Fix coordination test

kylecarbs · kylecarbs · commit 8dfc261c7bb6 · 2022-10-13T19:26:36.000Z
diff --git a/coderd/coderd.go b/coderd/coderd.go
@@ -558,7 +558,10 @@ func (api *API) Close() error {
 	api.websocketWaitMutex.Unlock()
 
 	api.metricsCache.Close()
-
+	coordinator := api.TailnetCoordinator.Load()
+	if coordinator != nil {
+		_ = (*coordinator).Close()
+	}
 	return api.workspaceAgentCache.Close()
 }
 
diff --git a/enterprise/coderd/workspaceagents_test.go b/enterprise/coderd/workspaceagents_test.go
@@ -89,9 +89,9 @@ func setupWorkspaceAgent(t *testing.T, client *codersdk.Client, user codersdk.Cr
 		CoordinatorDialer: agentClient.ListenWorkspaceAgentTailnet,
 		Logger:            slogtest.Make(t, nil).Named("agent"),
 	})
-	defer func() {
+	t.Cleanup(func() {
 		_ = agentCloser.Close()
-	}()
+	})
 	resources := coderdtest.AwaitWorkspaceAgents(t, client, workspace.ID)
 	return resources[0].Agents[0].ID
 }
