110 lines
4.1 KiB
Go
110 lines
4.1 KiB
Go
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])
|
|
}
|
|
}
|