修复(#75): 按数量精确匹配套餐列表促销
- loadSubscribePromoMap 返回 map[subscribeID]map[quantity]*SubscribePromo 二级映射 - 候选查询按 subscribe_id、quantity、priority DESC 排序,一次取出所有 quantity 档位 - 下单路径 QueryEligibleRules 将 quantity 条件移至 JOIN,精确匹配 - 永久订阅(expire_time=0)用 CASE WHEN 排序兜底,不再误判为回归促销 - 新增 promoEligibility、model、subscribe promo 单元测试 Co-authored-by: multica-agent <github@multica.ai>
This commit is contained in:
@@ -11,6 +11,7 @@ import (
|
|||||||
"github.com/perfect-panel/server/pkg/xerr"
|
"github.com/perfect-panel/server/pkg/xerr"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/clause"
|
||||||
)
|
)
|
||||||
|
|
||||||
type PromoResult struct {
|
type PromoResult struct {
|
||||||
@@ -148,7 +149,12 @@ func evaluateInactiveUserPromo(
|
|||||||
err := db.WithContext(ctx).
|
err := db.WithContext(ctx).
|
||||||
Model(&user.Subscribe{}).
|
Model(&user.Subscribe{}).
|
||||||
Where("user_id = ?", userID).
|
Where("user_id = ?", userID).
|
||||||
Order("expire_time DESC").
|
Order(clause.OrderBy{
|
||||||
|
Expression: clause.Expr{
|
||||||
|
SQL: "CASE WHEN expire_time = ? THEN 0 ELSE 1 END, expire_time DESC",
|
||||||
|
Vars: []interface{}{time.UnixMilli(0)},
|
||||||
|
},
|
||||||
|
}).
|
||||||
Limit(1).
|
Limit(1).
|
||||||
Take(&lastSub).Error
|
Take(&lastSub).Error
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -158,8 +164,16 @@ func evaluateInactiveUserPromo(
|
|||||||
return false, time.Time{}, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query promo inactive user subscription failed")
|
return false, time.Time{}, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query promo inactive user subscription failed")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return evaluateInactiveUserExpire(lastSub.ExpireTime, params, now), ruleExpiresAt, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func evaluateInactiveUserExpire(lastExpire time.Time, params promoRuleParams, now time.Time) bool {
|
||||||
|
if lastExpire.Equal(time.UnixMilli(0)) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
threshold := now.AddDate(0, -params.InactiveMonths, 0)
|
threshold := now.AddDate(0, -params.InactiveMonths, 0)
|
||||||
return lastSub.ExpireTime.Before(threshold) || lastSub.ExpireTime.Equal(threshold), ruleExpiresAt, nil
|
return lastExpire.Before(threshold) || lastExpire.Equal(threshold)
|
||||||
}
|
}
|
||||||
|
|
||||||
func promoRuleExpiresAt(rule *promo.RuleWithPrice) time.Time {
|
func promoRuleExpiresAt(rule *promo.RuleWithPrice) time.Time {
|
||||||
|
|||||||
@@ -0,0 +1,46 @@
|
|||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestEvaluateInactiveUserExpire(t *testing.T) {
|
||||||
|
now := time.Date(2026, 5, 27, 8, 0, 0, 0, time.UTC)
|
||||||
|
params := promoRuleParams{InactiveMonths: 3}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
lastExpire time.Time
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "permanent subscription is not inactive",
|
||||||
|
lastExpire: time.UnixMilli(0),
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "active subscription is not inactive",
|
||||||
|
lastExpire: now.Add(time.Hour),
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "expire at threshold is inactive",
|
||||||
|
lastExpire: now.AddDate(0, -3, 0),
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "expire before threshold is inactive",
|
||||||
|
lastExpire: now.AddDate(0, -3, -1),
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := evaluateInactiveUserExpire(tt.lastExpire, params, now); got != tt.want {
|
||||||
|
t.Fatalf("evaluateInactiveUserExpire() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -32,9 +32,7 @@ func (m *fakePromoModel) InsertUsage(context.Context, *promo.Usage, ...*gorm.DB)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestCalculatePurchasePricePromoSkipsPercentDiscount(t *testing.T) {
|
func TestCalculatePurchasePricePromoSkipsPercentDiscount(t *testing.T) {
|
||||||
svcCtx := &svc.ServiceContext{
|
model := &fakePromoModel{rules: []*promo.RuleWithPrice{
|
||||||
DB: &gorm.DB{},
|
|
||||||
PromoModel: &fakePromoModel{rules: []*promo.RuleWithPrice{
|
|
||||||
{
|
{
|
||||||
Rule: promo.Rule{
|
Rule: promo.Rule{
|
||||||
Id: 9,
|
Id: 9,
|
||||||
@@ -44,7 +42,10 @@ func TestCalculatePurchasePricePromoSkipsPercentDiscount(t *testing.T) {
|
|||||||
},
|
},
|
||||||
PromoPrice: 600,
|
PromoPrice: 600,
|
||||||
},
|
},
|
||||||
}},
|
}}
|
||||||
|
svcCtx := &svc.ServiceContext{
|
||||||
|
DB: &gorm.DB{},
|
||||||
|
PromoModel: model,
|
||||||
}
|
}
|
||||||
|
|
||||||
result, err := calculatePurchasePrice(
|
result, err := calculatePurchasePrice(
|
||||||
@@ -77,12 +78,13 @@ func TestCalculatePurchasePricePromoSkipsPercentDiscount(t *testing.T) {
|
|||||||
if result.PromoDiscount != 1200 {
|
if result.PromoDiscount != 1200 {
|
||||||
t.Fatalf("PromoDiscount = %d, want 1200", result.PromoDiscount)
|
t.Fatalf("PromoDiscount = %d, want 1200", result.PromoDiscount)
|
||||||
}
|
}
|
||||||
|
if model.lastQuantity != 3 {
|
||||||
|
t.Fatalf("promo query quantity = %d, want 3", model.lastQuantity)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCalculatePurchasePriceIgnoresInvalidPromoPrice(t *testing.T) {
|
func TestCalculatePurchasePriceIgnoresInvalidPromoPrice(t *testing.T) {
|
||||||
svcCtx := &svc.ServiceContext{
|
model := &fakePromoModel{rules: []*promo.RuleWithPrice{
|
||||||
DB: &gorm.DB{},
|
|
||||||
PromoModel: &fakePromoModel{rules: []*promo.RuleWithPrice{
|
|
||||||
{
|
{
|
||||||
Rule: promo.Rule{
|
Rule: promo.Rule{
|
||||||
Id: 10,
|
Id: 10,
|
||||||
@@ -92,7 +94,10 @@ func TestCalculatePurchasePriceIgnoresInvalidPromoPrice(t *testing.T) {
|
|||||||
},
|
},
|
||||||
PromoPrice: 1000,
|
PromoPrice: 1000,
|
||||||
},
|
},
|
||||||
}},
|
}}
|
||||||
|
svcCtx := &svc.ServiceContext{
|
||||||
|
DB: &gorm.DB{},
|
||||||
|
PromoModel: model,
|
||||||
}
|
}
|
||||||
|
|
||||||
result, err := calculatePurchasePrice(
|
result, err := calculatePurchasePrice(
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ import (
|
|||||||
"github.com/perfect-panel/server/pkg/xerr"
|
"github.com/perfect-panel/server/pkg/xerr"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/clause"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -89,7 +90,16 @@ func loadSubscribePromoMap(ctx context.Context, svcCtx *svc.ServiceContext, subs
|
|||||||
|
|
||||||
func querySubscribePromoCandidates(ctx context.Context, svcCtx *svc.ServiceContext, subscribeIDs []int64, loggedIn bool) ([]subscribePromoCandidate, error) {
|
func querySubscribePromoCandidates(ctx context.Context, svcCtx *svc.ServiceContext, subscribeIDs []int64, loggedIn bool) ([]subscribePromoCandidate, error) {
|
||||||
var candidates []subscribePromoCandidate
|
var candidates []subscribePromoCandidate
|
||||||
query := svcCtx.DB.WithContext(ctx).
|
err := subscribePromoCandidatesQuery(ctx, svcCtx.DB, subscribeIDs, loggedIn).
|
||||||
|
Scan(&candidates).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query subscribe promo candidates failed: %v", err)
|
||||||
|
}
|
||||||
|
return candidates, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func subscribePromoCandidatesQuery(ctx context.Context, db *gorm.DB, subscribeIDs []int64, loggedIn bool) *gorm.DB {
|
||||||
|
query := db.WithContext(ctx).
|
||||||
Table("subscribe_promo AS sp").
|
Table("subscribe_promo AS sp").
|
||||||
Select("sp.subscribe_id, sp.quantity, sp.promo_price, pr.name AS rule_name, pr.type AS rule_type, pr.params, pr.start_time, pr.end_time").
|
Select("sp.subscribe_id, sp.quantity, sp.promo_price, pr.name AS rule_name, pr.type AS rule_type, pr.params, pr.start_time, pr.end_time").
|
||||||
Joins("JOIN promo_rule AS pr ON pr.id = sp.promo_rule_id AND pr.deleted_at IS NULL").
|
Joins("JOIN promo_rule AS pr ON pr.id = sp.promo_rule_id AND pr.deleted_at IS NULL").
|
||||||
@@ -97,16 +107,11 @@ func querySubscribePromoCandidates(ctx context.Context, svcCtx *svc.ServiceConte
|
|||||||
if !loggedIn {
|
if !loggedIn {
|
||||||
query = query.Where("pr.type = ?", promoRuleTypeCampaign)
|
query = query.Where("pr.type = ?", promoRuleTypeCampaign)
|
||||||
}
|
}
|
||||||
err := query.
|
return query.
|
||||||
Order("sp.subscribe_id ASC").
|
Order("sp.subscribe_id ASC").
|
||||||
Order("sp.quantity ASC").
|
Order("sp.quantity ASC").
|
||||||
Order("pr.priority DESC").
|
Order("pr.priority DESC").
|
||||||
Order("pr.id ASC").
|
Order("pr.id ASC")
|
||||||
Scan(&candidates).Error
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query subscribe promo candidates failed: %v", err)
|
|
||||||
}
|
|
||||||
return candidates, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c subscribePromoCandidate) isActive(now time.Time) bool {
|
func (c subscribePromoCandidate) isActive(now time.Time) bool {
|
||||||
@@ -179,11 +184,7 @@ func (e *promoEligibilityEvaluator) lastSubscribeExpireAt() (time.Time, error) {
|
|||||||
return *e.lastExpire, nil
|
return *e.lastExpire, nil
|
||||||
}
|
}
|
||||||
var item user.Subscribe
|
var item user.Subscribe
|
||||||
err := e.db.WithContext(e.ctx).
|
err := e.lastSubscribeExpireQuery().
|
||||||
Model(&user.Subscribe{}).
|
|
||||||
Where("user_id = ?", e.userInfo.Id).
|
|
||||||
Where("expire_time != ?", time.UnixMilli(0)).
|
|
||||||
Order("expire_time DESC").
|
|
||||||
Limit(1).
|
Limit(1).
|
||||||
Take(&item).Error
|
Take(&item).Error
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -198,6 +199,18 @@ func (e *promoEligibilityEvaluator) lastSubscribeExpireAt() (time.Time, error) {
|
|||||||
return item.ExpireTime, nil
|
return item.ExpireTime, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (e *promoEligibilityEvaluator) lastSubscribeExpireQuery() *gorm.DB {
|
||||||
|
return e.db.WithContext(e.ctx).
|
||||||
|
Model(&user.Subscribe{}).
|
||||||
|
Where("user_id = ?", e.userInfo.Id).
|
||||||
|
Order(clause.OrderBy{
|
||||||
|
Expression: clause.Expr{
|
||||||
|
SQL: "CASE WHEN expire_time = ? THEN 0 ELSE 1 END, expire_time DESC",
|
||||||
|
Vars: []interface{}{time.UnixMilli(0)},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func (c subscribePromoCandidate) expiresAt() time.Time {
|
func (c subscribePromoCandidate) expiresAt() time.Time {
|
||||||
if c.EndTime == nil {
|
if c.EndTime == nil {
|
||||||
return time.Time{}
|
return time.Time{}
|
||||||
|
|||||||
@@ -1,10 +1,15 @@
|
|||||||
package subscribe
|
package subscribe
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/perfect-panel/server/internal/model/user"
|
"github.com/perfect-panel/server/internal/model/user"
|
||||||
|
"github.com/perfect-panel/server/internal/types"
|
||||||
|
"gorm.io/driver/mysql"
|
||||||
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestPromoEligibilityEvaluatorMatch(t *testing.T) {
|
func TestPromoEligibilityEvaluatorMatch(t *testing.T) {
|
||||||
@@ -73,3 +78,83 @@ func TestSubscribePromoCandidateActiveWindow(t *testing.T) {
|
|||||||
t.Fatal("candidate after end time should not be active")
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQuerySubscribePromoCandidatesIncludesQuantity(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)
|
||||||
|
}
|
||||||
|
|
||||||
|
var candidates []subscribePromoCandidate
|
||||||
|
tx := subscribePromoCandidatesQuery(context.Background(), db, []int64{11, 12}, true).Scan(&candidates)
|
||||||
|
stmt := tx.Statement
|
||||||
|
sql := stmt.SQL.String()
|
||||||
|
if !strings.Contains(sql, "sp.subscribe_id, sp.quantity, sp.promo_price") {
|
||||||
|
t.Fatalf("SQL missing quantity select: %s", sql)
|
||||||
|
}
|
||||||
|
if !strings.Contains(sql, "ORDER BY sp.subscribe_id ASC,sp.quantity ASC,pr.priority DESC,pr.id ASC") {
|
||||||
|
t.Fatalf("SQL missing quantity order: %s", sql)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestApplySubscribeDiscountPromosMatchesQuantity(t *testing.T) {
|
||||||
|
subscribe := types.Subscribe{Discount: []types.SubscribeDiscount{
|
||||||
|
{Quantity: 1},
|
||||||
|
{Quantity: 12},
|
||||||
|
}}
|
||||||
|
promos := map[int64]*types.SubscribePromo{
|
||||||
|
3: {RuleName: "季度优惠", PromoPrice: 2900},
|
||||||
|
12: {RuleName: "年度优惠", PromoPrice: 9900},
|
||||||
|
}
|
||||||
|
|
||||||
|
applySubscribeDiscountPromos(&subscribe, promos)
|
||||||
|
if subscribe.Discount[0].Promo != nil {
|
||||||
|
t.Fatalf("quantity 1 promo should be nil, got %+v", subscribe.Discount[0].Promo)
|
||||||
|
}
|
||||||
|
if subscribe.Discount[1].Promo == nil {
|
||||||
|
t.Fatal("quantity 12 promo should match")
|
||||||
|
}
|
||||||
|
if got, want := subscribe.Discount[1].Promo.RuleName, "年度优惠"; got != want {
|
||||||
|
t.Fatalf("promo rule name = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
|
||||||
|
subscribe = types.Subscribe{Discount: []types.SubscribeDiscount{{Quantity: 6}}}
|
||||||
|
applySubscribeDiscountPromos(&subscribe, promos)
|
||||||
|
if subscribe.Discount[0].Promo != nil {
|
||||||
|
t.Fatalf("promo should be nil when quantity does not match, got %+v", subscribe.Discount[0].Promo)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -27,18 +27,22 @@ func NewModel(db *gorm.DB, _ *redis.Client) Model {
|
|||||||
|
|
||||||
func (m *defaultPromoModel) QueryEligibleRules(ctx context.Context, subscribeId int64, quantity int64) ([]*RuleWithPrice, error) {
|
func (m *defaultPromoModel) QueryEligibleRules(ctx context.Context, subscribeId int64, quantity int64) ([]*RuleWithPrice, error) {
|
||||||
var list []*RuleWithPrice
|
var list []*RuleWithPrice
|
||||||
err := m.db.WithContext(ctx).
|
err := m.eligibleRulesQuery(ctx, subscribeId, quantity).
|
||||||
Table("promo_rule AS pr").
|
|
||||||
Select("pr.*, sp.promo_price").
|
|
||||||
Joins("JOIN subscribe_promo AS sp ON sp.promo_rule_id = pr.id").
|
|
||||||
Where("sp.subscribe_id = ? AND sp.quantity = ? AND sp.promo_price > 0 AND pr.enabled = ?", subscribeId, quantity, true).
|
|
||||||
Where("pr.deleted_at IS NULL").
|
|
||||||
Order("pr.priority DESC").
|
|
||||||
Order("pr.id ASC").
|
|
||||||
Find(&list).Error
|
Find(&list).Error
|
||||||
return list, err
|
return list, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (m *defaultPromoModel) eligibleRulesQuery(ctx context.Context, subscribeId int64, quantity int64) *gorm.DB {
|
||||||
|
return m.db.WithContext(ctx).
|
||||||
|
Table("promo_rule AS pr").
|
||||||
|
Select("pr.*, sp.promo_price").
|
||||||
|
Joins("JOIN subscribe_promo AS sp ON sp.promo_rule_id = pr.id AND sp.quantity = ?", quantity).
|
||||||
|
Where("sp.subscribe_id = ? AND sp.promo_price > 0 AND pr.enabled = ?", subscribeId, true).
|
||||||
|
Where("pr.deleted_at IS NULL").
|
||||||
|
Order("pr.priority DESC").
|
||||||
|
Order("pr.id ASC")
|
||||||
|
}
|
||||||
|
|
||||||
func (m *defaultPromoModel) InsertUsage(ctx context.Context, data *Usage, tx ...*gorm.DB) error {
|
func (m *defaultPromoModel) InsertUsage(ctx context.Context, data *Usage, tx ...*gorm.DB) error {
|
||||||
db := m.db.WithContext(ctx)
|
db := m.db.WithContext(ctx)
|
||||||
if len(tx) > 0 {
|
if len(tx) > 0 {
|
||||||
|
|||||||
@@ -0,0 +1,39 @@
|
|||||||
|
package promo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gorm.io/driver/mysql"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestQueryEligibleRulesFiltersByQuantity(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)
|
||||||
|
}
|
||||||
|
|
||||||
|
model := &defaultPromoModel{db: db}
|
||||||
|
var list []*RuleWithPrice
|
||||||
|
tx := model.eligibleRulesQuery(context.Background(), 11, 3).Find(&list)
|
||||||
|
stmt := tx.Statement
|
||||||
|
sql := stmt.SQL.String()
|
||||||
|
if !strings.Contains(sql, "JOIN subscribe_promo AS sp ON sp.promo_rule_id = pr.id AND sp.quantity = ?") {
|
||||||
|
t.Fatalf("SQL missing quantity join condition: %s", sql)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(stmt.Vars) < 2 {
|
||||||
|
t.Fatalf("SQL vars length = %d, want at least 2; vars=%v", len(stmt.Vars), stmt.Vars)
|
||||||
|
}
|
||||||
|
if got, want := stmt.Vars[0], int64(3); got != want {
|
||||||
|
t.Fatalf("first SQL var = %v, want quantity %d; vars=%v", got, want, stmt.Vars)
|
||||||
|
}
|
||||||
|
if got, want := stmt.Vars[1], int64(11); got != want {
|
||||||
|
t.Fatalf("second SQL var = %v, want subscribe_id %d; vars=%v", got, want, stmt.Vars)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user