close conns

ethanndickson · ethanndickson · commit fd3628f5c311 · 2024-06-07T03:00:55.000Z
diff --git a/coderd/prometheusmetrics/insights/metricscollector_test.go b/coderd/prometheusmetrics/insights/metricscollector_test.go
@@ -99,6 +99,13 @@ func TestCollectInsights(t *testing.T) {
 		agentClients = append(agentClients, agentAPI)
 	}
 
+	defer func() {
+		for a := range agentClients {
+			err := agentClients[a].DRPCConn().Close()
+			require.NoError(t, err)
+		}
+	}()
+
 	// Fake app stats
 	_, err = agentClients[0].UpdateStats(context.Background(), &agentproto.UpdateStatsRequest{
 		Stats: &agentproto.Stats{
diff --git a/coderd/workspaceagents_test.go b/coderd/workspaceagents_test.go
@@ -988,6 +988,10 @@ func TestWorkspaceAgent_LifecycleState(t *testing.T) {
 		ac.SetSessionToken(r.AgentToken)
 		conn, err := ac.ConnectRPC(ctx)
 		require.NoError(t, err)
+		defer func() {
+			cErr := conn.Close()
+			require.NoError(t, cErr)
+		}()
 		agentAPI := agentproto.NewDRPCAgentClient(conn)
 
 		tests := []struct {
