From f7b5fc5f64fb27c78b6387f19dfb2c7aa560b733 Mon Sep 17 00:00:00 2001 From: Robin Joseph Date: Thu, 17 Oct 2024 11:44:33 -0500 Subject: [PATCH] add t.Run to migrations_test --- migrations_test.go | 57 ++++++++++++++++++++++++++-------------------- 1 file changed, 32 insertions(+), 25 deletions(-) diff --git a/migrations_test.go b/migrations_test.go index 6b54bde..a21979d 100644 --- a/migrations_test.go +++ b/migrations_test.go @@ -40,36 +40,43 @@ func TestRunWithOptions(t *testing.T) { Database: os.Getenv("TEST_DATABASE_NAME"), }) db.AddQueryHook(logQueryHook{}) - dropMigrationTables(t, db) - err := RunWithOptions(db, tmp, []string{"cmd", "migrate"}, RunOptions{}) - assert.Nil(t, err) - assertTable(t, db, "migrations", true) - assertTable(t, db, "migration_lock", true) - assertTable(t, db, "custom_migrations", false) - assertTable(t, db, "custom_migration_lock", false) + t.Run("default", func(tt *testing.T) { + dropMigrationTables(t, db) + + err := RunWithOptions(db, tmp, []string{"cmd", "migrate"}, RunOptions{}) + assert.Nil(t, err) + assertTable(t, db, "migrations", true) + assertTable(t, db, "migration_lock", true) + assertTable(t, db, "custom_migrations", false) + assertTable(t, db, "custom_migration_lock", false) + }) - dropMigrationTables(t, db) + t.Run("custom tables - migrate", func(tt *testing.T) { + dropMigrationTables(t, db) - err = RunWithOptions(db, tmp, []string{"cmd", "migrate"}, RunOptions{ - MigrationsTableName: "custom_migrations", - MigrationLockTableName: "custom_migration_lock", + err := RunWithOptions(db, tmp, []string{"cmd", "migrate"}, RunOptions{ + MigrationsTableName: "custom_migrations", + MigrationLockTableName: "custom_migration_lock", + }) + assert.Nil(t, err) + assertTable(t, db, "custom_migrations", true) + assertTable(t, db, "custom_migration_lock", true) + assertTable(t, db, "migrations", false) + assertTable(t, db, "migration_lock", false) }) - assert.Nil(t, err) - assertTable(t, db, "custom_migrations", true) - assertTable(t, db, "custom_migration_lock", true) - assertTable(t, db, "migrations", false) - assertTable(t, db, "migration_lock", false) - dropMigrationTables(t, db) + t.Run("custom tables - rollback", func(tt *testing.T) { + dropMigrationTables(t, db) - err = RunWithOptions(db, tmp, []string{"cmd", "rollback"}, RunOptions{ - MigrationsTableName: "custom_migrations", - MigrationLockTableName: "custom_migration_lock", + err := RunWithOptions(db, tmp, []string{"cmd", "rollback"}, RunOptions{ + MigrationsTableName: "custom_migrations", + MigrationLockTableName: "custom_migration_lock", + }) + assert.Nil(t, err) + assertTable(t, db, "custom_migrations", true) + assertTable(t, db, "custom_migration_lock", true) + assertTable(t, db, "migrations", false) + assertTable(t, db, "migration_lock", false) }) - assert.Nil(t, err) - assertTable(t, db, "custom_migrations", true) - assertTable(t, db, "custom_migration_lock", true) - assertTable(t, db, "migrations", false) - assertTable(t, db, "migration_lock", false) }