fix: Add sync.Once to prevent double close in test (#1124)

https://github.com/coder/coder/runs/6151451291?check_suite_focus=true
This commit is contained in:
Kyle Carberry
2022-04-24 23:06:18 -05:00
committed by GitHub
parent abc13c5a92
commit 68f67c54b6
+4 -1
View File
@@ -310,12 +310,15 @@ func TestProvisionerd(t *testing.T) {
didLog atomic.Bool
didAcquireJob atomic.Bool
)
var complete sync.Once
completeChan := make(chan struct{})
closer := createProvisionerd(t, func(ctx context.Context) (proto.DRPCProvisionerDaemonClient, error) {
return createProvisionerDaemonClient(t, provisionerDaemonTestServer{
acquireJob: func(ctx context.Context, _ *proto.Empty) (*proto.AcquiredJob, error) {
if didAcquireJob.Load() {
close(completeChan)
complete.Do(func() {
close(completeChan)
})
return &proto.AcquiredJob{}, nil
}
didAcquireJob.Store(true)