composite_routes_cn_providers_migration_test.go 560 B

123456789101112131415161718
  1. package migrations
  2. import (
  3. "strings"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestCompositeRoutesCNProvidersMigration(t *testing.T) {
  8. content, err := FS.ReadFile("227_composite_routes_add_cn_providers.sql")
  9. require.NoError(t, err)
  10. sql := strings.Join(strings.Fields(string(content)), " ")
  11. require.Contains(t, sql, "DROP CONSTRAINT IF EXISTS composite_model_routes_target_platform_check")
  12. require.Contains(t, sql,
  13. "CHECK (target_platform IN ('anthropic', 'openai', 'gemini', 'antigravity', 'grok', 'kimi', 'zhipu', 'deepseek'))")
  14. }