mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-11-02 00:30:32 +00:00
feeeei
4a4b50d807
user_platform_quotas 只保存至少配置了一档限额的记录；不存在的行等价于不限额。 - 注册播种只为配置了限额的平台建行，一个都没有时不写入 - IncrementUsageWithReset 无行时不再 fail-open 建行 - BatchSnapshotUsage 改为只更新既有活跃行，不再插入无限额的新行 - 仓储 BulkInsertInitial/UpsertForUser 把三档全空的记录视为未配置； 管理员全量替换接口同样剔除，清空限额即软删并放弃累计用量 - 迁移 238 删除三档全 NULL 的历史行 - 用户仪表盘平台卡片 = 有用量的平台 ∪ 配置了限额的平台，计数改为卡片数 - 管理员限额弹窗对未保存限额的平台禁用"重置窗口"
236 lines
8.7 KiB
Go
236 lines
8.7 KiB
Go
//go:build unit
|
||
|
||
package service
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"testing"
|
||
"time"
|
||
|
||
dbent "github.com/Wei-Shaw/sub2api/ent"
|
||
)
|
||
|
||
// fakeInsertRecorder 记录 BulkInsertInitial 调用，实现 UserPlatformQuotaRepository port。
|
||
type fakeInsertRecorder struct {
|
||
records []UserPlatformQuotaRecord
|
||
err error
|
||
calls int // BulkInsertInitial 被调用的次数
|
||
lastCtx context.Context // 捕获最后一次 BulkInsertInitial 收到的 ctx（用于断言事务隔离）
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) GetByUserPlatform(_ context.Context, _ int64, _ string) (*UserPlatformQuotaRecord, error) {
|
||
return nil, nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) BulkInsertInitial(ctx context.Context, recs []UserPlatformQuotaRecord) error {
|
||
f.calls++
|
||
f.lastCtx = ctx
|
||
if f.err != nil {
|
||
return f.err
|
||
}
|
||
f.records = append(f.records, recs...)
|
||
return nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) IncrementUsageWithReset(_ context.Context, _ int64, _ string, _ float64, _ time.Time) error {
|
||
return nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) ListByUser(_ context.Context, _ int64) ([]UserPlatformQuotaRecord, error) {
|
||
return nil, nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) UpsertForUser(_ context.Context, _ int64, _ []UserPlatformQuotaRecord) error {
|
||
return nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) ResetExpiredWindow(_ context.Context, _ int64, _ string, _ string, _ time.Time) error {
|
||
return nil
|
||
}
|
||
|
||
func (f *fakeInsertRecorder) BatchSnapshotUsage(_ context.Context, _ []UserPlatformQuotaSnapshot, _ time.Time) error {
|
||
return nil
|
||
}
|
||
|
||
// TestSnapshotPlatformQuotaDefaults_PassesToRepoBulkInsert 锁定不变式：只有至少配置了一档限额的
|
||
// 平台才建行；三档全空（含 nil 条目）的平台不进入 BulkInsertInitial。
|
||
func TestSnapshotPlatformQuotaDefaults_PassesToRepoBulkInsert(t *testing.T) {
|
||
fakeRepo := &fakeInsertRecorder{}
|
||
s := &AuthService{userPlatformQuotaRepo: fakeRepo}
|
||
|
||
five := 5.0
|
||
zero := 0.0
|
||
plan := &signupGrantPlan{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &five},
|
||
"grok": {MonthlyLimitUSD: &zero}, // 0 = 显式禁用，属于已配置
|
||
"openai": {},
|
||
"gemini": {},
|
||
"antigravity": nil,
|
||
},
|
||
}
|
||
if err := s.snapshotPlatformQuotaDefaults(context.Background(), 999, plan); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(fakeRepo.records) != 2 {
|
||
t.Fatalf("expected 2 records (anthropic, grok), got %d: %+v", len(fakeRepo.records), fakeRepo.records)
|
||
}
|
||
byPlatform := map[string]UserPlatformQuotaRecord{}
|
||
for _, r := range fakeRepo.records {
|
||
if r.UserID != 999 {
|
||
t.Errorf("unexpected user id %d", r.UserID)
|
||
}
|
||
byPlatform[r.Platform] = r
|
||
}
|
||
if r, ok := byPlatform["anthropic"]; !ok || r.DailyLimitUSD == nil || *r.DailyLimitUSD != 5 {
|
||
t.Error("anthropic daily = 5 not snapshotted")
|
||
}
|
||
if r, ok := byPlatform["grok"]; !ok || r.MonthlyLimitUSD == nil || *r.MonthlyLimitUSD != 0 {
|
||
t.Error("grok monthly = 0 (explicit disable) must be snapshotted")
|
||
}
|
||
for _, p := range []string{"openai", "gemini", "antigravity"} {
|
||
if _, ok := byPlatform[p]; ok {
|
||
t.Errorf("platform %q has no configured limit and must not get a row", p)
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestSnapshotPlatformQuotaDefaults_AllUnlimitedSkipsInsert 锁定：没有任何平台配置限额时，
|
||
// 不调用 BulkInsertInitial（不产生全空的占位行）。
|
||
func TestSnapshotPlatformQuotaDefaults_AllUnlimitedSkipsInsert(t *testing.T) {
|
||
fakeRepo := &fakeInsertRecorder{}
|
||
s := &AuthService{userPlatformQuotaRepo: fakeRepo}
|
||
|
||
plan := &signupGrantPlan{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {},
|
||
"openai": {},
|
||
"gemini": nil,
|
||
},
|
||
}
|
||
if err := s.snapshotPlatformQuotaDefaults(context.Background(), 999, plan); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if fakeRepo.calls != 0 {
|
||
t.Fatalf("BulkInsertInitial must not be called when no platform has a configured limit, got %d calls", fakeRepo.calls)
|
||
}
|
||
if len(fakeRepo.records) != 0 {
|
||
t.Errorf("expected no records, got %d", len(fakeRepo.records))
|
||
}
|
||
}
|
||
|
||
// TestSnapshotPlatformQuotaDefaults_DetachesCallerTransaction 锁定 fix① 不变量：
|
||
// 平台配额快照是 best-effort，必须脱离调用方事务执行——这样它失败（例如某平台
|
||
// 违反 user_platform_quotas 的 CHECK 约束）也不会把调用方的注册主事务标记为 aborted。
|
||
// 历史 bug：snapshot 在 OAuth pending handler 的 binding tx 中执行，grok 违约毒化整个
|
||
// 事务 → consumePendingOAuthBrowserSessionTx 撞 "transaction aborted" → 500 → 清 cookie → 404。
|
||
func TestSnapshotPlatformQuotaDefaults_DetachesCallerTransaction(t *testing.T) {
|
||
fakeRepo := &fakeInsertRecorder{}
|
||
s := &AuthService{userPlatformQuotaRepo: fakeRepo}
|
||
|
||
five := 5.0
|
||
plan := &signupGrantPlan{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &five},
|
||
},
|
||
}
|
||
|
||
// 模拟调用方（OAuth pending handler）在事务 ctx 中调用快照
|
||
txCtx := dbent.NewTxContext(context.Background(), &dbent.Tx{})
|
||
|
||
if err := s.snapshotPlatformQuotaDefaults(txCtx, 999, plan); err != nil {
|
||
t.Fatalf("snapshot should not error (fail-open): %v", err)
|
||
}
|
||
|
||
if fakeRepo.lastCtx == nil {
|
||
t.Fatal("expected BulkInsertInitial to be called")
|
||
}
|
||
if dbent.TxFromContext(fakeRepo.lastCtx) != nil {
|
||
t.Error("快照必须脱离调用方事务执行（best-effort，失败不得毒化注册事务），但 repo 收到了仍携带事务的 ctx")
|
||
}
|
||
}
|
||
|
||
func TestSnapshotPlatformQuotaDefaults_NilPlanIsNoop(t *testing.T) {
|
||
fakeRepo := &fakeInsertRecorder{}
|
||
s := &AuthService{userPlatformQuotaRepo: fakeRepo}
|
||
if err := s.snapshotPlatformQuotaDefaults(context.Background(), 1, nil); err != nil {
|
||
t.Errorf("nil plan should be noop, got %v", err)
|
||
}
|
||
if len(fakeRepo.records) != 0 {
|
||
t.Errorf("expected no records, got %d", len(fakeRepo.records))
|
||
}
|
||
}
|
||
|
||
func TestSnapshotPlatformQuotaDefaults_RepoErrorFailsOpen(t *testing.T) {
|
||
fakeRepo := &fakeInsertRecorder{err: fmt.Errorf("db down")}
|
||
s := &AuthService{userPlatformQuotaRepo: fakeRepo}
|
||
five := 5.0
|
||
plan := &signupGrantPlan{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{
|
||
"anthropic": {DailyLimitUSD: &five},
|
||
},
|
||
}
|
||
if err := s.snapshotPlatformQuotaDefaults(context.Background(), 1, plan); err != nil {
|
||
t.Errorf("fail-open: expected nil even on repo error, got %v", err)
|
||
}
|
||
}
|
||
|
||
func TestSnapshotPlatformQuotaDefaults_NilRepoIsNoop(t *testing.T) {
|
||
s := &AuthService{userPlatformQuotaRepo: nil}
|
||
five := 5.0
|
||
plan := &signupGrantPlan{
|
||
PlatformQuotas: map[string]*DefaultPlatformQuotaSetting{"a": {DailyLimitUSD: &five}},
|
||
}
|
||
if err := s.snapshotPlatformQuotaDefaults(context.Background(), 1, plan); err != nil {
|
||
t.Errorf("nil repo should be noop, got %v", err)
|
||
}
|
||
}
|
||
|
||
// resolveSignupGrantPlan 测试：依赖完整的 AuthService 构造，需要 SettingService（含 settingRepoStub）。
|
||
// settingRepoStub 已在 auth_service_register_test.go 中定义，同 package 可直接使用。
|
||
func TestResolveSignupGrantPlan_GlobalQuotaLoadedBeforeAuthSource(t *testing.T) {
|
||
// 全局 quota JSON key（新格式）
|
||
settings := map[string]string{
|
||
SettingKeyRegistrationEnabled: "true",
|
||
SettingKeyDefaultPlatformQuotas: `{
|
||
"anthropic": {"daily": 10, "weekly": 50, "monthly": 200},
|
||
"openai": {"daily": 5, "weekly": 25, "monthly": 100},
|
||
"gemini": {"daily": 5, "weekly": 25, "monthly": 100},
|
||
"antigravity": {"daily": 5, "weekly": 25, "monthly": 100}
|
||
}`,
|
||
}
|
||
svc := newAuthService(nil, settings, nil, nil)
|
||
plan := svc.resolveSignupGrantPlan(context.Background(), "email")
|
||
if plan.PlatformQuotas == nil {
|
||
t.Fatal("expected PlatformQuotas to be non-nil after loading global quota KVs")
|
||
}
|
||
q := plan.PlatformQuotas["anthropic"]
|
||
if q == nil {
|
||
t.Fatal("expected anthropic quota to be set")
|
||
}
|
||
if q.DailyLimitUSD == nil || *q.DailyLimitUSD != 10 {
|
||
t.Errorf("expected anthropic daily=10, got %v", q.DailyLimitUSD)
|
||
}
|
||
}
|
||
|
||
// TestResolveSignupGrantPlan_DisabledAuthSourceStillCarriesGlobalQuota 验证 P1 约束：
|
||
// !enabled 早退路径仍携带全局 quota（GetDefaultPlatformQuotas 在 ResolveAuthSourceGrantSettings 之前）。
|
||
func TestResolveSignupGrantPlan_DisabledAuthSourceStillCarriesGlobalQuota(t *testing.T) {
|
||
settings := map[string]string{
|
||
SettingKeyRegistrationEnabled: "true",
|
||
// auth source 不配置（=> !enabled 路径）
|
||
SettingKeyDefaultPlatformQuotas: `{"anthropic": {"daily": 10, "weekly": 50, "monthly": 200}}`,
|
||
}
|
||
svc := newAuthService(nil, settings, nil, nil)
|
||
plan := svc.resolveSignupGrantPlan(context.Background(), "email")
|
||
// !enabled 路径：plan.PlatformQuotas 应已含全局层（不是 nil）
|
||
if plan.PlatformQuotas == nil {
|
||
t.Fatal("P1 violated: PlatformQuotas is nil even with global quota KVs set")
|
||
}
|
||
// P1 核心断言：disabled auth source 路径不能丢失全局 quota
|
||
if _, ok := plan.PlatformQuotas["anthropic"]; !ok {
|
||
t.Error("P1 violated: disabled auth source path dropped global platform quota")
|
||
}
|
||
}
|