fix: require client pipe to be closed in provisionerd test

kylecarbs · kylecarbs · commit 00671bf48f24 · 2023-02-13T15:47:38.000Z
https://github.com/coder/coder/actions/runs/4165019548/jobs/7207442687
diff --git a/provisionerd/provisionerd_test.go b/provisionerd/provisionerd_test.go
@@ -1090,10 +1090,15 @@ func createProvisionerDaemonClient(t *testing.T, server provisionerDaemonTestSer
 	require.NoError(t, err)
 	srv := drpcserver.New(mux)
 	ctx, cancelFunc := context.WithCancel(context.Background())
-	t.Cleanup(cancelFunc)
+	closed := make(chan struct{})
 	go func() {
+		defer close(closed)
 		_ = srv.Serve(ctx, serverPipe)
 	}()
+	t.Cleanup(func() {
+		cancelFunc()
+		<-closed
+	})
 	return proto.NewDRPCProvisionerDaemonClient(clientPipe)
 }
 
