-

databricks · Jan 29, 2025 · ca17487 · ca17487
1 parent 3e7acdb
commit ca17487
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/acceptance/acceptance_test.go b/acceptance/acceptance_test.go
@@ -211,6 +211,8 @@ func runTest(t *testing.T, dir, coverDir string, repls testdiff.ReplacementsCont
 	// specifies a custom server stubs.
 	if len(config.Server) > 0 {
 		server := testserver.New(t)
+		t.Cleanup(server.Close)
+
 		for _, stub := range config.Server {
 			require.NotEmpty(t, stub.Pattern)
 			require.NotEmpty(t, stub.Response.Body)
@@ -220,9 +222,6 @@ func runTest(t *testing.T, dir, coverDir string, repls testdiff.ReplacementsCont
 			})
 		}
 		cmdEnv = setEnv(cmdEnv, "DATABRICKS_HOST", server.URL)
-		t.Cleanup(func() {
-			server.Close()
-		})
 	}
 
 	repls.SetPathWithParents(tmpDir, "$TMPDIR")