Files
affiliate_dash/backend/internal/service/fulfillment_test.go
T
yml2213 bafe41a3ea 重构: 手续费改为百分比/固定二选一,清理旧 Skin/Order/ShipLog 演示链路
手续费:
- 新增 fee_type 字段(rate/fixed),百分比与固定金额二选一,不再叠加
- calculateServiceFee 按 fee_type 分支计算
- 商户创建/更新校验 fee_type,订单落库快照 fee_type
- 前端表单改为下拉选择手续费类型,动态显示对应输入框
- 测试拆为 TestFeeRate + TestFeeFixed

清理旧链路:
- 删除旧 Skin/Order/ShipLog 模型及 OrderService/SkinService
- Dashboard 迁移到 FulfillmentService
- 上游 /api/open/v1 改为基于 FulfillmentOrder 实现,接口契约不变
- 推送留痕改用 AuditLog,不再建 ShipLog 表
- 删除前端 Skins/Orders/ShipLogs/Distributors 页面及路由、菜单、API、类型
- 新增迁移 003: 添加 fee_type 列并 DROP 旧表
2026-07-30 14:12:27 +08:00

376 lines
12 KiB
Go

package service
import (
"strings"
"testing"
"affiliate_dash/internal/model"
"affiliate_dash/internal/testdb"
"gorm.io/gorm"
)
func newServiceTestDB(t *testing.T) *gorm.DB {
t.Helper()
return testdb.New(t,
&model.Merchant{},
&model.Product{},
&model.MerchantProduct{},
&model.WalletAccount{},
&model.WalletLedgerEntry{},
&model.FulfillmentOrder{},
&model.FulfillmentJob{},
&model.AuditLog{},
)
}
func seedFulfillmentMerchant(t *testing.T, db *gorm.DB, code string, balance, stock, price int64) (uint, model.MerchantProduct) {
t.Helper()
merchant := model.Merchant{Code: code, Name: code, Status: model.MerchantStatusActive}
if err := db.Create(&merchant).Error; err != nil {
t.Fatalf("create merchant: %v", err)
}
if err := db.Create(&model.WalletAccount{
MerchantID: merchant.ID,
Currency: "CNY",
AvailableBalance: balance,
}).Error; err != nil {
t.Fatalf("create wallet: %v", err)
}
product := model.Product{Code: code + "-product", Name: "测试商品", Status: model.ProductStatusActive}
if err := db.Create(&product).Error; err != nil {
t.Fatalf("create product: %v", err)
}
merchantProduct := model.MerchantProduct{
MerchantID: merchant.ID,
ProductID: product.ID,
SKU: "sku-basic",
DisplayName: "测试商品",
PriceAmount: price,
Currency: "CNY",
Stock: stock,
Status: model.ProductStatusActive,
}
if err := db.Create(&merchantProduct).Error; err != nil {
t.Fatalf("create merchant product: %v", err)
}
return merchant.ID, merchantProduct
}
func TestFulfillmentCreateOrderDebitsWalletAndIsIdempotent(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-a", 1000, 5, 200)
svc := NewFulfillmentService(db, nil)
first, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 11,
ClientOrderNo: "client-001",
SKU: product.SKU,
Quantity: 2,
RequestData: map[string]string{
"account": "player-1",
},
})
if err != nil {
t.Fatalf("create order: %v", err)
}
if first.Idempotent {
t.Fatalf("first create should not be idempotent")
}
if first.Order.Amount != 400 || first.Order.PaymentStatus != model.PaymentStatusPaid || first.Order.FulfillmentStatus != model.FulfillmentStatusPending {
t.Fatalf("unexpected order: %+v", first.Order)
}
second, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 11,
ClientOrderNo: "client-001",
SKU: product.SKU,
Quantity: 2,
})
if err != nil {
t.Fatalf("idempotent create: %v", err)
}
if !second.Idempotent || second.Order.OrderNo != first.Order.OrderNo {
t.Fatalf("expected existing order, got %+v", second)
}
var wallet model.WalletAccount
if err := db.Where("merchant_id = ?", merchantID).First(&wallet).Error; err != nil {
t.Fatalf("query wallet: %v", err)
}
if wallet.AvailableBalance != 600 {
t.Fatalf("wallet should debit once, got %d", wallet.AvailableBalance)
}
var refreshed model.MerchantProduct
if err := db.First(&refreshed, product.ID).Error; err != nil {
t.Fatalf("query product: %v", err)
}
if refreshed.Stock != 3 {
t.Fatalf("stock should decrease once, got %d", refreshed.Stock)
}
var ledgerCount int64
db.Model(&model.WalletLedgerEntry{}).Where("merchant_id = ?", merchantID).Count(&ledgerCount)
if ledgerCount != 1 {
t.Fatalf("ledger should have one debit entry, got %d", ledgerCount)
}
}
func TestFulfillmentCreateOrderAppliesMerchantFeeRate(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-fee-rate", 1000, 5, 200)
if err := db.Model(&model.Merchant{}).Where("id = ?", merchantID).Updates(map[string]interface{}{
"fee_type": model.FeeTypeRate,
"fee_rate_bp": int64(250),
}).Error; err != nil {
t.Fatalf("update merchant fee: %v", err)
}
svc := NewFulfillmentService(db, nil)
created, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 21,
ClientOrderNo: "client-fee-rate",
SKU: product.SKU,
Quantity: 2,
})
if err != nil {
t.Fatalf("create order: %v", err)
}
// baseAmount = 200 * 2 = 400; rate 250BP = 400 * 250 / 10000 = 10; total = 410
if created.Order.BaseAmount != 400 || created.Order.ServiceFeeAmount != 10 || created.Order.Amount != 410 {
t.Fatalf("unexpected rate fee snapshot: %+v", created.Order)
}
if created.Order.FeeType != model.FeeTypeRate || created.Order.FeeRateBP != 250 {
t.Fatalf("unexpected rate fee config snapshot: %+v", created.Order)
}
var wallet model.WalletAccount
if err := db.Where("merchant_id = ?", merchantID).First(&wallet).Error; err != nil {
t.Fatalf("query wallet: %v", err)
}
if wallet.AvailableBalance != 590 {
t.Fatalf("wallet should debit total amount, got %d", wallet.AvailableBalance)
}
}
func TestFulfillmentCreateOrderAppliesMerchantFeeFixed(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-fee-fixed", 1000, 5, 200)
if err := db.Model(&model.Merchant{}).Where("id = ?", merchantID).Updates(map[string]interface{}{
"fee_type": model.FeeTypeFixed,
"fee_fixed_amount": int64(30),
}).Error; err != nil {
t.Fatalf("update merchant fee: %v", err)
}
svc := NewFulfillmentService(db, nil)
created, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 22,
ClientOrderNo: "client-fee-fixed",
SKU: product.SKU,
Quantity: 2,
})
if err != nil {
t.Fatalf("create order: %v", err)
}
// baseAmount = 400; fixed fee = 30; total = 430
if created.Order.BaseAmount != 400 || created.Order.ServiceFeeAmount != 30 || created.Order.Amount != 430 {
t.Fatalf("unexpected fixed fee snapshot: %+v", created.Order)
}
if created.Order.FeeType != model.FeeTypeFixed || created.Order.FeeFixedAmount != 30 {
t.Fatalf("unexpected fixed fee config snapshot: %+v", created.Order)
}
var wallet model.WalletAccount
if err := db.Where("merchant_id = ?", merchantID).First(&wallet).Error; err != nil {
t.Fatalf("query wallet: %v", err)
}
if wallet.AvailableBalance != 570 {
t.Fatalf("wallet should debit total amount, got %d", wallet.AvailableBalance)
}
}
func TestFulfillmentCancelRefundsOnceAndRestoresStock(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-b", 1000, 2, 300)
svc := NewFulfillmentService(db, nil)
created, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 12,
ClientOrderNo: "client-cancel",
SKU: product.SKU,
Quantity: 1,
})
if err != nil {
t.Fatalf("create order: %v", err)
}
cancelled, err := svc.CancelOrder(merchantID, 12, created.Order.OrderNo, "用户取消")
if err != nil {
t.Fatalf("cancel order: %v", err)
}
if cancelled.PaymentStatus != model.PaymentStatusRefunded || cancelled.FulfillmentStatus != model.FulfillmentStatusCancelled {
t.Fatalf("unexpected cancelled order: %+v", cancelled)
}
if _, err := svc.CancelOrder(merchantID, 12, created.Order.OrderNo, "重复取消"); err != nil {
t.Fatalf("repeat cancel should be idempotent: %v", err)
}
var wallet model.WalletAccount
_ = db.Where("merchant_id = ?", merchantID).First(&wallet).Error
if wallet.AvailableBalance != 1000 {
t.Fatalf("wallet should refund once, got %d", wallet.AvailableBalance)
}
var refreshed model.MerchantProduct
_ = db.First(&refreshed, product.ID).Error
if refreshed.Stock != 2 {
t.Fatalf("stock should restore once, got %d", refreshed.Stock)
}
var ledgerCount int64
db.Model(&model.WalletLedgerEntry{}).Where("merchant_id = ?", merchantID).Count(&ledgerCount)
if ledgerCount != 2 {
t.Fatalf("ledger should have debit and refund, got %d", ledgerCount)
}
}
func TestFulfillmentStatusTransitions(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-c", 1000, -1, 100)
svc := NewFulfillmentService(db, nil)
created, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 13,
ClientOrderNo: "client-status",
SKU: product.SKU,
})
if err != nil {
t.Fatalf("create order: %v", err)
}
processing, err := svc.UpdateFulfillment(FulfillmentUpdateInput{
MerchantID: merchantID,
APIClientID: 13,
OrderNo: created.Order.OrderNo,
Status: model.FulfillmentStatusProcessing,
})
if err != nil {
t.Fatalf("mark processing: %v", err)
}
if processing.FulfillmentStatus != model.FulfillmentStatusProcessing {
t.Fatalf("expected processing, got %s", processing.FulfillmentStatus)
}
succeeded, err := svc.UpdateFulfillment(FulfillmentUpdateInput{
MerchantID: merchantID,
APIClientID: 13,
OrderNo: created.Order.OrderNo,
Status: model.FulfillmentStatusSucceeded,
ProviderOrderNo: "provider-1",
ResultData: map[string]string{"ok": "true"},
})
if err != nil {
t.Fatalf("mark succeeded: %v", err)
}
if succeeded.FulfillmentStatus != model.FulfillmentStatusSucceeded || succeeded.DeliveredAt == nil {
t.Fatalf("unexpected succeeded order: %+v", succeeded)
}
_, err = svc.UpdateFulfillment(FulfillmentUpdateInput{
MerchantID: merchantID,
APIClientID: 13,
OrderNo: created.Order.OrderNo,
Status: model.FulfillmentStatusFailed,
})
if err == nil {
t.Fatalf("should reject rollback after success")
}
}
func TestFulfillmentMerchantIsolation(t *testing.T) {
db := newServiceTestDB(t)
merchantA, productA := seedFulfillmentMerchant(t, db, "merchant-d", 1000, 1, 100)
merchantB, _ := seedFulfillmentMerchant(t, db, "merchant-e", 1000, 1, 100)
svc := NewFulfillmentService(db, nil)
created, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantA,
APIClientID: 14,
ClientOrderNo: "client-isolation",
SKU: productA.SKU,
})
if err != nil {
t.Fatalf("create order: %v", err)
}
if _, err := svc.GetOrder(merchantB, created.Order.OrderNo); err == nil {
t.Fatalf("other merchant should not read the order")
}
}
func TestWalletAdjustIsIdempotentPerMerchant(t *testing.T) {
db := newServiceTestDB(t)
merchantA, _ := seedFulfillmentMerchant(t, db, "merchant-f", 0, -1, 100)
merchantB, _ := seedFulfillmentMerchant(t, db, "merchant-g", 0, -1, 100)
svc := NewFulfillmentService(db, nil)
for _, merchantID := range []uint{merchantA, merchantB} {
wallet, err := svc.AdjustWallet(WalletAdjustInput{
MerchantID: merchantID,
ActorUserID: 1,
Amount: 500,
IdempotencyKey: "same-key",
Note: "充值",
})
if err != nil {
t.Fatalf("adjust wallet merchant %d: %v", merchantID, err)
}
if wallet.AvailableBalance != 500 {
t.Fatalf("unexpected balance for merchant %d: %d", merchantID, wallet.AvailableBalance)
}
}
wallet, err := svc.AdjustWallet(WalletAdjustInput{
MerchantID: merchantA,
ActorUserID: 1,
Amount: 500,
IdempotencyKey: "same-key",
Note: "重复充值",
})
if err != nil {
t.Fatalf("repeat adjust: %v", err)
}
if wallet.AvailableBalance != 500 {
t.Fatalf("repeat adjust should not change balance, got %d", wallet.AvailableBalance)
}
}
func TestCanFulfill(t *testing.T) {
ok, reason := CanFulfill(&model.FulfillmentOrder{
PaymentStatus: model.PaymentStatusPaid,
FulfillmentStatus: model.FulfillmentStatusFailed,
})
if !ok || reason != "" {
t.Fatalf("failed paid order should be fulfillable")
}
ok, _ = CanFulfill(&model.FulfillmentOrder{
PaymentStatus: model.PaymentStatusRefunded,
FulfillmentStatus: model.FulfillmentStatusPending,
})
if ok {
t.Fatalf("refunded order should not be fulfillable")
}
}
func TestCreateOrderRejectsInsufficientBalance(t *testing.T) {
db := newServiceTestDB(t)
merchantID, product := seedFulfillmentMerchant(t, db, "merchant-h", 50, 1, 100)
svc := NewFulfillmentService(db, nil)
_, err := svc.CreateOrder(CreateFulfillmentOrderInput{
MerchantID: merchantID,
APIClientID: 15,
ClientOrderNo: "client-low-balance",
SKU: product.SKU,
})
if err == nil || !strings.Contains(err.Error(), "余额不足") {
t.Fatalf("expected insufficient balance error, got %v", err)
}
var wallet model.WalletAccount
_ = db.Where("merchant_id = ?", merchantID).First(&wallet).Error
if wallet.AvailableBalance != 50 {
t.Fatalf("balance should remain unchanged, got %d", wallet.AvailableBalance)
}
}