mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user