package adminfinance

import (
	"testing"
	"time"

	"hfb_sys/backend/internal/timeutil"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

func TestDisbursementStatisticsUseActualPaymentTimeAndAmount(t *testing.T) {
	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.Exec(`CREATE TABLE rental_orders (
		id INTEGER PRIMARY KEY,
		settlement_mode TEXT NOT NULL,
		offline_settlement_status TEXT NOT NULL,
		offline_settlement_amount_cent INTEGER NOT NULL,
		offline_settled_at DATETIME NULL
	)`).Error; err != nil {
		t.Fatalf("创建订单表失败: %v", err)
	}
	if err := db.Exec(`CREATE TABLE withdrawal_requests (
		id INTEGER PRIMARY KEY,
		status TEXT NOT NULL,
		amount_cent INTEGER NOT NULL,
		fee_cent INTEGER NOT NULL,
		actual_amount_cent INTEGER NOT NULL,
		paid_at DATETIME NULL
	)`).Error; err != nil {
		t.Fatalf("创建提现表失败: %v", err)
	}

	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)
	outOfRange := time.Date(2026, 6, 30, 23, 59, 0, 0, loc)

	for _, args := range [][]any{
		{1, "platform_managed", "settled", 12000, inRangeOffline},
		{2, "platform_managed", "settled", 3000, outOfRange},
		{3, "platform_managed", "pending", 8000, nil},
		{4, "platform_managed", "pending", 0, nil},
		{5, "owner_wallet", "pending", 9000, nil},
	} {
		if err := db.Exec(`INSERT INTO rental_orders
			(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, "completed", 10000, 100, 9900, inRangeWithdrawal},
		{2, "completed", 5000, 0, 5000, outOfRange},
		{3, "processing", 5000, 200, 4800, nil},
		{4, "pending", 2100, 100, 2000, nil},
		{5, "rejected", 6000, 100, 5900, nil},
	} {
		if err := db.Exec(`INSERT INTO withdrawal_requests
			(id, 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 != 21900 || summary.PaidCount != 2 {
		t.Fatalf("实际出款 = %d/%d, want 21900/2", summary.PaidAmountCent, summary.PaidCount)
	}
	if summary.OfflineSettlementPaidAmountCent != 12000 || summary.OfflineSettlementPaidCount != 1 {
		t.Fatalf("代管打款 = %d/%d, want 12000/1", 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.PendingPaymentAmountCent != 12800 || summary.PendingPaymentCount != 2 {
		t.Fatalf("待出款 = %d/%d, want 12800/2", summary.PendingPaymentAmountCent, summary.PendingPaymentCount)
	}
	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) != 2 {
		t.Fatalf("每日出款行数 = %d, want 2", len(daily))
	}
	if daily[0].Date != "2026-07-01" || daily[0].OfflineSettlementPaidAmountCent != 12000 {
		t.Fatalf("第一日代管出款 = %+v", daily[0])
	}
	if daily[1].Date != "2026-07-02" || daily[1].WithdrawalPaidAmountCent != 9900 {
		t.Fatalf("第二日提现出款 = %+v", daily[1])
	}
}