Fix test panic in docker driver test

This commit is contained in:
Isabel Suchanek 2021-05-07 10:01:00 -07:00 committed by Isabel Suchanek
parent 3e4a3ff8eb
commit b5a2f48c78
1 changed files with 5 additions and 3 deletions

View File

@ -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")