优化结构

internal/data/migrations_test.go

@@ -3,18 +3,29 @@ package data
 import (
 	"testing"
 
+	integrationmodule "kra/internal/modules/integration"
+	paymentmodule "kra/internal/modules/payment"
+	systemmodule "kra/internal/modules/system"
+	taskmodule "kra/internal/modules/task"
 	"kra/pkg/database/migration"
+	platformmodule "kra/pkg/module"
 )
 
+func testCatalog() platformmodule.Catalog {
+	return platformmodule.Catalog{Definitions: []platformmodule.Definition{
+		systemmodule.Definition(), integrationmodule.Definition(), taskmodule.Definition(), paymentmodule.Definition(),
+	}}
+}
+
 func TestMigrateAllRunsModuleSchemasWithoutBootstrapSeed(t *testing.T) {
 	db, err := openWithDriver("sqlite", "file:"+t.Name()+"?mode=memory&cache=shared")
 	if err != nil {
 		t.Fatal(err)
 	}
-	if err = migrateAll(db); err != nil {
+	if err = migrateAll(db, testCatalog()); err != nil {
 		t.Fatal(err)
 	}
-	if err = migrateAll(db); err != nil {
+	if err = migrateAll(db, testCatalog()); err != nil {
 		t.Fatalf("second migration run: %v", err)
 	}
 	for _, table := range []string{"sys_integration_configs", "sys_users", "sys_base_menus", "pay_orders"} {
@@ -26,10 +37,10 @@ func TestMigrateAllRunsModuleSchemasWithoutBootstrapSeed(t *testing.T) {
 	if err = db.Table(migration.TableName).Count(&versions).Error; err != nil {
 		t.Fatal(err)
 	}
-	if versions != 7 {
-		t.Fatalf("migration versions = %d, want 7", versions)
+	if versions != 8 {
+		t.Fatalf("migration versions = %d, want 8", versions)
 	}
-	var communicationRows []integrationConfigPO
+	var communicationRows []dataintegration.ConfigPO
 	if err = db.Where("kind IN ?", []string{"mq", "websocket"}).Order("kind, provider").Find(&communicationRows).Error; err != nil {
 		t.Fatal(err)
 	}