Create migration set factory

krab/action_migrate_up_hooks_test.go

@@ -19,24 +19,12 @@ func TestActionMigrateUpHooks(t *testing.T) {
 		db.MustExec("CREATE SCHEMA tenants")
 		defer db.MustExec("DROP SCHEMA tenants CASCADE")
 
-		set := &MigrationSet{
-			Hooks: &Hooks{
-				Before: "SET search_path TO tenants",
-			},
-			RefName: "tenants",
-			Migrations: []*Migration{
-				{
-					Version: "v1",
-					Up: MigrationUp{
-						SQL: `CREATE TABLE animals(name VARCHAR)`,
-					},
-					Down: MigrationDown{
-						SQL: `DROP TABLE animals`,
-					},
-				},
-			},
-		}
-		set.InitDefaults()
+		set := createMigrationSet("tenants",
+			"v1",
+			`CREATE TABLE animals(name VARCHAR)`,
+			`DROP TABLE animals`,
+		)
+		set.Hooks = &Hooks{Before: "SET search_path TO tenants"}
 
 		err := (&ActionMigrateUp{Set: set}).Do(ctx, db, cli.NullUI())
 		assert.NoError(err, "First migration should pass")

krab/factories_test.go

@@ -0,0 +1,22 @@
+package krab
+
+func createMigrationSet(
+	refName string,
+	migrationData ...string,
+) *MigrationSet {
+	migrations := make([]*Migration, len(migrationData)/3)
+
+	for i := 0; i < len(migrationData); i += 3 {
+		migrations[i/3] = &Migration{
+			Version: migrationData[i],
+			Up: MigrationUp{
+				SQL: migrationData[i+1],
+			},
+			Down: MigrationDown{
+				SQL: migrationData[i+2],
+			},
+		}
+	}
+
+	return &MigrationSet{RefName: refName, Migrations: migrations, Hooks: &Hooks{}}
+}