package pickup import ( "encoding/json" "errors" "testing" "hfb_sys/backend/internal/auditlog" "hfb_sys/backend/internal/model" "gorm.io/datatypes" "gorm.io/driver/sqlite" "gorm.io/gorm" "gorm.io/gorm/logger" ) func TestBuildPickupPriceSnapshotSplitsOwnerPrice(t *testing.T) { raw, err := json.Marshal(map[string]any{ "price_breakdown": map[string]any{ "seller_coin_base_price": 233, "seller_total_price": 353, "platform_markup_amount": 30, "consumable_price": 120, "seller_ratio": 42.9, "buyer_ratio": 35.1, }, }) if err != nil { t.Fatalf("构造资产快照失败: %v", err) } snapshot := buildPickupPriceSnapshot( model.RentalListing{PriceCent: 38300}, model.GameAccount{HafCoinAmount: 100000000, AssetSummary: datatypes.JSON(raw)}, ) assertInt64(t, "网站售价", snapshot.ListingPriceCent, 38300) assertInt64(t, "号主纯比价格", snapshot.OwnerPurePriceCent, 23300) assertInt64(t, "额外物品价格", snapshot.OwnerExtraItemPriceCent, 12000) assertInt64(t, "号主总价", snapshot.OwnerTotalPriceCent, 35300) assertInt64(t, "兼容号主价", snapshot.OwnerPriceCent, 35300) assertInt64(t, "网站加价", snapshot.WebsiteProfitCent, 3000) if snapshot.SellerRatio != 42.9 || snapshot.BuyerRatio != 35.1 { t.Fatalf("比例 = %.2f/%.2f, want 42.90/35.10", snapshot.SellerRatio, snapshot.BuyerRatio) } } func TestBuildPickupPriceSnapshotFallsBackToSplitSum(t *testing.T) { raw, err := json.Marshal(map[string]any{ "price_breakdown": map[string]any{ "seller_coin_base_price": 200, "consumable_price": 50, }, }) if err != nil { t.Fatalf("构造资产快照失败: %v", err) } snapshot := buildPickupPriceSnapshot( model.RentalListing{PriceCent: 30000}, model.GameAccount{AssetSummary: datatypes.JSON(raw)}, ) assertInt64(t, "拆分合计总价", snapshot.OwnerTotalPriceCent, 25000) assertInt64(t, "纯比价格", snapshot.OwnerPurePriceCent, 20000) assertInt64(t, "额外物品价格", snapshot.OwnerExtraItemPriceCent, 5000) assertInt64(t, "网站加价兜底", snapshot.WebsiteProfitCent, 5000) } func TestRepositoryUpdateProfitAllowsPickingUp(t *testing.T) { repo, db := newPickupTestRepo(t) pickup := seedPickup(t, db, StatusPickingUp) item, err := repo.UpdateProfit(t.Context(), pickup.ID, UpdateProfitRequest{ ProfitAmountCent: 2500, Reason: "线下扣点调整", }, 99, auditlog.Meta{RequestID: "req-profit"}) if err != nil { t.Fatalf("UpdateProfit() error = %v", err) } assertInt64(t, "修改后利润", item.ProfitAmountCent, 2500) var stored model.AdminPickup if err := db.First(&stored, pickup.ID).Error; err != nil { t.Fatalf("查询提号失败: %v", err) } assertInt64(t, "数据库利润", stored.ProfitAmountCent, 2500) var audit model.AuditLog if err := db.Where("action = ?", "pickup_profit_update").First(&audit).Error; err != nil { t.Fatalf("查询审计日志失败: %v", err) } var detail map[string]any if err := json.Unmarshal(audit.Detail, &detail); err != nil { t.Fatalf("解析审计日志失败: %v", err) } assertJSONNumber(t, "旧利润", detail["old_profit_amount_cent"], 1000) assertJSONNumber(t, "新利润", detail["new_profit_amount_cent"], 2500) if detail["reason"] != "线下扣点调整" { t.Fatalf("reason = %v, want 线下扣点调整", detail["reason"]) } } func TestRepositoryUpdateProfitRejectsCompleted(t *testing.T) { repo, db := newPickupTestRepo(t) pickup := seedPickup(t, db, StatusCompleted) _, err := repo.UpdateProfit(t.Context(), pickup.ID, UpdateProfitRequest{ProfitAmountCent: 2500}, 99, auditlog.Meta{}) if !errors.Is(err, ErrPickupNotPickingUp) { t.Fatalf("UpdateProfit() error = %v, want ErrPickupNotPickingUp", err) } var stored model.AdminPickup if err := db.First(&stored, pickup.ID).Error; err != nil { t.Fatalf("查询提号失败: %v", err) } assertInt64(t, "已完成提号利润", stored.ProfitAmountCent, 1000) var auditCount int64 if err := db.Model(&model.AuditLog{}).Where("action = ?", "pickup_profit_update").Count(&auditCount).Error; err != nil { t.Fatalf("统计审计日志失败: %v", err) } if auditCount != 0 { t.Fatalf("审计日志数量 = %d, want 0", auditCount) } } func TestRepositoryListAdminFiltersShopName(t *testing.T) { repo, db := newPickupTestRepo(t) first := seedPickup(t, db, StatusPickingUp) if err := db.Model(&first).Updates(map[string]any{ "platform": "淘宝", "shop_name": "大锤一店", }).Error; err != nil { t.Fatalf("设置第一条提号店铺失败: %v", err) } second := first second.ID = 0 second.PickupNo = "PK-PROFIT-002" second.ShopName = "大锤二店" if err := db.Create(&second).Error; err != nil { t.Fatalf("创建第二条提号记录失败: %v", err) } result, err := repo.ListAdmin(t.Context(), AdminPickupQuery{ShopName: " 大锤一店 "}) if err != nil { t.Fatalf("ListAdmin() error = %v", err) } if result.Total != 1 || len(result.Items) != 1 { t.Fatalf("店铺筛选结果 = total %d, items %d, want 1", result.Total, len(result.Items)) } if result.Items[0].ShopName != "大锤一店" { t.Fatalf("店铺名称 = %q, want 大锤一店", result.Items[0].ShopName) } } func TestRepositoryListShopNamesReturnsDistinctPlatformOptions(t *testing.T) { repo, db := newPickupTestRepo(t) first := seedPickup(t, db, StatusPickingUp) if err := db.Model(&first).Updates(map[string]any{ "platform": "淘宝", "shop_name": "大锤一店", }).Error; err != nil { t.Fatalf("设置第一条提号店铺失败: %v", err) } for index, row := range []model.AdminPickup{ {PickupNo: "PK-SHOP-002", ListingID: first.ListingID, AccountID: first.AccountID, OwnerID: first.OwnerID, AdminID: 1, Platform: "淘宝", ShopName: "大锤一店", Status: StatusPickingUp}, {PickupNo: "PK-SHOP-003", ListingID: first.ListingID, AccountID: first.AccountID, OwnerID: first.OwnerID, AdminID: 1, Platform: "拼多多", ShopName: "大锤二店", Status: StatusPickingUp}, } { if err := db.Create(&row).Error; err != nil { t.Fatalf("创建第 %d 条店铺选项失败: %v", index+2, err) } } names, err := repo.ListShopNames(t.Context(), "淘宝") if err != nil { t.Fatalf("ListShopNames() error = %v", err) } if len(names) != 1 || names[0] != "大锤一店" { t.Fatalf("淘宝店铺选项 = %v, want [大锤一店]", names) } } func assertInt64(t *testing.T, name string, got, want int64) { t.Helper() if got != want { t.Fatalf("%s = %d, want %d", name, got, want) } } func assertJSONNumber(t *testing.T, name string, got any, want float64) { t.Helper() value, ok := got.(float64) if !ok || value != want { t.Fatalf("%s = %v, want %.0f", name, got, want) } } func newPickupTestRepo(t *testing.T) (*Repository, *gorm.DB) { t.Helper() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{ Logger: logger.Default.LogMode(logger.Silent), }) if err != nil { t.Fatalf("打开测试数据库失败: %v", err) } if err := db.AutoMigrate( &model.User{}, &model.GameAccount{}, &model.RentalListing{}, &model.AdminPickup{}, &model.AuditLog{}, ); err != nil { t.Fatalf("迁移测试表失败: %v", err) } return NewRepository(db), db } func seedPickup(t *testing.T, db *gorm.DB, status string) model.AdminPickup { t.Helper() owner := model.User{Phone: "18800000000"} if err := db.Create(&owner).Error; err != nil { t.Fatalf("创建号主失败: %v", err) } account := model.GameAccount{ OwnerID: owner.ID, ServerRegion: "测试区", LoginPlatform: "微信", Title: "测试账号", Status: "rented", } if err := db.Create(&account).Error; err != nil { t.Fatalf("创建账号失败: %v", err) } listing := model.RentalListing{ ListingNo: "LST-PROFIT-001", AccountID: account.ID, OwnerID: owner.ID, PriceCent: 10000, Status: "rented", ReviewStatus: "approved", } if err := db.Create(&listing).Error; err != nil { t.Fatalf("创建上架记录失败: %v", err) } pickup := model.AdminPickup{ PickupNo: "PK-PROFIT-001", ListingID: listing.ID, AccountID: account.ID, OwnerID: owner.ID, AdminID: 1, ProfitAmountCent: 1000, Status: status, } if err := db.Create(&pickup).Error; err != nil { t.Fatalf("创建提号记录失败: %v", err) } return pickup }