Fix test panic in docker driver test

hashicorp · May 7, 2021 · 1b22964 · 1b22964
1 parent 334de88
commit 1b22964
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/drivers/docker/driver_test.go b/drivers/docker/driver_test.go
@@ -3032,11 +3032,10 @@ func TestDockerDriver_StopSignal(t *testing.T) {
 			require.NoError(t, err)
 			require.NoError(t, d.WaitUntilStarted(task.ID, 5*time.Second))
 
+			stopErr := make(chan error)
 			go func() {
 				err := d.StopTask(task.ID, 1*time.Second, c.jobKillSignal)
-				if err != nil {
-					t.Errorf("stop task failed: %v", err)
-				}
+				stopErr <- err
 			}()
 
 			timeout := time.After(10 * time.Second)
@@ -3054,6 +3053,9 @@ func TestDockerDriver_StopSignal(t *testing.T) {
 							break WAIT
 						}
 					}
+				case err := <-stopErr:
+					t.Errorf("stop task failed: %v", err)
+					break WAIT
 				case <-timeout:
 					// timeout waiting for signals
 					require.Equal(t, c.expectedSignals, receivedSignals, "timed out waiting for expected signals")