fix: require client pipe to be closed in provisionerd test (#6188)

https://github.com/coder/coder/actions/runs/4165019548/jobs/7207442687
This commit is contained in:
Kyle Carberry
2023-02-14 00:46:05 +00:00
committed by GitHub
parent 733f58c76d
commit 2d0a69ba47
+7 -1
View File
@@ -1090,10 +1090,16 @@ 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()
_ = serverPipe.Close()
<-closed
})
return proto.NewDRPCProvisionerDaemonClient(clientPipe)
}