fix: ensure embedded postgres cleanup in TestMain

cmd/apply/apply_integration_test.go

@@ -30,11 +30,7 @@ func TestMain(m *testing.M) {
 	sharedEmbeddedPG = testutil.SetupPostgres(nil)
 	defer sharedEmbeddedPG.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // TestApplyCommand_TransactionRollback verifies that the apply command uses proper

cmd/migrate_integration_test.go

@@ -37,11 +37,7 @@ func TestMain(m *testing.M) {
 	sharedEmbeddedPG = testutil.SetupPostgres(nil)
 	defer sharedEmbeddedPG.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // TestPlanAndApply tests the complete CLI (plan and apply) workflow using test cases

internal/diff/diff_test.go

@@ -19,11 +19,7 @@ func TestMain(m *testing.M) {
 	sharedTestPostgres = testutil.SetupPostgres(nil)
 	defer sharedTestPostgres.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // buildSQLFromSteps builds a SQL string from collected plan diffs

internal/plan/plan_test.go

@@ -26,11 +26,7 @@ func TestMain(m *testing.M) {
 	sharedTestPostgres = testutil.SetupPostgres(nil)
 	defer sharedTestPostgres.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // discoverTestDataVersions discovers available test data versions in the testdata directory