Fix TestMain temp directory cleanup

ingydotnet · ingydotnet · commit 866cff5db5d8 · 2026-01-20T13:58:44.000-08:00
diff --git a/cmd/go-yaml/cli_test.go b/cmd/go-yaml/cli_test.go
@@ -23,19 +23,23 @@ func TestMain(m *testing.M) {
 		fmt.Fprintf(os.Stderr, "Failed to create temp dir: %v\n", err)
 		os.Exit(1)
 	}
-	defer os.RemoveAll(tmpDir)
 
 	// Build the binary once
 	testBinary = filepath.Join(tmpDir, "go-yaml")
 	cmd := exec.Command("go", "build", "-o", testBinary, ".")
 	cmd.Stderr = os.Stderr
 	if err := cmd.Run(); err != nil {
 		fmt.Fprintf(os.Stderr, "Failed to build test binary: %v\n", err)
+		os.RemoveAll(tmpDir)
 		os.Exit(1)
 	}
 
 	// Run tests
-	os.Exit(m.Run())
+	code := m.Run()
+
+	// Cleanup before exit (defer won't run with os.Exit)
+	os.RemoveAll(tmpDir)
+	os.Exit(code)
 }
 
 // TestCase represents a single test case from a test file
