hfb_sys/backend/internal/modules/adminfinance/disbursement_test.go

package adminfinance

import (
	"testing"
	"time"

	"hfb_sys/backend/internal/database"
	"hfb_sys/backend/internal/timeutil"
)

func TestDisbursementStatisticsUseActualPaymentTimeAndAmount(t *testing.T) {
	db := database.NewTestDB()

	loc := timeutil.ShanghaiLocation()
	inRangeOffline := time.Date(2026, 7, 1, 10, 0, 0, 0, loc)
	inRangeWithdrawal := time.Date(2026, 7, 2, 11, 0, 0, 0, loc)
	inRangeManual := time.Date(2026, 7, 3, 12, 0, 0, 0, loc)
	outOfRange := time.Date(2026, 6, 30, 23, 59, 0, 0, loc)

	for _, args := range [][]any{
		{1, "ORD-1", 1, 1, 1, 1, "platform_managed", "settled", 12000, inRangeOffline},
		{2, "ORD-2", 2, 2, 2, 2, "platform_managed", "settled", 3000, outOfRange},
		{3, "ORD-3", 3, 3, 3, 3, "platform_managed", "pending", 8000, nil},
		{4, "ORD-4", 4, 4, 4, 4, "platform_managed", "pending", 0, nil},
		{5, "ORD-5", 5, 5, 5, 5, "owner_wallet", "pending", 9000, nil},
	} {
		if err := db.Exec(`INSERT INTO rental_orders
			(id, order_no, listing_id, account_id, owner_id, renter_id,
			 settlement_mode, offline_settlement_status, offline_settlement_amount_cent, offline_settled_at)
			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, args...).Error; err != nil {
			t.Fatalf("写入订单失败: %v", err)
		}
	}
	for _, args := range [][]any{
		{1, "PKR-1", 1, 1, 1, 1, "platform_managed", "settled", 4000, inRangeOffline, inRangeOffline},
		{2, "PKR-2", 2, 2, 2, 2, "platform_managed", "pending", 2500, nil, inRangeOffline},
	} {
		if err := db.Exec(`INSERT INTO admin_pickups
			(id, pickup_no, listing_id, account_id, owner_id, admin_id, settlement_mode,
			 offline_settlement_status, settle_amount_cent, offline_settled_at, created_at)
			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, args...).Error; err != nil {
			t.Fatalf("写入提号失败: %v", err)
		}
	}
	for _, args := range [][]any{
		{1, 1, "platform_managed", "settled", "补差价", 1, 500, inRangeManual, inRangeManual},
		{2, 2, "platform_managed", "pending", "补差价", 1, 300, nil, inRangeManual},
		{3, 3, "platform_managed", "recovery_pending", "补差价", 1, -200, nil, inRangeManual},
	} {
		if err := db.Exec(`INSERT INTO admin_pickup_financial_adjustments
			(id, pickup_id, settlement_mode, status, reason, created_by, settle_delta_cent, settled_at, created_at)
			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, args...).Error; err != nil {
			t.Fatalf("写入提号调整失败: %v", err)
		}
	}
	for _, args := range [][]any{
		{1, "MD-1", "推广投放", "测试收款方", "测试备注", 1, "paid", 2500, inRangeManual},
		{2, "MD-2", "推广投放", "测试收款方", "测试备注", 1, "paid", 1000, outOfRange},
		{3, "MD-3", "推广投放", "测试收款方", "测试备注", 1, "voided", 6000, inRangeManual},
	} {
		if err := db.Exec(`INSERT INTO manual_disbursements
			(id, disbursement_no, category, payee_name, remark, created_by, status, amount_cent, paid_at)
			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, args...).Error; err != nil {
			t.Fatalf("写入其他线下出款失败: %v", err)
		}
	}
	for _, args := range [][]any{
		{1, "WD-1", 1, "alipay", "测试户名", "test@example.com", "completed", 10000, 100, 9900, inRangeWithdrawal},
		{2, "WD-2", 1, "alipay", "测试户名", "test@example.com", "completed", 5000, 0, 5000, outOfRange},
		{3, "WD-3", 1, "alipay", "测试户名", "test@example.com", "processing", 5000, 200, 4800, nil},
		{4, "WD-4", 1, "alipay", "测试户名", "test@example.com", "pending", 2100, 100, 2000, nil},
		{5, "WD-5", 1, "alipay", "测试户名", "test@example.com", "rejected", 6000, 100, 5900, nil},
	} {
		if err := db.Exec(`INSERT INTO withdrawal_requests
			(id, withdraw_no, user_id, account_type, account_name, account_no,
			 status, amount_cent, fee_cent, actual_amount_cent, paid_at)
			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, args...).Error; err != nil {
			t.Fatalf("写入提现单失败: %v", err)
		}
	}

	query := DashboardQuery{
		StartDate: time.Date(2026, 7, 1, 0, 0, 0, 0, loc),
		EndDate:   time.Date(2026, 7, 3, 23, 59, 59, 0, loc),
	}
	repo := NewRepository(db)
	summary, err := repo.disbursementSummary(t.Context(), query)
	if err != nil {
		t.Fatalf("disbursementSummary() error = %v", err)
	}
	if summary.PaidAmountCent != 28900 || summary.PaidCount != 5 {
		t.Fatalf("实际出款 = %d/%d, want 28900/5", summary.PaidAmountCent, summary.PaidCount)
	}
	if summary.OfflineSettlementPaidAmountCent != 16500 || summary.OfflineSettlementPaidCount != 3 {
		t.Fatalf("代管打款 = %d/%d, want 16500/3", summary.OfflineSettlementPaidAmountCent, summary.OfflineSettlementPaidCount)
	}
	if summary.WithdrawalAmountCent != 10000 || summary.WithdrawalFeeAmountCent != 100 || summary.WithdrawalPaidAmountCent != 9900 {
		t.Fatalf("提现申请/手续费/实付 = %d/%d/%d, want 10000/100/9900", summary.WithdrawalAmountCent, summary.WithdrawalFeeAmountCent, summary.WithdrawalPaidAmountCent)
	}
	if summary.ManualPaidAmountCent != 2500 || summary.ManualPaidCount != 1 {
		t.Fatalf("其他线下出款 = %d/%d, want 2500/1", summary.ManualPaidAmountCent, summary.ManualPaidCount)
	}
	if summary.PendingPaymentAmountCent != 15600 || summary.PendingPaymentCount != 4 {
		t.Fatalf("待出款 = %d/%d, want 15600/4", summary.PendingPaymentAmountCent, summary.PendingPaymentCount)
	}
	if summary.OfflineSettlementPendingAmountCent != 10800 || summary.OfflineSettlementPendingCount != 3 {
		t.Fatalf("线下结算待出款 = %d/%d, want 10800/3", summary.OfflineSettlementPendingAmountCent, summary.OfflineSettlementPendingCount)
	}
	if summary.WithdrawalReviewAmountCent != 2000 || summary.WithdrawalReviewCount != 1 {
		t.Fatalf("待审核提现 = %d/%d, want 2000/1", summary.WithdrawalReviewAmountCent, summary.WithdrawalReviewCount)
	}

	daily, err := repo.dailyDisbursements(t.Context(), query)
	if err != nil {
		t.Fatalf("dailyDisbursements() error = %v", err)
	}
	if len(daily) != 3 {
		t.Fatalf("每日出款行数 = %d, want 3", len(daily))
	}
	if daily[0].Date != "2026-07-01" || daily[0].OfflineSettlementPaidAmountCent != 16000 {
		t.Fatalf("第一日代管出款 = %+v", daily[0])
	}
	if daily[1].Date != "2026-07-02" || daily[1].WithdrawalPaidAmountCent != 9900 {
		t.Fatalf("第二日提现出款 = %+v", daily[1])
	}
	if daily[2].Date != "2026-07-03" || daily[2].ManualPaidAmountCent != 2500 || daily[2].OfflineSettlementPaidAmountCent != 500 {
		t.Fatalf("第三日其他线下出款 = %+v", daily[2])
	}
}