mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
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:
@@ -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)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user