lib: improve wait.For returns. (#1403)

This commit is contained in:
Ludovic Fernandez 2021-05-14 17:37:45 +02:00 committed by GitHub
parent e3d7c85cfc
commit c53c3d08db
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions

View file

@ -1,6 +1,7 @@
package wait
import (
"errors"
"fmt"
"time"
@ -16,6 +17,9 @@ func For(msg string, timeout, interval time.Duration, f func() (bool, error)) er
for {
select {
case <-timeUp:
if lastErr == nil {
return errors.New("time limit exceeded")
}
return fmt.Errorf("time limit exceeded: last error: %w", lastErr)
default:
}

View file

@ -13,7 +13,7 @@ func TestForTimeout(t *testing.T) {
})
}()
timeout := time.After(5 * time.Second)
timeout := time.After(6 * time.Second)
select {
case <-timeout:
t.Fatal("timeout exceeded")
@ -21,5 +21,6 @@ func TestForTimeout(t *testing.T) {
if err == nil {
t.Errorf("expected timeout error; got %v", err)
}
t.Logf("%v", err)
}
}