fix: fix test leaking file (#114)

coder · Nov 27, 2024 · ed4baf8 · ed4baf8
1 parent b6eb32d
commit ed4baf8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/integration/docker_test.go b/integration/docker_test.go
@@ -425,6 +425,9 @@ func TestDocker(t *testing.T) {
 			HostKeyPath:  regKeyPath,
 			Image:        integrationtest.UbuntuImage,
 			TLSPort:      strconv.Itoa(registryAddr.Port),
+			PasswordDir:  dir,
+			Username:     "user",
+			Password:     "password",
 		})
 
 		envs := []string{

diff --git a/integration/integrationtest/os.go b/integration/integrationtest/os.go
@@ -17,6 +17,12 @@ func TmpDir(t *testing.T) string {
 	tmpdir, err := os.MkdirTemp("", strings.ReplaceAll(t.Name(), "/", "_"))
 	require.NoError(t, err)
 	t.Logf("using tmpdir %s", tmpdir)
+	t.Cleanup(func() {
+		if !t.Failed() {
+			// Could be useful in case of test failure.
+			_ = os.RemoveAll(tmpdir)
+		}
+	})
 	return tmpdir
 }