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