Add tests for inferred ALTER SCHEMA migrations

xataio · Jun 24, 2024 · 6480058 · 6480058
1 parent e7c5ac6
commit 6480058
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pkg/state/state_test.go b/pkg/state/state_test.go
@@ -201,7 +201,8 @@ func TestInferredMigration(t *testing.T) {
 				name: "create/drop schema",
 				sqlStmts: []string{
 					"CREATE SCHEMA foo",
-					"DROP SCHEMA foo",
+					"ALTER SCHEMA foo RENAME TO bar",
+					"DROP SCHEMA bar",
 				},
 				wantMigrations: []migrations.Migration{
 					{
@@ -211,7 +212,12 @@ func TestInferredMigration(t *testing.T) {
 					},
 					{
 						Operations: migrations.Operations{
-							&migrations.OpRawSQL{Up: "DROP SCHEMA foo"},
+							&migrations.OpRawSQL{Up: "ALTER SCHEMA foo RENAME TO bar"},
+						},
+					},
+					{
+						Operations: migrations.Operations{
+							&migrations.OpRawSQL{Up: "DROP SCHEMA bar"},
 						},
 					},
 				},