junhong_cmp_fiber/internal/service/package/activation_terms_integration_test.go

package packagepkg

import (
	"context"
	"strconv"
	"testing"
	"time"

	"github.com/break/junhong_cmp_fiber/internal/model"
	"github.com/break/junhong_cmp_fiber/internal/testutil"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

// TestActivateByRealnameUsesPurchasedSnapshot 验证实名激活只使用购买快照计算起止时间。
func TestActivateByRealnameUsesPurchasedSnapshot(t *testing.T) {
	tx := testutil.NewPostgresTransaction(t)
	card := createActivationTermsCard(t, tx, constants.RealNameStatusVerified)
	pkg := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 99)
	purchasedAt := time.Now().Add(-48 * time.Hour).Truncate(time.Second)
	usage := createActivationTermsUsage(t, tx, pkg.ID, card.ID, 1, constants.PackageExpiryBaseFromPurchase, 10, true, purchasedAt)
	service := NewActivationService(tx, nil, nil, nil, nil, zap.NewNop())
	if err := service.ActivateByRealname(context.Background(), constants.AssetTypeIotCard, card.ID); err != nil {
		t.Fatalf("实名激活失败：%v", err)
	}
	var refreshed model.PackageUsage
	if err := tx.First(&refreshed, usage.ID).Error; err != nil {
		t.Fatalf("查询激活结果失败：%v", err)
	}
	expectedExpiry := CalculateExpiryTime(constants.PackageCalendarTypeByDay, refreshed.CreatedAt, 0, 10)
	if refreshed.ActivatedAt == nil || refreshed.ExpiresAt == nil || refreshed.ActivatedAt.Unix() != refreshed.CreatedAt.Unix() || refreshed.ExpiresAt.Unix() != expectedExpiry.Unix() {
		t.Fatalf("实名激活未使用购买快照：%+v", refreshed)
	}
}

// TestActivateNextPendingMainPackageUsesQueueSnapshotAndIsIdempotent 验证连续排队按优先级和各自快照接续。
func TestActivateNextPendingMainPackageUsesQueueSnapshotAndIsIdempotent(t *testing.T) {
	tx := testutil.NewPostgresTransaction(t)
	redisClient := testutil.NewRedisClient(t)
	card := createActivationTermsCard(t, tx, constants.RealNameStatusVerified)
	firstPackage := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 90)
	secondPackage := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 90)
	first := createActivationTermsUsage(t, tx, firstPackage.ID, card.ID, 1, constants.PackageExpiryBaseFromActivation, 3, false, time.Now().Add(-time.Hour))
	second := createActivationTermsUsage(t, tx, secondPackage.ID, card.ID, 2, constants.PackageExpiryBaseFromActivation, 4, false, time.Now())
	service := NewActivationService(tx, redisClient, nil, nil, nil, zap.NewNop())
	activated, err := service.ActivateNextPendingMainPackage(context.Background(), constants.AssetTypeIotCard, card.ID)
	if err != nil || !activated {
		t.Fatalf("激活队首套餐失败：activated=%v err=%v", activated, err)
	}
	activated, err = service.ActivateNextPendingMainPackage(context.Background(), constants.AssetTypeIotCard, card.ID)
	if err != nil || activated {
		t.Fatalf("已有生效主套餐时重复接续应幂等：activated=%v err=%v", activated, err)
	}
	if err := tx.Model(&model.PackageUsage{}).Where("id = ?", first.ID).Update("status", constants.PackageUsageStatusExpired).Error; err != nil {
		t.Fatalf("结束队首套餐失败：%v", err)
	}
	activated, err = service.ActivateNextPendingMainPackage(context.Background(), constants.AssetTypeIotCard, card.ID)
	if err != nil || !activated {
		t.Fatalf("激活第二个排队套餐失败：activated=%v err=%v", activated, err)
	}
	var refreshed []model.PackageUsage
	if err := tx.Where("id IN ?", []uint{first.ID, second.ID}).Order("priority").Find(&refreshed).Error; err != nil {
		t.Fatalf("查询队列接续结果失败：%v", err)
	}
	if len(refreshed) != 2 || refreshed[1].ExpiresAt == nil || refreshed[1].ActivatedAt == nil || !refreshed[1].ExpiresAt.Equal(CalculateExpiryTime(constants.PackageCalendarTypeByDay, *refreshed[1].ActivatedAt, 0, 4)) {
		t.Fatalf("第二个排队套餐未使用自身快照：%+v", refreshed)
	}
}

// TestRefundFollowUpActivatesNextPackageFromSnapshot 验证退款失效后下一套餐按购买快照接续。
func TestRefundFollowUpActivatesNextPackageFromSnapshot(t *testing.T) {
	tx := testutil.NewPostgresTransaction(t)
	redisClient := testutil.NewRedisClient(t)
	card := createActivationTermsCard(t, tx, constants.RealNameStatusVerified)
	currentPackage := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 60)
	nextPackage := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 60)
	current := createActivationTermsUsage(t, tx, currentPackage.ID, card.ID, 1, constants.PackageExpiryBaseFromPurchase, 30, false, time.Now().Add(-time.Hour))
	now := time.Now()
	if err := tx.Model(current).Updates(map[string]any{"status": constants.PackageUsageStatusActive, "activated_at": now, "expires_at": now.AddDate(0, 0, 30)}).Error; err != nil {
		t.Fatalf("设置当前生效套餐失败：%v", err)
	}
	next := createActivationTermsUsage(t, tx, nextPackage.ID, card.ID, 2, constants.PackageExpiryBaseFromActivation, 6, false, time.Now())
	service := NewActivationService(tx, redisClient, nil, nil, nil, zap.NewNop())
	if err := service.InvalidatePackagesForRefund(context.Background(), constants.AssetTypeIotCard, card.ID, current.OrderID, 77, "UR55-REFUND", &current.ID); err != nil {
		t.Fatalf("退款失效当前套餐失败：%v", err)
	}
	activated, err := service.ActivateNextPendingMainPackage(context.Background(), constants.AssetTypeIotCard, card.ID)
	if err != nil || !activated {
		t.Fatalf("退款后接续下一套餐失败：activated=%v err=%v", activated, err)
	}
	var refreshed model.PackageUsage
	if err := tx.First(&refreshed, next.ID).Error; err != nil {
		t.Fatalf("查询退款接续结果失败：%v", err)
	}
	if refreshed.ExpiresAt == nil || refreshed.ActivatedAt == nil || !refreshed.ExpiresAt.Equal(CalculateExpiryTime(constants.PackageCalendarTypeByDay, *refreshed.ActivatedAt, 0, 6)) {
		t.Fatalf("退款接续未使用下一套餐快照：%+v", refreshed)
	}
}

// TestActivateSpecificPackageRejectsPartialSnapshot 验证新记录非法快照不会静默回退当前套餐配置。
func TestActivateSpecificPackageRejectsPartialSnapshot(t *testing.T) {
	tx := testutil.NewPostgresTransaction(t)
	redisClient := testutil.NewRedisClient(t)
	card := createActivationTermsCard(t, tx, constants.RealNameStatusVerified)
	pkg := createActivationTermsPackage(t, tx, constants.PackageExpiryBaseFromPurchase, 30)
	usage := createActivationTermsUsage(t, tx, pkg.ID, card.ID, 1, constants.PackageExpiryBaseFromPurchase, 30, false, time.Now())
	if err := tx.Exec("ALTER TABLE tb_package_usage DISABLE TRIGGER trg_validate_package_usage_terms_snapshot").Error; err != nil {
		t.Fatalf("禁用快照校验触发器失败：%v", err)
	}
	if err := tx.Model(usage).Update("calendar_type_snapshot", "invalid").Error; err != nil {
		t.Fatalf("构造非法快照失败：%v", err)
	}
	if err := tx.Exec("ALTER TABLE tb_package_usage ENABLE TRIGGER trg_validate_package_usage_terms_snapshot").Error; err != nil {
		t.Fatalf("恢复快照校验触发器失败：%v", err)
	}
	service := NewActivationService(tx, redisClient, nil, nil, nil, zap.NewNop())
	if err := service.ActivateSpecificPackage(context.Background(), usage.ID); err == nil {
		t.Fatal("非法部分快照必须拒绝激活")
	}
	var refreshed model.PackageUsage
	if err := tx.First(&refreshed, usage.ID).Error; err != nil || refreshed.Status != constants.PackageUsageStatusPending {
		t.Fatalf("非法快照失败后状态必须保持待生效：status=%d err=%v", refreshed.Status, err)
	}
}

func createActivationTermsCard(t *testing.T, tx *gorm.DB, realnameStatus int) *model.IotCard {
	t.Helper()
	suffix := strconv.FormatInt(time.Now().UnixNano()%100000000000000000, 10)
	iccid := "87" + suffix
	if len(iccid) < 19 {
		iccid += "0000000000000000000"[:19-len(iccid)]
	}
	iccid = iccid[:19]
	card := &model.IotCard{ICCID: iccid, ICCID19: iccid, CarrierID: 1, RealNameStatus: realnameStatus, AssetStatus: constants.AssetStatusInStock, Generation: 1}
	if err := tx.Create(card).Error; err != nil {
		t.Fatalf("创建激活测试卡失败：%v", err)
	}
	return card
}

func createActivationTermsPackage(t *testing.T, tx *gorm.DB, expiryBase string, durationDays int) *model.Package {
	t.Helper()
	suffix := strconv.FormatInt(time.Now().UnixNano(), 10)
	pkg := &model.Package{PackageCode: "UR55-ACTIVATE-" + suffix, PackageName: "UR55激活测试套餐", PackageType: constants.PackageTypeFormal, DurationMonths: 1, DurationDays: durationDays, CalendarType: constants.PackageCalendarTypeByDay, ExpiryBase: expiryBase, Status: constants.StatusEnabled, ShelfStatus: 1, DataResetCycle: "monthly"}
	if err := tx.Create(pkg).Error; err != nil {
		t.Fatalf("创建激活测试套餐失败：%v", err)
	}
	return pkg
}

func createActivationTermsUsage(t *testing.T, tx *gorm.DB, packageID, cardID uint, priority int, expiryBase string, durationDays int, pendingRealname bool, createdAt time.Time) *model.PackageUsage {
	t.Helper()
	unique := uint(time.Now().UnixNano() & 0x7fffffff)
	usage := &model.PackageUsage{Model: gorm.Model{CreatedAt: createdAt}, OrderID: unique, OrderNo: "UR55-ACTIVATE-" + strconv.FormatUint(uint64(unique), 10), PackageID: packageID, PackageName: "UR55激活测试套餐", UsageType: constants.AssetWalletResourceTypeIotCard, IotCardID: cardID, DataLimitMB: 1, Status: constants.PackageUsageStatusPending, Priority: priority, PendingRealnameActivation: pendingRealname, Generation: 1, ExpiryBaseSnapshot: expiryBase, CalendarTypeSnapshot: constants.PackageCalendarTypeByDay, DurationDaysSnapshot: durationDays}
	if err := tx.Omit("status", "pending_realname_activation").Create(usage).Error; err != nil {
		t.Fatalf("创建激活测试使用记录失败：%v", err)
	}
	if err := tx.Model(usage).Updates(map[string]any{"status": constants.PackageUsageStatusPending, "pending_realname_activation": pendingRealname}).Error; err != nil {
		t.Fatalf("设置激活测试状态失败：%v", err)
	}
	return usage
}