110 lines
3.5 KiB
Go
110 lines
3.5 KiB
Go
package subscribe
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/perfect-panel/server/internal/model/user"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestPromoEligibilityEvaluatorMatch(t *testing.T) {
|
|
now := time.Unix(1710000000, 0)
|
|
campaignEnd := now.Add(2 * time.Hour)
|
|
|
|
campaign := subscribePromoCandidate{
|
|
RuleName: "限时活动",
|
|
RuleType: promoRuleTypeCampaign,
|
|
PromoPrice: 99,
|
|
EndTime: &campaignEnd,
|
|
}
|
|
ok, expiresAt, err := (&promoEligibilityEvaluator{}).match(campaign, now)
|
|
if err != nil {
|
|
t.Fatalf("campaign match error: %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatal("campaign promo should match without login")
|
|
}
|
|
if got, want := unixSeconds(expiresAt), campaignEnd.Unix(); got != want {
|
|
t.Fatalf("campaign expires_at = %d, want %d", got, want)
|
|
}
|
|
|
|
newUser := subscribePromoCandidate{
|
|
RuleName: "新客7天优惠",
|
|
RuleType: promoRuleTypeNewUser,
|
|
PromoPrice: 279,
|
|
Params: `{"window_hours":168}`,
|
|
}
|
|
ok, _, err = (&promoEligibilityEvaluator{}).match(newUser, now)
|
|
if err != nil {
|
|
t.Fatalf("anonymous new_user match error: %v", err)
|
|
}
|
|
if ok {
|
|
t.Fatal("new_user promo should not match without login")
|
|
}
|
|
|
|
userInfo := &user.User{Id: 1, CreatedAt: now.Add(-24 * time.Hour)}
|
|
ok, expiresAt, err = (&promoEligibilityEvaluator{userInfo: userInfo}).match(newUser, now)
|
|
if err != nil {
|
|
t.Fatalf("logged-in new_user match error: %v", err)
|
|
}
|
|
if !ok {
|
|
t.Fatal("new_user promo should match inside window")
|
|
}
|
|
if got, want := unixSeconds(expiresAt), userInfo.CreatedAt.Add(168*time.Hour).Unix(); got != want {
|
|
t.Fatalf("new_user expires_at = %d, want %d", got, want)
|
|
}
|
|
}
|
|
|
|
func TestSubscribePromoCandidateActiveWindow(t *testing.T) {
|
|
now := time.Unix(1710000000, 0)
|
|
start := now.Add(-time.Hour)
|
|
end := now.Add(time.Hour)
|
|
|
|
if !(subscribePromoCandidate{PromoPrice: 1, StartTime: &start, EndTime: &end}).isActive(now) {
|
|
t.Fatal("candidate inside active window should be active")
|
|
}
|
|
if (subscribePromoCandidate{PromoPrice: 0, StartTime: &start, EndTime: &end}).isActive(now) {
|
|
t.Fatal("candidate with zero promo price should not be active")
|
|
}
|
|
if (subscribePromoCandidate{PromoPrice: 1, StartTime: &end}).isActive(now) {
|
|
t.Fatal("candidate before start time should not be active")
|
|
}
|
|
if (subscribePromoCandidate{PromoPrice: 1, EndTime: &start}).isActive(now) {
|
|
t.Fatal("candidate after end time should not be active")
|
|
}
|
|
}
|
|
|
|
func TestLastSubscribeExpireAtPrioritizesPermanentSubscription(t *testing.T) {
|
|
db, err := gorm.Open(mysql.New(mysql.Config{
|
|
DSN: "gorm:gorm@tcp(localhost:9910)/gorm?charset=utf8&parseTime=True&loc=Local",
|
|
SkipInitializeWithVersion: true,
|
|
}), &gorm.Config{DryRun: true, DisableAutomaticPing: true})
|
|
if err != nil {
|
|
t.Fatalf("open dry-run db: %v", err)
|
|
}
|
|
|
|
evaluator := &promoEligibilityEvaluator{
|
|
db: db,
|
|
userInfo: &user.User{Id: 7},
|
|
}
|
|
var item user.Subscribe
|
|
tx := evaluator.lastSubscribeExpireQuery().Limit(1).Take(&item)
|
|
|
|
sql := tx.Statement.SQL.String()
|
|
if !strings.Contains(sql, "CASE WHEN expire_time = ? THEN 0 ELSE 1 END") {
|
|
t.Fatalf("SQL missing permanent subscription priority order: %s", sql)
|
|
}
|
|
if strings.Contains(sql, "expire_time !=") {
|
|
t.Fatalf("SQL should not filter out permanent subscriptions: %s", sql)
|
|
}
|
|
if len(tx.Statement.Vars) < 2 {
|
|
t.Fatalf("SQL vars length = %d, want at least 2; vars=%v", len(tx.Statement.Vars), tx.Statement.Vars)
|
|
}
|
|
if got, want := tx.Statement.Vars[1], time.UnixMilli(0); got != want {
|
|
t.Fatalf("permanent subscription order var = %v, want %v; vars=%v", got, want, tx.Statement.Vars)
|
|
}
|
|
}
|