李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
38 lines
1.6 KiB
Go
38 lines
1.6 KiB
Go
package migrations
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestChannelMonitorQuotaModeMigration(t *testing.T) {
|
|
content, err := FS.ReadFile("226_channel_monitor_quota_mode.sql")
|
|
require.NoError(t, err)
|
|
|
|
sql := strings.Join(strings.Fields(string(content)), " ")
|
|
|
|
// provider CHECK 两张表扩到 8 平台，且带幂等守卫（仿 176 grok 迁移）。
|
|
require.Contains(t, sql, "channel_monitors_provider_check")
|
|
require.Contains(t, sql, "channel_monitor_request_templates_provider_check")
|
|
require.Contains(t, sql, "CHECK (provider IN ('openai', 'anthropic', 'gemini', 'grok', 'antigravity', 'kimi', 'zhipu', 'deepseek'))")
|
|
require.Contains(t, sql, "position('kimi' IN monitor_constraint_def) = 0")
|
|
require.Contains(t, sql, "position('kimi' IN template_constraint_def) = 0")
|
|
|
|
// check_mode 三态，默认 probe。
|
|
require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS check_mode VARCHAR(32) NOT NULL DEFAULT 'probe'")
|
|
require.Contains(t, sql, "CHECK (check_mode IN ('probe', 'quota', 'quota_probe'))")
|
|
|
|
// account_id 关联账号，账号删除置空（监控保留，运行时报「账号未关联」）。
|
|
require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS account_id BIGINT REFERENCES accounts(id) ON DELETE SET NULL")
|
|
require.Contains(t, sql, "CREATE INDEX IF NOT EXISTS idx_channel_monitors_account_id ON channel_monitors(account_id)")
|
|
|
|
// 历史表配额快照列。
|
|
require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS quota JSONB")
|
|
|
|
// 公开设置默认关闭。
|
|
require.Contains(t, sql, "VALUES ('channel_monitor_show_quota', 'false')")
|
|
require.Contains(t, sql, "ON CONFLICT (key) DO NOTHING")
|
|
}
|