hfb_sys/backend/internal/modules/pickup/repository_test.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 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
}