feat(finance): add daily paid order count

2026-08-10 13:42:40 +08:00
parent 89bf5f30ba
commit 6ddd7b664c
6 changed files with 109 additions and 1 deletions
@@ -0,0 +1,65 @@
+package adminfinance
+
+import (
+	"testing"
+	"time"
+
+	"hfb_sys/backend/internal/timeutil"
+
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+func TestDailyPaidOrdersUsePaidAtAndExcludeMohong(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 payment_orders (
+		id INTEGER PRIMARY KEY,
+		order_id INTEGER NOT NULL,
+		biz_type TEXT NOT NULL,
+		status TEXT NOT NULL,
+		paid_at DATETIME NULL
+	)`).Error; err != nil {
+		t.Fatalf("创建支付单表失败: %v", err)
+	}
+
+	loc := timeutil.ShanghaiLocation()
+	day := func(value int) time.Time { return time.Date(2026, 8, value, 10, 0, 0, 0, loc) }
+	rows := [][]any{
+		{1, 101, "order_pay", "paid", day(2)},
+		{2, 101, "order_pay", "paid", day(2)},
+		{3, 102, "order_pay", "paid", day(3)},
+		{4, 201, "mohong_pay", "paid", day(2)},
+		{5, 103, "order_pay", "created", nil},
+		{6, 104, "order_pay", "paid", day(1)},
+	}
+	for _, row := range rows {
+		if err := db.Exec(`INSERT INTO payment_orders (id, order_id, biz_type, status, paid_at)
+			VALUES (?, ?, ?, ?, ?)`, row...).Error; err != nil {
+			t.Fatalf("写入支付单失败: %v", err)
+		}
+	}
+
+	repo := NewRepository(db)
+	query := DashboardQuery{
+		StartDate: time.Date(2026, 8, 2, 0, 0, 0, 0, loc),
+		EndDate:   time.Date(2026, 8, 3, 23, 59, 59, 0, loc),
+	}
+	result, err := repo.dailyPaidOrders(t.Context(), query)
+	if err != nil {
+		t.Fatalf("读取每日已付款订单失败: %v", err)
+	}
+	if len(result) != 2 {
+		t.Fatalf("每日已付款记录数 = %d, want 2", len(result))
+	}
+	counts := make(map[string]int64, len(result))
+	for _, row := range result {
+		counts[row.Date] = row.PaidOrderCount
+	}
+	if counts["2026-08-02"] != 1 || counts["2026-08-03"] != 1 {
+		t.Fatalf("每日已付款订单 = %#v, want 8月2日和8月3日均为1", counts)
+	}
+}