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>
345 lines
13 KiB
Go
345 lines
13 KiB
Go
//go:build unit
|
||
|
||
package service
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/config"
|
||
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
func TestMergePlatformQuotaDefaults_PatchSemantics(t *testing.T) {
|
||
five := 5.0
|
||
base := DefaultPlatformQuotaSetting{
|
||
DailyLimitUSD: &five,
|
||
WeeklyLimitUSD: &five,
|
||
}
|
||
ten := 10.0
|
||
patch := DefaultPlatformQuotaSetting{DailyLimitUSD: &ten}
|
||
|
||
mergePlatformQuotaDefaults(&base, &patch)
|
||
if base.DailyLimitUSD == nil || *base.DailyLimitUSD != 10.0 {
|
||
t.Errorf("daily not patched: %+v", base.DailyLimitUSD)
|
||
}
|
||
if base.WeeklyLimitUSD == nil || *base.WeeklyLimitUSD != 5.0 {
|
||
t.Errorf("weekly should remain 5.0: %+v", base.WeeklyLimitUSD)
|
||
}
|
||
}
|
||
|
||
func TestMergePlatformQuotaDefaults_ZeroIsExplicitDisable(t *testing.T) {
|
||
five := 5.0
|
||
base := DefaultPlatformQuotaSetting{DailyLimitUSD: &five}
|
||
zero := 0.0
|
||
patch := DefaultPlatformQuotaSetting{DailyLimitUSD: &zero}
|
||
|
||
mergePlatformQuotaDefaults(&base, &patch)
|
||
if base.DailyLimitUSD == nil || *base.DailyLimitUSD != 0 {
|
||
t.Errorf("explicit 0 should patch base, got %+v", base.DailyLimitUSD)
|
||
}
|
||
}
|
||
|
||
func TestMergePlatformQuotaDefaults_NilSrcIsNoop(t *testing.T) {
|
||
five := 5.0
|
||
base := DefaultPlatformQuotaSetting{DailyLimitUSD: &five}
|
||
mergePlatformQuotaDefaults(&base, nil)
|
||
if base.DailyLimitUSD == nil || *base.DailyLimitUSD != 5.0 {
|
||
t.Errorf("nil src should be no-op: %+v", base.DailyLimitUSD)
|
||
}
|
||
}
|
||
|
||
func floatPtrPQ(v float64) *float64 { return &v }
|
||
|
||
func newSettingServiceForPlatformQuotaTest(seed map[string]string) *SettingService {
|
||
repo := newMockSettingRepo()
|
||
for k, v := range seed {
|
||
repo.data[k] = v
|
||
}
|
||
return NewSettingService(repo, &config.Config{})
|
||
}
|
||
|
||
func TestGetDefaultPlatformQuotas_ReturnsFourPlatforms(t *testing.T) {
|
||
zero := 0.0
|
||
svc := newSettingServiceForPlatformQuotaTest(map[string]string{
|
||
// 新 JSON 格式：anthropic daily=10.5, openai monthly=0, gemini/antigravity 无配置
|
||
SettingKeyDefaultPlatformQuotas: `{"anthropic":{"daily":10.5},"openai":{"monthly":0}}`,
|
||
})
|
||
got, err := svc.GetDefaultPlatformQuotas(context.Background())
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
// 必须包含全部 4 个 platform key（补齐契约）
|
||
for _, platform := range []string{"anthropic", "openai", "gemini", "antigravity"} {
|
||
if _, ok := got[platform]; !ok {
|
||
t.Errorf("missing platform key: %q", platform)
|
||
}
|
||
}
|
||
// anthropic daily = 10.5
|
||
if v := got["anthropic"].DailyLimitUSD; v == nil || *v != 10.5 {
|
||
t.Errorf("anthropic daily want 10.5, got %v", v)
|
||
}
|
||
// openai monthly = 0（显式禁用）
|
||
if v := got["openai"].MonthlyLimitUSD; v == nil || *v != zero {
|
||
t.Errorf("openai monthly want 0 (explicit disable), got %v", v)
|
||
}
|
||
// gemini 无配置 → weekly = nil
|
||
if v := got["gemini"].WeeklyLimitUSD; v != nil {
|
||
t.Errorf("gemini weekly want nil (not configured), got %v", *v)
|
||
}
|
||
// antigravity 无配置 → daily = nil
|
||
if v := got["antigravity"].DailyLimitUSD; v != nil {
|
||
t.Errorf("antigravity daily want nil (not configured), got %v", *v)
|
||
}
|
||
}
|
||
|
||
func TestGetAuthSourcePlatformQuotas_OnlyConfiguredReturned(t *testing.T) {
|
||
source := "email"
|
||
// 新 JSON 格式：anthropic daily=5, monthly=100；openai weekly=0；gemini/antigravity 无配置
|
||
svc := newSettingServiceForPlatformQuotaTest(map[string]string{
|
||
SettingKeyAuthSourcePlatformQuotas(source): `{"anthropic":{"daily":5,"monthly":100},"openai":{"weekly":0}}`,
|
||
})
|
||
got := svc.GetAuthSourcePlatformQuotas(context.Background(), source)
|
||
|
||
// anthropic 有配置 → 在结果中
|
||
anthro, ok := got["anthropic"]
|
||
if !ok {
|
||
t.Fatal("expected anthropic to be present")
|
||
}
|
||
if anthro.DailyLimitUSD == nil || *anthro.DailyLimitUSD != 5.0 {
|
||
t.Errorf("anthropic daily want 5.0, got %v", anthro.DailyLimitUSD)
|
||
}
|
||
if anthro.MonthlyLimitUSD == nil || *anthro.MonthlyLimitUSD != 100.0 {
|
||
t.Errorf("anthropic monthly want 100.0, got %v", anthro.MonthlyLimitUSD)
|
||
}
|
||
if anthro.WeeklyLimitUSD != nil {
|
||
t.Errorf("anthropic weekly not configured, want nil, got %v", *anthro.WeeklyLimitUSD)
|
||
}
|
||
|
||
// openai weekly=0 → 在结果中
|
||
oai, ok := got["openai"]
|
||
if !ok {
|
||
t.Fatal("expected openai to be present")
|
||
}
|
||
if oai.WeeklyLimitUSD == nil || *oai.WeeklyLimitUSD != 0 {
|
||
t.Errorf("openai weekly want 0, got %v", oai.WeeklyLimitUSD)
|
||
}
|
||
|
||
// gemini / antigravity 无配置 → 不在结果中（override 语义）
|
||
if _, ok := got["gemini"]; ok {
|
||
t.Error("gemini not configured, should be absent from result")
|
||
}
|
||
if _, ok := got["antigravity"]; ok {
|
||
t.Error("antigravity not configured, should be absent from result")
|
||
}
|
||
}
|
||
|
||
func TestGetAuthSourcePlatformQuotas_AllNegativeOrEmpty_NoEntry(t *testing.T) {
|
||
source := "linuxdo"
|
||
// 新 JSON 格式：未配置任何平台（空 JSON key）→ 返回空 map
|
||
svc := newSettingServiceForPlatformQuotaTest(map[string]string{
|
||
SettingKeyAuthSourcePlatformQuotas(source): `{}`,
|
||
})
|
||
got := svc.GetAuthSourcePlatformQuotas(context.Background(), source)
|
||
// 空 map → override 语义，无 openai 条目
|
||
if _, ok := got["openai"]; ok {
|
||
t.Error("empty JSON object should result in no openai entry")
|
||
}
|
||
if len(got) != 0 {
|
||
t.Errorf("expected empty result map, got %v", got)
|
||
}
|
||
}
|
||
|
||
// TestSystemPlatformQuotas_WriteReadRoundTrip 验证系统层 platform quota 经 buildSystemSettingsUpdates（写）
|
||
// 再由 GetDefaultPlatformQuotas（读）正确往返——覆盖真实 write→read 路径，锁住 4-key 补齐契约。
|
||
func TestSystemPlatformQuotas_WriteReadRoundTrip(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(nil)
|
||
ctx := context.Background()
|
||
|
||
ten := 10.0
|
||
ss := &SystemSettings{
|
||
DefaultPlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &ten, WeeklyLimitUSD: nil, MonthlyLimitUSD: nil},
|
||
},
|
||
}
|
||
if err := svc.UpdateSettings(ctx, ss); err != nil {
|
||
t.Fatalf("UpdateSettings: %v", err)
|
||
}
|
||
|
||
got, err := svc.GetDefaultPlatformQuotas(ctx)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
// 4-key 补齐契约：无论写了几个 platform，读回必须含全部 4 个
|
||
for _, p := range []string{"anthropic", "openai", "gemini", "antigravity"} {
|
||
if _, ok := got[p]; !ok {
|
||
t.Errorf("4-key contract violated: missing platform %q", p)
|
||
}
|
||
}
|
||
// 写入值正确往返
|
||
if v := got["anthropic"].DailyLimitUSD; v == nil || *v != ten {
|
||
t.Fatalf("anthropic daily round-trip failed: got %v, want 10", v)
|
||
}
|
||
// 未写入的平台字段为 nil
|
||
if got["openai"].DailyLimitUSD != nil {
|
||
t.Errorf("openai daily should be nil (not written), got %v", got["openai"].DailyLimitUSD)
|
||
}
|
||
}
|
||
|
||
// TestSystemPlatformQuotas_EmptyMapClearsAll 验证空 map 的整体替换语义：
|
||
// 写入 DefaultPlatformQuotas={} 后，GetDefaultPlatformQuotas 返回 4 个平台、所有字段均为 nil，
|
||
// 明确文档化"空 map = 清空全部配额"是有意为之的 whole-replace 语义。
|
||
func TestSystemPlatformQuotas_EmptyMapClearsAll(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(nil)
|
||
ctx := context.Background()
|
||
|
||
// 先写入有值的配置
|
||
ten := 10.0
|
||
if err := svc.UpdateSettings(ctx, &SystemSettings{
|
||
DefaultPlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &ten},
|
||
},
|
||
}); err != nil {
|
||
t.Fatalf("initial write: %v", err)
|
||
}
|
||
|
||
// 再写入空 map（整体替换语义：清空全部）
|
||
if err := svc.UpdateSettings(ctx, &SystemSettings{
|
||
DefaultPlatformQuotas: map[string]*DefaultPlatformQuotaSetting{},
|
||
}); err != nil {
|
||
t.Fatalf("empty map write: %v", err)
|
||
}
|
||
|
||
got, err := svc.GetDefaultPlatformQuotas(ctx)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
// 4 个 key 仍然存在（补齐契约）
|
||
for _, p := range []string{"anthropic", "openai", "gemini", "antigravity"} {
|
||
if _, ok := got[p]; !ok {
|
||
t.Errorf("4-key contract violated after empty write: missing %q", p)
|
||
}
|
||
}
|
||
// 所有字段 nil（全部已清空）
|
||
for _, p := range AllowedQuotaPlatforms {
|
||
pq := got[p]
|
||
if pq == nil {
|
||
continue
|
||
}
|
||
if pq.DailyLimitUSD != nil || pq.WeeklyLimitUSD != nil || pq.MonthlyLimitUSD != nil {
|
||
t.Errorf("platform %q should have all-nil limits after empty-map write, got %+v", p, pq)
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestUpdateSettingsWithAuthSourceDefaults_PlatformQuotaRoundTrip 验证 round-4 fix：
|
||
// PUT /admin/settings 携带的 auth source × platform × window 限额能完整写入并被 GetAuthSourcePlatformQuotas 读回。
|
||
// Round-4 之前 writeProviderDefaultGrantUpdates 完全没写 PQ key，前端配置静默丢失。
|
||
func TestUpdateSettingsWithAuthSourceDefaults_PlatformQuotaRoundTrip(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(nil)
|
||
systemSettings := &SystemSettings{}
|
||
authDefaults := &AuthSourceDefaultSettings{
|
||
Email: ProviderDefaultGrantSettings{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {
|
||
DailyLimitUSD: floatPtrPQ(5.0),
|
||
WeeklyLimitUSD: nil, // 无限额
|
||
MonthlyLimitUSD: floatPtrPQ(100.0),
|
||
},
|
||
"openai": {
|
||
DailyLimitUSD: floatPtrPQ(0), // 显式禁用
|
||
},
|
||
},
|
||
},
|
||
}
|
||
if err := svc.UpdateSettingsWithAuthSourceDefaults(context.Background(), systemSettings, authDefaults); err != nil {
|
||
t.Fatalf("UpdateSettingsWithAuthSourceDefaults: %v", err)
|
||
}
|
||
got := svc.GetAuthSourcePlatformQuotas(context.Background(), "email")
|
||
anthro := got["anthropic"]
|
||
if anthro == nil || anthro.DailyLimitUSD == nil || *anthro.DailyLimitUSD != 5.0 {
|
||
t.Errorf("anthropic daily round-trip failed: %+v", anthro)
|
||
}
|
||
if anthro != nil && anthro.WeeklyLimitUSD != nil {
|
||
t.Errorf("anthropic weekly want nil (无限额), got %v", *anthro.WeeklyLimitUSD)
|
||
}
|
||
if anthro == nil || anthro.MonthlyLimitUSD == nil || *anthro.MonthlyLimitUSD != 100.0 {
|
||
t.Errorf("anthropic monthly round-trip failed: %+v", anthro)
|
||
}
|
||
oai := got["openai"]
|
||
if oai == nil || oai.DailyLimitUSD == nil || *oai.DailyLimitUSD != 0 {
|
||
t.Errorf("openai daily=0 (禁用) round-trip failed: %+v", oai)
|
||
}
|
||
// 其他 source 不应有 quota（authDefaults 只填了 Email）
|
||
if linux := svc.GetAuthSourcePlatformQuotas(context.Background(), "linuxdo"); len(linux) != 0 {
|
||
t.Errorf("linuxdo should be empty, got %+v", linux)
|
||
}
|
||
}
|
||
|
||
// TestUpdateSettingsWithAuthSourceDefaults_NilPlatformQuotaPreservesExisting 验证 #2 防御：
|
||
// 请求未携带某 auth source 的 platform quota（nil）时跳过写入、保留既有配置，
|
||
// 而非整体替换为空 map 清空（与系统层 nil 守卫一致）。
|
||
func TestUpdateSettingsWithAuthSourceDefaults_NilPlatformQuotaPreservesExisting(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(map[string]string{
|
||
SettingKeyAuthSourcePlatformQuotas("email"): `{"anthropic":{"daily":5,"weekly":null,"monthly":null}}`,
|
||
})
|
||
// authDefaults 不携带 Email 的 PlatformQuotas（nil）——应保留既有配置
|
||
authDefaults := &AuthSourceDefaultSettings{
|
||
Email: ProviderDefaultGrantSettings{PlatformQuotas: nil},
|
||
}
|
||
if err := svc.UpdateSettingsWithAuthSourceDefaults(context.Background(), &SystemSettings{}, authDefaults); err != nil {
|
||
t.Fatalf("UpdateSettingsWithAuthSourceDefaults: %v", err)
|
||
}
|
||
anthro := svc.GetAuthSourcePlatformQuotas(context.Background(), "email")["anthropic"]
|
||
if anthro == nil || anthro.DailyLimitUSD == nil || *anthro.DailyLimitUSD != 5.0 {
|
||
t.Errorf("nil PlatformQuotas 应保留既有 anthropic daily=5，got %+v", anthro)
|
||
}
|
||
}
|
||
|
||
// TestGetAuthSourcePlatformQuotas_JSON 验证新 JSON key 读写语义：
|
||
// 写入 JSON，断言已配置平台在结果中、未配置平台不在结果中（override 语义）。
|
||
func TestGetAuthSourcePlatformQuotas_JSON(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(map[string]string{
|
||
SettingKeyAuthSourcePlatformQuotas("email"): `{"openai":{"daily":null,"weekly":null,"monthly":20}}`,
|
||
})
|
||
got := svc.GetAuthSourcePlatformQuotas(context.Background(), "email")
|
||
|
||
// openai monthly = 20
|
||
oai, ok := got["openai"]
|
||
if !ok {
|
||
t.Fatal("expected openai to be present")
|
||
}
|
||
if oai.MonthlyLimitUSD == nil || *oai.MonthlyLimitUSD != 20 {
|
||
t.Errorf("openai monthly want 20, got %v", oai.MonthlyLimitUSD)
|
||
}
|
||
if oai.DailyLimitUSD != nil {
|
||
t.Errorf("openai daily want nil, got %v", *oai.DailyLimitUSD)
|
||
}
|
||
if oai.WeeklyLimitUSD != nil {
|
||
t.Errorf("openai weekly want nil, got %v", *oai.WeeklyLimitUSD)
|
||
}
|
||
|
||
// anthropic 未配置 → 不在结果中（override 语义）
|
||
if _, ok := got["anthropic"]; ok {
|
||
t.Error("anthropic not configured, should be absent from result")
|
||
}
|
||
}
|
||
|
||
// TestUpdateSettingsWithAuthSourceDefaults_NegativeQuotaRejected 验证改动 C：
|
||
// auth-source platform quota 含负数时，UpdateSettingsWithAuthSourceDefaults 返回 BadRequest 错误。
|
||
func TestUpdateSettingsWithAuthSourceDefaults_NegativeQuotaRejected(t *testing.T) {
|
||
svc := newSettingServiceForPlatformQuotaTest(nil)
|
||
neg := -1.0
|
||
authDefaults := &AuthSourceDefaultSettings{
|
||
Email: ProviderDefaultGrantSettings{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &neg},
|
||
},
|
||
},
|
||
}
|
||
err := svc.UpdateSettingsWithAuthSourceDefaults(context.Background(), &SystemSettings{}, authDefaults)
|
||
require.Error(t, err, "expected error for negative quota")
|
||
require.Equal(t, "INVALID_DEFAULT_PLATFORM_QUOTA", infraerrors.Reason(err))
|
||
}
|