test: 为代码审核修复添加详细单元测试（7个测试文件，50+测试用例）

新增测试文件: - cors_test.go: CORS 条件化头部测试（12个测试，覆盖白名单/黑名单/通配符/凭证/多源/Vary） - gateway_helper_backoff_test.go: nextBackoff 退避测试（6个测试+基准，验证指数增长/边界/抖动/收敛） - billing_cache_jitter_test.go: jitteredTTL 抖动测试（5个测试+基准，验证范围/上界/方差/均值） - subscription_calculate_progress_test.go: calculateProgress 纯函数测试（9个测试，覆盖日/周/月限额/超限截断/过期） - openai_gateway_handler_test.go: SSE JSON 转义测试（7个子用例，验证双引号/反斜杠/换行符安全）更新测试文件: - response_transformer_test.go: 增强 generateRandomID 测试（7个测试，含并发/字符集/降级计数器） - security_headers_test.go: 适配 GenerateNonce 新签名 - api_key_auth_test.go: 适配 NewSubscriptionService 新参数 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-07 22:14:07 +08:00
parent 9634494ba9
commit 00caf0bcd8
8 changed files with 913 additions and 6 deletions
--- a/backend/internal/service/subscription_calculate_progress_test.go
+++ b/backend/internal/service/subscription_calculate_progress_test.go
@@ -0,0 +1,231 @@
+package service
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// --- Task 5: 验证 calculateProgress 纯函数行为正确 ---
+
+func newTestSubscriptionService() *SubscriptionService {
+	return &SubscriptionService{}
+}
+
+func ptrFloat64(v float64) *float64 { return &v }
+func ptrTime(t time.Time) *time.Time { return &t }
+
+func TestCalculateProgress_BasicFields(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+
+	sub := &UserSubscription{
+		ID:        100,
+		ExpiresAt: now.Add(30 * 24 * time.Hour),
+	}
+	group := &Group{
+		Name: "Premium",
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	assert.Equal(t, int64(100), progress.ID)
+	assert.Equal(t, "Premium", progress.GroupName)
+	assert.Equal(t, sub.ExpiresAt, progress.ExpiresAt)
+	assert.Equal(t, 29, progress.ExpiresInDays) // 约 30 天
+	assert.Nil(t, progress.Daily, "无日限额时 Daily 应为 nil")
+	assert.Nil(t, progress.Weekly, "无周限额时 Weekly 应为 nil")
+	assert.Nil(t, progress.Monthly, "无月限额时 Monthly 应为 nil")
+}
+
+func TestCalculateProgress_DailyUsage(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+	dailyStart := now.Add(-12 * time.Hour)
+
+	sub := &UserSubscription{
+		ID:               1,
+		ExpiresAt:        now.Add(10 * 24 * time.Hour),
+		DailyUsageUSD:    3.0,
+		DailyWindowStart: ptrTime(dailyStart),
+	}
+	group := &Group{
+		Name:          "Pro",
+		DailyLimitUSD: ptrFloat64(10.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Daily, "有日限额和窗口时 Daily 不应为 nil")
+	assert.Equal(t, 10.0, progress.Daily.LimitUSD)
+	assert.Equal(t, 3.0, progress.Daily.UsedUSD)
+	assert.Equal(t, 7.0, progress.Daily.RemainingUSD)
+	assert.Equal(t, 30.0, progress.Daily.Percentage)
+	assert.Equal(t, dailyStart, progress.Daily.WindowStart)
+}
+
+func TestCalculateProgress_WeeklyUsage(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+	weeklyStart := now.Add(-3 * 24 * time.Hour)
+
+	sub := &UserSubscription{
+		ID:                1,
+		ExpiresAt:         now.Add(10 * 24 * time.Hour),
+		WeeklyUsageUSD:    25.0,
+		WeeklyWindowStart: ptrTime(weeklyStart),
+	}
+	group := &Group{
+		Name:           "Pro",
+		WeeklyLimitUSD: ptrFloat64(50.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Weekly, "有周限额和窗口时 Weekly 不应为 nil")
+	assert.Equal(t, 50.0, progress.Weekly.LimitUSD)
+	assert.Equal(t, 25.0, progress.Weekly.UsedUSD)
+	assert.Equal(t, 25.0, progress.Weekly.RemainingUSD)
+	assert.Equal(t, 50.0, progress.Weekly.Percentage)
+}
+
+func TestCalculateProgress_MonthlyUsage(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+	monthlyStart := now.Add(-15 * 24 * time.Hour)
+
+	sub := &UserSubscription{
+		ID:                 1,
+		ExpiresAt:          now.Add(10 * 24 * time.Hour),
+		MonthlyUsageUSD:    80.0,
+		MonthlyWindowStart: ptrTime(monthlyStart),
+	}
+	group := &Group{
+		Name:            "Enterprise",
+		MonthlyLimitUSD: ptrFloat64(100.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Monthly, "有月限额和窗口时 Monthly 不应为 nil")
+	assert.Equal(t, 100.0, progress.Monthly.LimitUSD)
+	assert.Equal(t, 80.0, progress.Monthly.UsedUSD)
+	assert.Equal(t, 20.0, progress.Monthly.RemainingUSD)
+	assert.Equal(t, 80.0, progress.Monthly.Percentage)
+}
+
+func TestCalculateProgress_OverLimit_ClampedTo100Percent(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+
+	sub := &UserSubscription{
+		ID:               1,
+		ExpiresAt:        now.Add(10 * 24 * time.Hour),
+		DailyUsageUSD:    15.0, // 超过限额
+		DailyWindowStart: ptrTime(now.Add(-1 * time.Hour)),
+	}
+	group := &Group{
+		Name:          "Pro",
+		DailyLimitUSD: ptrFloat64(10.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Daily)
+	assert.Equal(t, 100.0, progress.Daily.Percentage, "超额使用应被截断为 100%")
+	assert.Equal(t, 0.0, progress.Daily.RemainingUSD, "超额使用时剩余应为 0")
+}
+
+func TestCalculateProgress_NoWindowStart_NoProgress(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+
+	// 有限额但无窗口起始时间（订阅未激活）
+	sub := &UserSubscription{
+		ID:             1,
+		ExpiresAt:      now.Add(10 * 24 * time.Hour),
+		DailyUsageUSD:  0,
+		WeeklyUsageUSD: 0,
+	}
+	group := &Group{
+		Name:           "Pro",
+		DailyLimitUSD:  ptrFloat64(10.0),
+		WeeklyLimitUSD: ptrFloat64(50.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	assert.Nil(t, progress.Daily, "无 DailyWindowStart 时 Daily 应为 nil")
+	assert.Nil(t, progress.Weekly, "无 WeeklyWindowStart 时 Weekly 应为 nil")
+}
+
+func TestCalculateProgress_AllLimits(t *testing.T) {
+	svc := newTestSubscriptionService()
+	now := time.Now()
+
+	sub := &UserSubscription{
+		ID:                 1,
+		ExpiresAt:          now.Add(10 * 24 * time.Hour),
+		DailyUsageUSD:      5.0,
+		WeeklyUsageUSD:     20.0,
+		MonthlyUsageUSD:    60.0,
+		DailyWindowStart:   ptrTime(now.Add(-6 * time.Hour)),
+		WeeklyWindowStart:  ptrTime(now.Add(-3 * 24 * time.Hour)),
+		MonthlyWindowStart: ptrTime(now.Add(-15 * 24 * time.Hour)),
+	}
+	group := &Group{
+		Name:            "Full",
+		DailyLimitUSD:   ptrFloat64(10.0),
+		WeeklyLimitUSD:  ptrFloat64(50.0),
+		MonthlyLimitUSD: ptrFloat64(100.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Daily)
+	require.NotNil(t, progress.Weekly)
+	require.NotNil(t, progress.Monthly)
+
+	assert.Equal(t, 50.0, progress.Daily.Percentage)
+	assert.Equal(t, 40.0, progress.Weekly.Percentage)
+	assert.Equal(t, 60.0, progress.Monthly.Percentage)
+}
+
+func TestCalculateProgress_ExpiredSubscription(t *testing.T) {
+	svc := newTestSubscriptionService()
+
+	sub := &UserSubscription{
+		ID:        1,
+		ExpiresAt: time.Now().Add(-24 * time.Hour), // 已过期
+	}
+	group := &Group{Name: "Expired"}
+
+	progress := svc.calculateProgress(sub, group)
+
+	assert.Equal(t, 0, progress.ExpiresInDays, "过期订阅的剩余天数应为 0")
+}
+
+func TestCalculateProgress_ResetsInSeconds_NotNegative(t *testing.T) {
+	svc := newTestSubscriptionService()
+	// 使用过去的窗口起始时间，使得重置时间已过
+	pastStart := time.Now().Add(-48 * time.Hour)
+
+	sub := &UserSubscription{
+		ID:               1,
+		ExpiresAt:        time.Now().Add(10 * 24 * time.Hour),
+		DailyUsageUSD:    1.0,
+		DailyWindowStart: ptrTime(pastStart),
+	}
+	group := &Group{
+		Name:          "Test",
+		DailyLimitUSD: ptrFloat64(10.0),
+	}
+
+	progress := svc.calculateProgress(sub, group)
+
+	require.NotNil(t, progress.Daily)
+	assert.GreaterOrEqual(t, progress.Daily.ResetsInSeconds, int64(0),
+		"ResetsInSeconds 不应为负数")
+}