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

backend/internal/service/auth_email_oauth_auto_test.go

@@ -0,0 +1,88 @@
+//go:build unit
+
+package service
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Wei-Shaw/sub2api/internal/config"
+	"github.com/stretchr/testify/require"
+)
+
+func newEmailOAuthAutoAuthService(
+	userRepo UserRepository,
+	settings map[string]string,
+	quotaRepo UserPlatformQuotaRepository,
+) *AuthService {
+	cfg := &config.Config{
+		JWT: config.JWTConfig{
+			Secret:                   "test-secret",
+			ExpireHour:               1,
+			AccessTokenExpireMinutes: 60,
+			RefreshTokenExpireDays:   7,
+		},
+		Default: config.DefaultConfig{
+			UserBalance:     3.5,
+			UserConcurrency: 2,
+		},
+	}
+
+	settingService := NewSettingService(&settingRepoStub{values: settings}, cfg)
+
+	return NewAuthService(
+		nil, // entClient — nil, updateUserSignupSource early return
+		userRepo,
+		nil, // redeemRepo — invitationCode="" 时不触发
+		&refreshTokenCacheStub{},
+		cfg,
+		settingService,
+		nil, // emailService
+		nil, // turnstileService
+		nil, // emailQueueService
+		nil, // promoService
+		nil, // defaultSubAssigner — nil, assignSubscriptions early return
+		nil, // affiliateService — nil, bindOAuthAffiliate early return
+		quotaRepo,
+	)
+}
+
+func TestEmailOAuthAuto_SnapshotsPlatformQuotaDefaults(t *testing.T) {
+	userRepo := &userRepoStub{nextID: 88}
+	quotaRepo := &userPlatformQuotaRepoStub{}
+
+	svc := newEmailOAuthAutoAuthService(
+		userRepo,
+		map[string]string{
+			SettingKeyRegistrationEnabled:   "true",
+			SettingKeyDefaultPlatformQuotas: `{"gemini": {"monthly": 100.0}}`,
+		},
+		quotaRepo,
+	)
+
+	user, err := svc.createEmailOAuthUser(
+		context.Background(),
+		"newoauth@example.com",
+		"newoauth",
+		"github",
+		"", // invitationCode
+		"", // affiliateCode
+	)
+	require.NoError(t, err)
+	require.NotNil(t, user)
+	require.Equal(t, int64(88), user.ID)
+
+	require.Len(t, quotaRepo.bulkInsertCalls, 1, "createEmailOAuthUser must snapshot platform quotas via BulkInsertInitial")
+
+	records := quotaRepo.bulkInsertCalls[0]
+	var geminiRecord *UserPlatformQuotaRecord
+	for i := range records {
+		if records[i].Platform == "gemini" {
+			geminiRecord = &records[i]
+			break
+		}
+	}
+	require.NotNil(t, geminiRecord, "expected gemini platform record")
+	require.NotNil(t, geminiRecord.MonthlyLimitUSD)
+	require.InDelta(t, 100.0, *geminiRecord.MonthlyLimitUSD, 0.0001)
+}