rogee
3e809c4fc7
抖音 profile/other 响应本就携带画像数据但被丢弃，同步链路也无回填。 - migration 1046: creator_competitor 增加 signature、total_favorited 列及非负约束 - douyin 解析层: TargetProfile 增加签名与计数，total_favorited 兼容字符串大数 - store: UpdateCompetitorProfile 改为画像回填（nil 计数保留原值） - 同步链路: 作品采集成功后拉取 profile 回填，失败仅记日志不阻断
76 lines
4.2 KiB
Go
76 lines
4.2 KiB
Go
package environment
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
)
|
|
|
|
func TestMigration043ConsolidatedSchemaShape(t *testing.T) {
|
|
databaseURL := os.Getenv("CREATORHUB_POSTGRES_TEST_URL")
|
|
if databaseURL == "" {
|
|
t.Skip("set CREATORHUB_POSTGRES_TEST_URL")
|
|
}
|
|
ctx := context.Background()
|
|
testURL := isolatedDatabaseURL(t, databaseURL)
|
|
store, err := Open(ctx, testURL)
|
|
if err != nil {
|
|
t.Fatalf("open store (applies all migrations incl 043): %v", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
db, err := sql.Open("pgx", testURL)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// 最终 19 张表
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.tables WHERE table_schema = current_schema()
|
|
AND table_name IN ('social_account','gateway','network_exit','browser_env','audit_event',
|
|
'creator_competitor','creator_competitor_share_job','creator_work','creator_work_metric','creator_work_cover',
|
|
'creator_comment','creator_comment_rule_result','creator_lead_rule','creator_account_profile','creator_account_password',
|
|
'creator_account_metric','creator_collection_checkpoint','creator_settings','schema_migration')`, 19)
|
|
// 已删表不再存在
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.tables WHERE table_schema = current_schema()
|
|
AND table_name IN ('runtime_instance','environment_binding','credential_reference','content_draft','confirmation',
|
|
'operation_task','execution_attempt','runtime_use_lease','creator_event','creator_strategy','creator_operation',
|
|
'creator_conversation','creator_message','creator_cooldown','creator_listener_state','creator_listener_boundary',
|
|
'creator_relation','creator_material_job','creator_event_strategy_trace','creator_work_source','creator_metric_plan',
|
|
'creator_source_sync_lease','creator_schema_migration')`, 0)
|
|
// browser_env 合并列
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'browser_env' AND column_name IN ('account_id','exit_id','gateway_id','runtime_cleanup_pending',
|
|
'runtime_cleanup_binding_version','runtime_cleanup_runtime_id','runtime_cleanup_network_id',
|
|
'runtime_id','runtime_lease_until','runtime_network_id','runtime_node_id','updated_at')`, 12)
|
|
// social_account 凭据列
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'social_account' AND column_name IN ('credential_provider','credential_key')`, 2)
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'social_account' AND column_name = 'credential_reference_id'`, 0)
|
|
// creator_work 调度列
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'creator_work' AND column_name IN ('metric_plan_published_at','metric_monitoring_end','metric_plan_next_at',
|
|
'metric_plan_interval_seconds','metric_plan_multiplier','metric_plan_max_interval_seconds','metric_plan_point_index','metric_plan_stopped')`, 8)
|
|
// checkpoint 支持 sync kind
|
|
var kindCheck string
|
|
if err := db.QueryRow(`SELECT pg_get_constraintdef(oid) FROM pg_constraint WHERE conname = 'creator_collection_checkpoint_collection_kind_check' AND conrelid = 'creator_collection_checkpoint'::regclass`).Scan(&kindCheck); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(kindCheck, "'sync'") {
|
|
t.Fatalf("checkpoint kind check missing sync: %s", kindCheck)
|
|
}
|
|
// creator_settings 转写列已删
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'creator_settings' AND column_name LIKE 'transcription%'`, 0)
|
|
// audit_event 任务列已删
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM information_schema.columns WHERE table_schema = current_schema()
|
|
AND table_name = 'audit_event' AND column_name IN ('confirmation_id','confirmation_version','attempt_id','task_id','runtime_instance_id')`, 0)
|
|
// 统一登记表：1-38（除 36）、1017-1045、43 全部登记
|
|
assertDatabaseCount(t, db, `SELECT count(*) FROM schema_migration`, 53)
|
|
}
|