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

backend/internal/handler/user_platform_quotas_handler_test.go

@@ -0,0 +1,212 @@
+//go:build unit
+
+package handler
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/Wei-Shaw/sub2api/internal/handler/quotaview"
+	"github.com/Wei-Shaw/sub2api/internal/pkg/timezone"
+	middleware2 "github.com/Wei-Shaw/sub2api/internal/server/middleware"
+	"github.com/Wei-Shaw/sub2api/internal/service"
+	"github.com/gin-gonic/gin"
+)
+
+// fakeQuotaRepoForUserHandler 实现 service.UserPlatformQuotaRepository 最小子集
+type fakeQuotaRepoForUserHandler struct {
+	service.UserPlatformQuotaRepository
+	records []service.UserPlatformQuotaRecord
+}
+
+func (f *fakeQuotaRepoForUserHandler) ListByUser(_ context.Context, _ int64) ([]service.UserPlatformQuotaRecord, error) {
+	return f.records, nil
+}
+
+func TestGetMyPlatformQuotas_EmptyReturns200WithEmptyArray(t *testing.T) {
+	repo := &fakeQuotaRepoForUserHandler{records: nil}
+	h := &UserHandler{userPlatformQuotaRepo: repo}
+	gin.SetMode(gin.TestMode)
+	w := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(w)
+	c.Request = httptest.NewRequest(http.MethodGet, "/user/platform-quotas", nil)
+	c.Set(string(middleware2.ContextKeyUser), middleware2.AuthSubject{UserID: 42})
+	h.GetMyPlatformQuotas(c)
+	if w.Code != 200 {
+		t.Fatalf("expected 200, got %d. body: %s", w.Code, w.Body.String())
+	}
+	var body struct {
+		Code int `json:"code"`
+		Data struct {
+			PlatformQuotas []any `json:"platform_quotas"`
+		} `json:"data"`
+	}
+	if err := json.Unmarshal(w.Body.Bytes(), &body); err != nil {
+		t.Fatalf("unmarshal error: %v, body: %s", err, w.Body.String())
+	}
+	if body.Code != 0 {
+		t.Errorf("expected code=0, got %d", body.Code)
+	}
+	if body.Data.PlatformQuotas == nil {
+		// nil 和 empty slice 均视为可接受（JSON 可能序列化为 null 或 []）
+		// 此断言只验证 HTTP 200 + code=0 即可
+	}
+}
+
+func TestGetMyPlatformQuotas_D14_LazyZeroForExpiredWindow(t *testing.T) {
+	pastStart := time.Now().UTC().AddDate(0, 0, -2)
+	daily := 5.0
+	repo := &fakeQuotaRepoForUserHandler{records: []service.UserPlatformQuotaRecord{{
+		UserID:           42,
+		Platform:         "anthropic",
+		DailyLimitUSD:    &daily,
+		DailyUsageUSD:    3.0,
+		DailyWindowStart: &pastStart,
+	}}}
+	h := &UserHandler{userPlatformQuotaRepo: repo}
+	gin.SetMode(gin.TestMode)
+	w := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(w)
+	c.Request = httptest.NewRequest(http.MethodGet, "/user/platform-quotas", nil)
+	c.Set(string(middleware2.ContextKeyUser), middleware2.AuthSubject{UserID: 42})
+	h.GetMyPlatformQuotas(c)
+
+	if w.Code != 200 {
+		t.Fatalf("expected 200, got %d. body: %s", w.Code, w.Body.String())
+	}
+
+	// 解析 response，验证过期 daily 的 usage_usd=0 且 window_resets_at=null
+	body := w.Body.String()
+	if !strings.Contains(body, `"daily_usage_usd":0`) {
+		t.Errorf("expected daily_usage_usd:0 in body, got: %s", body)
+	}
+	if !strings.Contains(body, `"daily_window_resets_at":null`) {
+		t.Errorf("expected daily_window_resets_at:null in body, got: %s", body)
+	}
+}
+
+func TestGetMyPlatformQuotas_NilRepo_Returns200Empty(t *testing.T) {
+	h := &UserHandler{userPlatformQuotaRepo: nil}
+	gin.SetMode(gin.TestMode)
+	w := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(w)
+	c.Request = httptest.NewRequest(http.MethodGet, "/user/platform-quotas", nil)
+	c.Set(string(middleware2.ContextKeyUser), middleware2.AuthSubject{UserID: 99})
+	h.GetMyPlatformQuotas(c)
+	if w.Code != 200 {
+		t.Fatalf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetMyPlatformQuotas_NoAuth_Returns401(t *testing.T) {
+	h := &UserHandler{userPlatformQuotaRepo: nil}
+	gin.SetMode(gin.TestMode)
+	w := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(w)
+	c.Request = httptest.NewRequest(http.MethodGet, "/user/platform-quotas", nil)
+	// 不设置 auth subject
+	h.GetMyPlatformQuotas(c)
+	if w.Code != 401 {
+		t.Fatalf("expected 401, got %d", w.Code)
+	}
+}
+
+func TestLazyZeroQuotaForResponse_UserViewStripsWindowStart(t *testing.T) {
+	start := time.Now().UTC().Add(-1 * time.Hour)
+	r := service.UserPlatformQuotaRecord{
+		Platform:         "anthropic",
+		DailyUsageUSD:    1.0,
+		DailyWindowStart: &start,
+	}
+	out := quotaview.LazyZeroQuotaForResponse(r, time.Now().UTC(), false)
+	if _, ok := out["daily_window_start"]; ok {
+		t.Error("user view should not include daily_window_start")
+	}
+}
+
+func TestLazyZeroQuotaForResponse_AdminViewIncludesWindowStart(t *testing.T) {
+	start := time.Now().UTC().Add(-1 * time.Hour)
+	r := service.UserPlatformQuotaRecord{
+		Platform:         "anthropic",
+		DailyWindowStart: &start,
+	}
+	out := quotaview.LazyZeroQuotaForResponse(r, time.Now().UTC(), true)
+	if _, ok := out["daily_window_start"]; !ok {
+		t.Error("admin view should include daily_window_start")
+	}
+}
+
+func TestLazyZeroQuotaForResponse_ActiveWindowPreservesUsage(t *testing.T) {
+	// 今天的窗口起始时间（不过期）：按全局时区取当天 0 点，与 view 层同口径
+	now := time.Now()
+	today := timezone.StartOfDay(now)
+	usage := 2.5
+	r := service.UserPlatformQuotaRecord{
+		Platform:         "openai",
+		DailyUsageUSD:    usage,
+		DailyWindowStart: &today,
+	}
+	out := quotaview.LazyZeroQuotaForResponse(r, now, false)
+	if out["daily_usage_usd"] != usage {
+		t.Errorf("expected daily_usage_usd=%v, got %v", usage, out["daily_usage_usd"])
+	}
+	// 活跃窗口应有 resets_at（非 nil）
+	if out["daily_window_resets_at"] == nil {
+		t.Error("active window should have daily_window_resets_at set")
+	}
+}
+
+func TestNeedsDailyReset_NilStart_ReturnsFalse(t *testing.T) {
+	if quotaview.NeedsDailyReset(nil, time.Now().UTC()) {
+		t.Error("nil start should not need reset")
+	}
+}
+
+func TestNeedsDailyReset_OldStart_ReturnsTrue(t *testing.T) {
+	old := time.Now().UTC().AddDate(0, 0, -1)
+	if !quotaview.NeedsDailyReset(&old, time.Now().UTC()) {
+		t.Error("yesterday start should need daily reset")
+	}
+}
+
+func TestNeedsWeeklyReset_NilStart_ReturnsFalse(t *testing.T) {
+	if quotaview.NeedsWeeklyReset(nil, time.Now().UTC()) {
+		t.Error("nil start should not need weekly reset")
+	}
+}
+
+func TestNeedsMonthlyReset_NilStart_ReturnsFalse(t *testing.T) {
+	if quotaview.NeedsMonthlyReset(nil, time.Now().UTC()) {
+		t.Error("nil start should not need monthly reset")
+	}
+}
+
+// TestNeedsMonthlyReset_30DayRolling 验证 30 天滚动语义（C-NEW-1）。
+func TestNeedsMonthlyReset_30DayRolling_Expired(t *testing.T) {
+	start := time.Now().UTC().Add(-31 * 24 * time.Hour) // 31 天前，已过期
+	if !quotaview.NeedsMonthlyReset(&start, time.Now().UTC()) {
+		t.Error("31 days ago should need monthly reset (30-day rolling)")
+	}
+}
+
+func TestNeedsMonthlyReset_30DayRolling_Active(t *testing.T) {
+	start := time.Now().UTC().Add(-15 * 24 * time.Hour) // 15 天前，窗口有效
+	if quotaview.NeedsMonthlyReset(&start, time.Now().UTC()) {
+		t.Error("15 days ago should NOT need monthly reset (30-day rolling, still active)")
+	}
+}
+
+// TestNeedsMonthlyReset_CrossMonthBoundary 验证跨自然月时 30 天未满不重置（旧自然月语义会提前重置）。
+func TestNeedsMonthlyReset_CrossMonthBoundary(t *testing.T) {
+	// 窗口起始 4 月 20 日；5 月 1 日仅过了 11 天，不足 30 天，不应重置
+	windowStart := time.Date(2026, 4, 20, 0, 0, 0, 0, time.UTC)
+	now := time.Date(2026, 5, 1, 0, 0, 0, 0, time.UTC)
+	if quotaview.NeedsMonthlyReset(&windowStart, now) {
+		t.Error("cross-month boundary within 30 days should NOT trigger reset (30-day rolling)")
+	}
+}