Copy TEST_TMPDIR env variable to TMPDIR before running tests

cockroachdb · Oct 9, 2024 · 6b9e1e5 · 6b9e1e5
1 parent 74dae57
commit 6b9e1e5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/go/tools/builders/generate_test_main.go b/go/tools/builders/generate_test_main.go
@@ -165,6 +165,12 @@ func testsInShard() []testing.InternalTest {
 }
 
 func main() {
+	// NOTE(ricky): Bazel sets the TEST_TMPDIR env variable, but Cockroach
+	// tests generally consult TMPDIR.
+	err := os.Setenv("TMPDIR", os.Getenv("TEST_TMPDIR"))
+	if err != nil {
+		panic(err)
+	}
 	if bzltestutil.ShouldWrap() {
 		err := bzltestutil.Wrap("{{.Pkgname}}")
 		exitCode := 0