user_platform_quota_cn_providers_migration_test.go 929 B

12345678910111213141516171819202122
  1. package migrations
  2. import (
  3. "strings"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. )
  7. // TestUserPlatformQuotasCNProvidersMigration 校验 224 号迁移把 kimi/zhipu/deepseek
  8. // 加入 user_platform_quotas.platform 的 CHECK 约束(对照 157 号 grok 迁移)。
  9. // 约束未放宽时,注册预填充 8 平台默认配额会整条 INSERT 中止 → 新用户零配额行
  10. // (缺失配额行 = 无限额),管理端设置国产平台配额直接 500。
  11. func TestUserPlatformQuotasCNProvidersMigration(t *testing.T) {
  12. content, err := FS.ReadFile("224_user_platform_quotas_add_cn_providers.sql")
  13. require.NoError(t, err)
  14. sql := strings.Join(strings.Fields(string(content)), " ")
  15. require.Contains(t, sql, "DROP CONSTRAINT IF EXISTS user_platform_quotas_platform_check")
  16. require.Contains(t, sql,
  17. "CHECK (platform IN ('anthropic', 'openai', 'gemini', 'antigravity', 'grok', 'kimi', 'zhipu', 'deepseek'))")
  18. }