test rollback doesn't fail

xataio · Jan 15, 2024 · 84ce30e · 84ce30e
1 parent 1beeeaf
commit 84ce30e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/roll/execute_test.go b/pkg/roll/execute_test.go
@@ -63,6 +63,14 @@ func TestDisabledSchemaManagement(t *testing.T) {
 			t.Errorf("Expected schema %q to not exist", version)
 		}
 
+		if err := mig.Rollback(ctx); err != nil {
+			t.Fatalf("Failed to rollback migration: %v", err)
+		}
+
+		if err := mig.Start(ctx, &migrations.Migration{Name: version, Operations: migrations.Operations{createTableOp("table1")}}); err != nil {
+			t.Fatalf("Failed to start migration again: %v", err)
+		}
+
 		// complete the migration, check that the schema still doesn't exist
 		if err := mig.Complete(ctx); err != nil {
 			t.Fatalf("Failed to complete migration: %v", err)