Files
hi-server/internal/model/promo/model_test.go
T
2026-05-27 06:03:25 -07:00

40 lines
1.2 KiB
Go

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)
}
}