Files
hfb_sys/backend/internal/modules/pickup/repository_test.go
T

705 lines
27 KiB
Go

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 TestRepositoryListAdminFiltersOwnerPhone(t *testing.T) {
repo, db := newPickupTestRepo(t)
pickup := seedPickup(t, db, StatusPickingUp)
result, err := repo.ListAdmin(t.Context(), AdminPickupQuery{Keyword: "000000"})
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].ID != pickup.ID || result.Items[0].OwnerPhone != "18800000000" {
t.Fatalf(
"手机号搜索结果 = id %d, phone %q, want id %d, phone 18800000000",
result.Items[0].ID,
result.Items[0].OwnerPhone,
pickup.ID,
)
}
}
func TestRepositoryListAdminFiltersOfflineSettlementStatus(t *testing.T) {
repo, db := newPickupTestRepo(t)
first := seedPickup(t, db, StatusCompleted)
if err := db.Model(&first).Updates(map[string]any{
"settlement_mode": SettlementModePlatformManaged,
"offline_settlement_status": OfflineSettlementStatusPending,
}).Error; err != nil {
t.Fatalf("设置待线下结算提号失败: %v", err)
}
second := first
second.ID = 0
second.PickupNo = "PK-OFFLINE-SETTLED-001"
second.OfflineSettlementStatus = OfflineSettlementStatusSettled
if err := db.Create(&second).Error; err != nil {
t.Fatalf("创建已线下结算提号失败: %v", err)
}
result, err := repo.ListAdmin(t.Context(), AdminPickupQuery{OfflineSettlementStatus: OfflineSettlementStatusPending})
if err != nil {
t.Fatalf("ListAdmin() error = %v", err)
}
if result.Total != 1 || len(result.Items) != 1 || result.Items[0].OfflineSettlementStatus != OfflineSettlementStatusPending {
t.Fatalf("待线下结算筛选结果 = %+v", result)
}
}
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 TestRepositoryListAvailableListingsFiltersAccountSource(t *testing.T) {
repo, db := newPickupTestRepo(t)
internal := seedAvailableListing(t, db, "LST-SOURCE-INTERNAL", "站内账号", false)
external := seedAvailableListing(t, db, "LST-SOURCE-EXTERNAL", "外部账号", true)
if err := db.Create(&model.ListingUpload{
UploaderName: "外部号主",
SourceChannel: "淘宝",
ListingID: &external.ID,
Status: "listing_created",
}).Error; err != nil {
t.Fatalf("创建外部上传记录失败: %v", err)
}
all, err := repo.ListAvailableListings(t.Context(), AvailableListingQuery{Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("ListAvailableListings() error = %v", err)
}
if all.Total != 2 || len(all.Items) != 2 {
t.Fatalf("全部来源结果 = total %d, items %d, want 2", all.Total, len(all.Items))
}
externalResult, err := repo.ListAvailableListings(t.Context(), AvailableListingQuery{
SourceType: ListingSourceExternal,
Page: 1,
PageSize: 20,
})
if err != nil {
t.Fatalf("筛选外部上传 error = %v", err)
}
if externalResult.Total != 1 || len(externalResult.Items) != 1 {
t.Fatalf("外部上传结果 = total %d, items %d, want 1", externalResult.Total, len(externalResult.Items))
}
externalItem := externalResult.Items[0]
if externalItem.ID != external.ID || !externalItem.IsExternalUpload {
t.Fatalf("外部上传账号 = id %d, external %t, want id %d", externalItem.ID, externalItem.IsExternalUpload, external.ID)
}
if externalItem.AccountSource != AccountSourceExternalPlatformManaged || externalItem.SourceChannel != "淘宝" {
t.Fatalf("外部上传来源 = %q/%q", externalItem.AccountSource, externalItem.SourceChannel)
}
if externalItem.HandoffMode != "platform" || externalItem.SettlementMode != "platform_managed" {
t.Fatalf("外部上传代管模式 = %q/%q", externalItem.HandoffMode, externalItem.SettlementMode)
}
internalResult, err := repo.ListAvailableListings(t.Context(), AvailableListingQuery{
SourceType: ListingSourceInternal,
Page: 1,
PageSize: 20,
})
if err != nil {
t.Fatalf("筛选站内上传 error = %v", err)
}
if internalResult.Total != 1 || len(internalResult.Items) != 1 {
t.Fatalf("站内上传结果 = total %d, items %d, want 1", internalResult.Total, len(internalResult.Items))
}
internalItem := internalResult.Items[0]
if internalItem.ID != internal.ID || internalItem.IsExternalUpload || internalItem.AccountSource != AccountSourceInternal {
t.Fatalf("站内上传账号 = id %d, external %t, source %q", internalItem.ID, internalItem.IsExternalUpload, internalItem.AccountSource)
}
}
func TestRepositoryCreateSnapshotsExternalAccountSource(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-SOURCE-CREATE", "待提号外部账号", true)
upload := model.ListingUpload{
UploaderName: "外部号主",
SourceChannel: "闲鱼",
ListingID: &listing.ID,
Status: "listing_created",
}
if err := db.Create(&upload).Error; err != nil {
t.Fatalf("创建外部上传记录失败: %v", err)
}
item, err := repo.Create(t.Context(), CreateRequest{
ListingID: listing.ID,
Platform: "微信",
ProfitAmountCent: 1200,
}, 99, auditlog.Meta{RequestID: "req-source"})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if item.AccountSource != AccountSourceExternalPlatformManaged || item.SourceChannel != "闲鱼" {
t.Fatalf("创建来源快照 = %q/%q", item.AccountSource, item.SourceChannel)
}
if err := db.Delete(&upload).Error; err != nil {
t.Fatalf("删除上传关联失败: %v", err)
}
stored, err := repo.FindByID(t.Context(), item.ID)
if err != nil {
t.Fatalf("FindByID() error = %v", err)
}
if stored.AccountSource != AccountSourceExternalPlatformManaged || stored.SourceChannel != "闲鱼" {
t.Fatalf("关联删除后的来源快照 = %q/%q", stored.AccountSource, stored.SourceChannel)
}
}
func TestRepositoryCompleteInternalPickupCreditsWallet(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-COMPLETE-INTERNAL", "站内结算账号", false)
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
completed, err := repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 12000}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Complete() error = %v", err)
}
if completed.Status != StatusCompleted || completed.SettlementMode != SettlementModeOwnerWallet {
t.Fatalf("站内提号完成状态 = %q/%q", completed.Status, completed.SettlementMode)
}
if completed.OfflineSettlementStatus != OfflineSettlementStatusNone {
t.Fatalf("站内提号线下结算状态 = %q, want none", completed.OfflineSettlementStatus)
}
var ledger model.WalletLedger
if err := db.Where("biz_type = ? AND biz_no = ?", "admin_pickup_settle", pickup.PickupNo).First(&ledger).Error; err != nil {
t.Fatalf("站内提号应生成钱包流水: %v", err)
}
assertInt64(t, "站内提号钱包入账", ledger.AmountCent, 12000)
}
func TestRepositoryCompletePlatformManagedPickupUsesOfflineSettlement(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-COMPLETE-EXTERNAL", "代管结算账号", true)
if err := db.Create(&model.ListingUpload{
UploaderName: "外部号主",
SourceChannel: "闲鱼",
ListingID: &listing.ID,
Status: "listing_created",
}).Error; err != nil {
t.Fatalf("创建外部上传记录失败: %v", err)
}
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if pickup.SettlementMode != SettlementModePlatformManaged {
t.Fatalf("平台代管提号结算模式 = %q", pickup.SettlementMode)
}
completed, err := repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 13500}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Complete() error = %v", err)
}
if completed.OfflineSettlementStatus != OfflineSettlementStatusPending {
t.Fatalf("平台代管提号线下结算状态 = %q, want pending", completed.OfflineSettlementStatus)
}
if completed.OfflineSettledAt != nil || completed.OfflineSettledBy != nil {
t.Fatalf("待线下结算不应记录确认信息: %+v", completed)
}
var ledgerCount int64
if err := db.Model(&model.WalletLedger{}).
Where("biz_type = ? AND biz_no = ?", "admin_pickup_settle", pickup.PickupNo).
Count(&ledgerCount).Error; err != nil {
t.Fatalf("统计钱包流水失败: %v", err)
}
if ledgerCount != 0 {
t.Fatalf("平台代管提号不应生成钱包流水, got %d", ledgerCount)
}
settled, err := repo.MarkOfflineSettlement(t.Context(), pickup.ID, OfflineSettlementRequest{Remark: "已转账"}, 100, auditlog.Meta{})
if err != nil {
t.Fatalf("MarkOfflineSettlement() error = %v", err)
}
if settled.OfflineSettlementStatus != OfflineSettlementStatusSettled || settled.OfflineSettledAt == nil || settled.OfflineSettledBy == nil || *settled.OfflineSettledBy != 100 {
t.Fatalf("平台代管线下结算确认结果 = %+v", settled)
}
if settled.OfflineSettlementRemark != "已转账" {
t.Fatalf("线下结算备注 = %q", settled.OfflineSettlementRemark)
}
var audit model.AuditLog
if err := db.Where("action = ? AND biz_id = ?", "pickup_offline_settlement", pickup.ID).First(&audit).Error; err != nil {
t.Fatalf("确认线下结算应写审计日志: %v", err)
}
}
func TestRepositoryFinancialAdjustmentCreditsInternalPickupWallet(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-ADJUST-INTERNAL", "站内补款账号", false)
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID, ProfitAmountCent: 1000}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if _, err = repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 12000}, 99, auditlog.Meta{}); err != nil {
t.Fatalf("Complete() error = %v", err)
}
updated, err := repo.CreateFinancialAdjustment(t.Context(), pickup.ID, FinancialAdjustmentRequest{
ProfitAmountCent: ptrInt64(1500),
SettleAmountCent: ptrInt64(13500),
Reason: "补录实际成交金额",
}, 100, auditlog.Meta{})
if err != nil {
t.Fatalf("CreateFinancialAdjustment() error = %v", err)
}
assertInt64(t, "原始利润不变", updated.ProfitAmountCent, 1000)
assertInt64(t, "有效利润", updated.EffectiveProfitAmountCent, 1500)
assertInt64(t, "原始结算额不变", updated.SettleAmountCent, 12000)
assertInt64(t, "有效结算额", updated.EffectiveSettleAmountCent, 13500)
var adjustment model.AdminPickupFinancialAdjustment
if err := db.Where("pickup_id = ?", pickup.ID).First(&adjustment).Error; err != nil {
t.Fatalf("查询调整流水失败: %v", err)
}
if adjustment.Status != FinancialAdjustmentStatusSettled || adjustment.ProfitDeltaCent != 500 || adjustment.SettleDeltaCent != 1500 {
t.Fatalf("调整流水 = %+v", adjustment)
}
var ledger model.WalletLedger
if err := db.Where("biz_type = ?", "admin_pickup_adjustment").First(&ledger).Error; err != nil {
t.Fatalf("站内补款应生成钱包流水: %v", err)
}
assertInt64(t, "站内补款钱包入账", ledger.AmountCent, 1500)
}
func TestRepositoryFinancialAdjustmentUsesRecoveryForInternalReduction(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-ADJUST-RECOVERY", "站内追回账号", false)
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if _, err = repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 12000}, 99, auditlog.Meta{}); err != nil {
t.Fatalf("Complete() error = %v", err)
}
updated, err := repo.CreateFinancialAdjustment(t.Context(), pickup.ID, FinancialAdjustmentRequest{
SettleAmountCent: ptrInt64(10000),
Reason: "实际应结算金额减少",
}, 100, auditlog.Meta{})
if err != nil {
t.Fatalf("CreateFinancialAdjustment() error = %v", err)
}
assertInt64(t, "追回后的有效结算额", updated.EffectiveSettleAmountCent, 10000)
var adjustment model.AdminPickupFinancialAdjustment
if err := db.Where("pickup_id = ?", pickup.ID).First(&adjustment).Error; err != nil {
t.Fatalf("查询追回调整失败: %v", err)
}
if adjustment.Status != FinancialAdjustmentStatusRecovery || adjustment.SettleDeltaCent != -2000 {
t.Fatalf("追回调整 = %+v", adjustment)
}
var audit model.AuditLog
if err := db.Where("action = ?", "pickup_financial_adjustment_create").First(&audit).Error; err != nil {
t.Fatalf("查询财务调整审计日志失败: %v", err)
}
if audit.BizType != "admin_pickup" {
t.Fatalf("财务调整审计业务类型 = %q, want admin_pickup", audit.BizType)
}
var ledgerCount int64
if err := db.Model(&model.WalletLedger{}).Where("biz_type = ?", "admin_pickup_adjustment").Count(&ledgerCount).Error; err != nil {
t.Fatalf("统计补款流水失败: %v", err)
}
if ledgerCount != 0 {
t.Fatalf("待追回不能直接扣钱包, got %d 条流水", ledgerCount)
}
}
func TestRepositoryFinancialAdjustmentSettlesPlatformManagedPickupSeparately(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-ADJUST-PLATFORM", "代管调整账号", true)
if err := db.Create(&model.ListingUpload{UploaderName: "外部号主", ListingID: &listing.ID, Status: "listing_created"}).Error; err != nil {
t.Fatalf("创建外部上传记录失败: %v", err)
}
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if _, err = repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 12000}, 99, auditlog.Meta{}); err != nil {
t.Fatalf("Complete() error = %v", err)
}
if _, err = repo.MarkOfflineSettlement(t.Context(), pickup.ID, OfflineSettlementRequest{}, 100, auditlog.Meta{}); err != nil {
t.Fatalf("MarkOfflineSettlement() error = %v", err)
}
updated, err := repo.CreateFinancialAdjustment(t.Context(), pickup.ID, FinancialAdjustmentRequest{
ProfitAmountCent: ptrInt64(2000),
SettleAmountCent: ptrInt64(13500),
Reason: "补录代管打款",
}, 101, auditlog.Meta{})
if err != nil {
t.Fatalf("CreateFinancialAdjustment() error = %v", err)
}
if updated.OfflineSettlementStatus != OfflineSettlementStatusSettled {
t.Fatalf("原线下结算状态不应被重置: %q", updated.OfflineSettlementStatus)
}
adjustments, err := repo.ListFinancialAdjustments(t.Context(), pickup.ID)
if err != nil || len(adjustments) != 1 {
t.Fatalf("ListFinancialAdjustments() = %+v, %v", adjustments, err)
}
if adjustments[0].Status != FinancialAdjustmentStatusPending {
t.Fatalf("代管补款状态 = %q, want pending", adjustments[0].Status)
}
var createAudit model.AuditLog
if err := db.Where("action = ?", "pickup_financial_adjustment_create").First(&createAudit).Error; err != nil {
t.Fatalf("查询创建财务调整审计日志失败: %v", err)
}
if createAudit.BizType != "admin_pickup" {
t.Fatalf("创建财务调整审计业务类型 = %q, want admin_pickup", createAudit.BizType)
}
if err := repo.SettleFinancialAdjustment(t.Context(), adjustments[0].ID, FinancialAdjustmentSettlementRequest{Remark: "已补款"}, 102, auditlog.Meta{}); err != nil {
t.Fatalf("SettleFinancialAdjustment() error = %v", err)
}
adjustments, err = repo.ListFinancialAdjustments(t.Context(), pickup.ID)
if err != nil || adjustments[0].Status != FinancialAdjustmentStatusSettled || adjustments[0].SettledBy == nil || *adjustments[0].SettledBy != 102 {
t.Fatalf("确认后的调整流水 = %+v, %v", adjustments, err)
}
var settleAudit model.AuditLog
if err := db.Where("action = ?", "pickup_financial_adjustment_settle").First(&settleAudit).Error; err != nil {
t.Fatalf("查询确认财务调整审计日志失败: %v", err)
}
if settleAudit.BizType != "admin_pickup" {
t.Fatalf("确认财务调整审计业务类型 = %q, want admin_pickup", settleAudit.BizType)
}
}
func TestRepositoryOfflineSettlementProcessesPendingPickupAdjustments(t *testing.T) {
repo, db := newPickupTestRepo(t)
listing := seedAvailableListing(t, db, "LST-ADJUST-WITH-SETTLEMENT", "代管合并结算账号", true)
if err := db.Create(&model.ListingUpload{UploaderName: "外部号主", ListingID: &listing.ID, Status: "listing_created"}).Error; err != nil {
t.Fatalf("创建外部上传记录失败: %v", err)
}
pickup, err := repo.Create(t.Context(), CreateRequest{ListingID: listing.ID}, 99, auditlog.Meta{})
if err != nil {
t.Fatalf("Create() error = %v", err)
}
if _, err = repo.Complete(t.Context(), pickup.ID, CompleteRequest{SettleAmountCent: 12000}, 99, auditlog.Meta{}); err != nil {
t.Fatalf("Complete() error = %v", err)
}
if _, err = repo.CreateFinancialAdjustment(t.Context(), pickup.ID, FinancialAdjustmentRequest{
SettleAmountCent: ptrInt64(10000),
Reason: "付款前修正应付金额",
}, 100, auditlog.Meta{}); err != nil {
t.Fatalf("CreateFinancialAdjustment() error = %v", err)
}
adjustments, err := repo.ListFinancialAdjustments(t.Context(), pickup.ID)
if err != nil || len(adjustments) != 1 || adjustments[0].Status != FinancialAdjustmentStatusPending {
t.Fatalf("结算前调整流水 = %+v, %v", adjustments, err)
}
settled, err := repo.MarkOfflineSettlement(t.Context(), pickup.ID, OfflineSettlementRequest{Remark: "按修正后金额打款"}, 101, auditlog.Meta{})
if err != nil {
t.Fatalf("MarkOfflineSettlement() error = %v", err)
}
if settled.OfflineSettlementStatus != OfflineSettlementStatusSettled || settled.EffectiveSettleAmountCent != 10000 {
t.Fatalf("合并线下结算结果 = %+v", settled)
}
adjustments, err = repo.ListFinancialAdjustments(t.Context(), pickup.ID)
if err != nil || adjustments[0].Status != FinancialAdjustmentStatusSettled || adjustments[0].SettledBy == nil || *adjustments[0].SettledBy != 101 {
t.Fatalf("合并结算后的调整流水 = %+v, %v", adjustments, err)
}
}
func ptrInt64(value int64) *int64 {
return &value
}
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.AdminUser{},
&model.GameAccount{},
&model.RentalListing{},
&model.ListingUpload{},
&model.AdminPickup{},
&model.AdminPickupFinancialAdjustment{},
&model.WalletAccount{},
&model.WalletLedger{},
&model.AuditLog{},
&model.ProcessEvent{},
&model.Notification{},
&model.ListingStatusEvent{},
); err != nil {
t.Fatalf("迁移测试表失败: %v", err)
}
return NewRepository(db), db
}
func seedAvailableListing(t *testing.T, db *gorm.DB, listingNo, title string, external bool) model.RentalListing {
t.Helper()
owner := model.User{Phone: "188" + listingNo}
if err := db.Create(&owner).Error; err != nil {
t.Fatalf("创建候选号主失败: %v", err)
}
account := model.GameAccount{
OwnerID: owner.ID,
ServerRegion: "测试区",
LoginPlatform: "微信",
Title: title,
Status: "published",
}
if err := db.Create(&account).Error; err != nil {
t.Fatalf("创建候选账号失败: %v", err)
}
listing := model.RentalListing{
ListingNo: listingNo,
AccountID: account.ID,
OwnerID: owner.ID,
PriceCent: 10000,
Status: "published",
ReviewStatus: "approved",
HandoffMode: "owner",
SettlementMode: "owner_wallet",
}
if external {
listing.HandoffMode = "platform"
listing.SettlementMode = "platform_managed"
}
if err := db.Create(&listing).Error; err != nil {
t.Fatalf("创建候选上架记录失败: %v", err)
}
return listing
}
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
}