6b39b344d8
为用户在 anthropic/openai/gemini/antigravity 四个平台上提供日/周/月 三个窗口的 USD 配额管控。配额语义：未设置=不限制，0=禁用，>0=美元上限。 两层模型： - 配置层：系统默认配额，以及 email/linuxdo/oidc/wechat/github/google/ dingtalk 七个鉴权来源的默认配额，存于 settings，以嵌套 JSON 整体读写 （系统 1 个 key + 每个来源 1 个 key），整体替换语义。 - 运行时层：user_platform_quota 表按用户记录实际配额，与配置层解耦。 后端：新增 ent schema 与 140_user_platform_quotas.sql 迁移、repository 与 service 端口、计费链路集成、管理端与用户端读写接口。 前端：管理端设置页配额编辑、用户配额管理 Modal、用户 Dashboard 展示、 中英文案。 Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
|
|
)
|
|
|
|
func TestPlatformFromAPIKey_NilSafe(t *testing.T) {
|
|
if got := PlatformFromAPIKey(nil); got != "" {
|
|
t.Errorf("nil APIKey should yield empty string, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestPlatformFromAPIKey_NilGroup(t *testing.T) {
|
|
k := &APIKey{Group: nil}
|
|
if got := PlatformFromAPIKey(k); got != "" {
|
|
t.Errorf("APIKey with nil Group should yield empty string, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestPlatformFromAPIKey_DerivesFromGroup(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
platform string
|
|
}{
|
|
{"anthropic", "anthropic"},
|
|
{"openai", "openai"},
|
|
{"gemini", "gemini"},
|
|
{"antigravity", "antigravity"},
|
|
{"empty", ""},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
k := &APIKey{
|
|
Group: &Group{Platform: tt.platform},
|
|
}
|
|
got := PlatformFromAPIKey(k)
|
|
if got != tt.platform {
|
|
t.Errorf("PlatformFromAPIKey(%q) = %q, want %q", tt.platform, got, tt.platform)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestQuotaPlatform 锁定配额计量口径：ForcePlatform 路由（如 /antigravity）按 ForcePlatform 计，
|
|
// 否则回退到 Group 平台。preflight 与 post-billing 共用此口径，保证一致。
|
|
func TestQuotaPlatform(t *testing.T) {
|
|
apiKey := &APIKey{Group: &Group{Platform: PlatformAnthropic}}
|
|
|
|
t.Run("no force platform falls back to group platform", func(t *testing.T) {
|
|
if got := QuotaPlatform(context.Background(), apiKey); got != PlatformAnthropic {
|
|
t.Errorf("QuotaPlatform without force = %q, want %q", got, PlatformAnthropic)
|
|
}
|
|
})
|
|
|
|
t.Run("force platform overrides group platform", func(t *testing.T) {
|
|
ctx := context.WithValue(context.Background(), ctxkey.ForcePlatform, PlatformAntigravity)
|
|
if got := QuotaPlatform(ctx, apiKey); got != PlatformAntigravity {
|
|
t.Errorf("QuotaPlatform with force = %q, want %q", got, PlatformAntigravity)
|
|
}
|
|
})
|
|
|
|
t.Run("empty force platform falls back to group platform", func(t *testing.T) {
|
|
ctx := context.WithValue(context.Background(), ctxkey.ForcePlatform, "")
|
|
if got := QuotaPlatform(ctx, apiKey); got != PlatformAnthropic {
|
|
t.Errorf("QuotaPlatform with empty force = %q, want %q", got, PlatformAnthropic)
|
|
}
|
|
})
|
|
|
|
t.Run("nil api key with force platform returns force platform", func(t *testing.T) {
|
|
ctx := context.WithValue(context.Background(), ctxkey.ForcePlatform, PlatformAntigravity)
|
|
if got := QuotaPlatform(ctx, nil); got != PlatformAntigravity {
|
|
t.Errorf("QuotaPlatform(nil) with force = %q, want %q", got, PlatformAntigravity)
|
|
}
|
|
})
|
|
}
|