@@ -84,6 +84,7 @@ func TestProvisionerd(t *testing.T) {
8484t .Run ("CloseCancelsJob" ,func (t * testing.T ) {
8585t .Parallel ()
8686completeChan := make (chan struct {})
87+ var completed sync.Once
8788var closer io.Closer
8889var closerMutex sync.Mutex
8990closerMutex .Lock ()
@@ -105,7 +106,9 @@ func TestProvisionerd(t *testing.T) {
105106},
106107updateJob :noopUpdateJob ,
107108failJob :func (ctx context.Context ,job * proto.FailedJob ) (* proto.Empty ,error ) {
108- close (completeChan )
109+ completed .Do (func () {
110+ close (completeChan )
111+ })
109112return & proto.Empty {},nil
110113},
111114}),nil
@@ -432,6 +435,8 @@ func TestProvisionerd(t *testing.T) {
432435
433436t .Run ("Shutdown" ,func (t * testing.T ) {
434437t .Parallel ()
438+ var updated sync.Once
439+ var completed sync.Once
435440updateChan := make (chan struct {})
436441completeChan := make (chan struct {})
437442server := createProvisionerd (t ,func (ctx context.Context ) (proto.DRPCProvisionerDaemonClient ,error ) {
@@ -453,12 +458,16 @@ func TestProvisionerd(t *testing.T) {
453458updateJob :func (ctx context.Context ,update * proto.UpdateJobRequest ) (* proto.UpdateJobResponse ,error ) {
454459if len (update .Logs )> 0 && update .Logs [0 ].Source == proto .LogSource_PROVISIONER {
455460// Close on a log so we know when the job is in progress!
456- close (updateChan )
461+ updated .Do (func () {
462+ close (updateChan )
463+ })
457464}
458465return & proto.UpdateJobResponse {},nil
459466},
460467failJob :func (ctx context.Context ,job * proto.FailedJob ) (* proto.Empty ,error ) {
461- close (completeChan )
468+ completed .Do (func () {
469+ close (completeChan )
470+ })
462471return & proto.Empty {},nil
463472},
464473}),nil
@@ -642,6 +651,7 @@ func TestProvisionerd(t *testing.T) {
642651
643652t .Run ("ReconnectAndComplete" ,func (t * testing.T ) {
644653t .Parallel ()
654+ var completed sync.Once
645655var second atomic.Bool
646656failChan := make (chan struct {})
647657failedChan := make (chan struct {})
@@ -650,7 +660,9 @@ func TestProvisionerd(t *testing.T) {
650660client := createProvisionerDaemonClient (t ,provisionerDaemonTestServer {
651661acquireJob :func (ctx context.Context ,_ * proto.Empty ) (* proto.AcquiredJob ,error ) {
652662if second .Load () {
653- close (completeChan )
663+ completed .Do (func () {
664+ close (completeChan )
665+ })
654666return & proto.AcquiredJob {},nil
655667}
656668return & proto.AcquiredJob {