Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/payment_stats_test.go

@@ -0,0 +1,84 @@
+//go:build unit
+
+package service
+
+import (
+	"testing"
+	"time"
+
+	dbent "github.com/Wei-Shaw/sub2api/ent"
+	"github.com/stretchr/testify/require"
+)
+
+func TestComputeBasicStatsGroupsAmountsByCurrency(t *testing.T) {
+	t.Parallel()
+
+	todayStart := time.Date(2026, time.July, 25, 0, 0, 0, 0, time.UTC)
+	yesterday := todayStart.Add(-time.Hour)
+	today := todayStart.Add(time.Hour)
+	orders := []*dbent.PaymentOrder{
+		paymentStatsTestOrder(1, "alice@example.com", "CNY", 10, &today),
+		paymentStatsTestOrder(2, "bob@example.com", "USD", 10, &today),
+		paymentStatsTestOrder(1, "alice@example.com", "CNY", 5, &yesterday),
+	}
+
+	stats := &DashboardStats{}
+	computeBasicStats(stats, orders, todayStart)
+
+	require.Equal(t, CurrencyAmounts{"CNY": 15, "USD": 10}, stats.TotalAmount)
+	require.Equal(t, CurrencyAmounts{"CNY": 10, "USD": 10}, stats.TodayAmount)
+	require.Equal(t, CurrencyAmounts{"CNY": 7.5, "USD": 10}, stats.AvgAmount)
+	require.Equal(t, 3, stats.TotalCount)
+	require.Equal(t, 2, stats.TodayCount)
+}
+
+func TestPaymentDashboardBreakdownsGroupAmountsAndRankingsByCurrency(t *testing.T) {
+	t.Parallel()
+
+	firstDay := time.Date(2026, time.July, 24, 12, 0, 0, 0, time.UTC)
+	secondDay := firstDay.AddDate(0, 0, 1)
+	orders := []*dbent.PaymentOrder{
+		paymentStatsTestOrder(1, "alice@example.com", "CNY", 5.555, &firstDay),
+		paymentStatsTestOrder(2, "bob@example.com", "CNY", 10, &firstDay),
+		paymentStatsTestOrder(1, "alice@example.com", "USD", 20, &secondDay),
+		paymentStatsTestOrder(2, "bob@example.com", "USD", 10, &secondDay),
+	}
+	orders[0].PaymentType = "stripe"
+	orders[1].PaymentType = "stripe"
+	orders[2].PaymentType = "stripe"
+	orders[3].PaymentType = "alipay"
+
+	daily := buildDailySeries(orders, firstDay.AddDate(0, 0, -1), 2)
+	require.Equal(t, []DailyStats{
+		{Date: "2026-07-24", Amount: CurrencyAmounts{"CNY": 15.56}, Count: 2},
+		{Date: "2026-07-25", Amount: CurrencyAmounts{"USD": 30}, Count: 2},
+	}, daily)
+
+	methods := buildMethodDistribution(orders)
+	require.Equal(t, []PaymentMethodStat{
+		{Type: "alipay", Amount: CurrencyAmounts{"USD": 10}, Count: 1},
+		{Type: "stripe", Amount: CurrencyAmounts{"CNY": 15.56, "USD": 20}, Count: 3},
+	}, methods)
+
+	users := buildTopUsers(orders)
+	require.Equal(t, TopUsersByCurrency{
+		"CNY": {
+			{UserID: 2, Email: "bob@example.com", Amount: 10},
+			{UserID: 1, Email: "alice@example.com", Amount: 5.56},
+		},
+		"USD": {
+			{UserID: 1, Email: "alice@example.com", Amount: 20},
+			{UserID: 2, Email: "bob@example.com", Amount: 10},
+		},
+	}, users)
+}
+
+func paymentStatsTestOrder(userID int64, email, currency string, amount float64, paidAt *time.Time) *dbent.PaymentOrder {
+	return &dbent.PaymentOrder{
+		UserID:           userID,
+		UserEmail:        email,
+		PayAmount:        amount,
+		PaidAt:           paidAt,
+		ProviderSnapshot: map[string]any{"currency": currency},
+	}
+}