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