Honor RemoveAll defer

hashicorp · Jul 11, 2020 · 014f135 · 014f135
1 parent fd57643
commit 014f135
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/tfexec/terraform_test.go b/tfexec/terraform_test.go
@@ -19,20 +19,20 @@ const testTerraformStateFileName = "terraform.tfstate"
 var tfPath string
 
 func TestMain(m *testing.M) {
-	var err error
-	td, err := ioutil.TempDir("", "tfinstall")
-	if err != nil {
-		panic(err)
-	}
-	defer os.RemoveAll(td)
-
-	tfPath, err = tfinstall.Find(tfinstall.LookPath(), tfinstall.LatestVersion(td, true))
-	if err != nil {
-		panic(err)
-	}
-	exitCode := m.Run()
-	os.Exit(exitCode)
-
+	os.Exit(func() int {
+		var err error
+		td, err := ioutil.TempDir("", "tfinstall")
+		if err != nil {
+			panic(err)
+		}
+		defer os.RemoveAll(td)
+
+		tfPath, err = tfinstall.Find(tfinstall.LookPath(), tfinstall.LatestVersion(td, true))
+		if err != nil {
+			panic(err)
+		}
+		return m.Run()
+	}())
 }
 
 func TestCheckpointDisablePropagation(t *testing.T) {