66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package waiterr_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/nalgeon/be"
|
|
|
|
"codeberg.org/danjones000/waiterr"
|
|
)
|
|
|
|
func TestGo(t *testing.T) {
|
|
we := new(waiterr.WaitErr)
|
|
err := errors.New("uh-oh")
|
|
var run bool
|
|
we.Go(func() error {
|
|
run = true
|
|
return err
|
|
})
|
|
|
|
be.Err(t, we.Wait(), err)
|
|
be.True(t, run)
|
|
}
|
|
|
|
func TestWait(t *testing.T) {
|
|
we := new(waiterr.WaitErr)
|
|
er1 := errors.New("uh-oh")
|
|
er2 := errors.New("oops")
|
|
we.Go(func() error { return er1 })
|
|
we.Go(func() error { return nil })
|
|
we.Go(func() error { return er2 })
|
|
|
|
err := we.Wait()
|
|
be.Err(t, err, er1, er2)
|
|
if ers, ok := err.(interface{ Unwrap() []error }); ok {
|
|
all := ers.Unwrap()
|
|
be.Equal(t, len(all), 2)
|
|
be.True(t, all[0] == er1 || all[0] == er2)
|
|
be.True(t, all[1] == er2 || all[1] == er1)
|
|
} else {
|
|
t.Fatal("Returned error should have Unwrap method")
|
|
}
|
|
}
|
|
|
|
func TestWaitForError(t *testing.T) {
|
|
we := new(waiterr.WaitErr)
|
|
er1 := errors.New("uh-oh")
|
|
er2 := errors.New("oops")
|
|
we.Go(func() error { return nil })
|
|
we.Go(func() error { return er1 })
|
|
we.Go(func() error { return er2 })
|
|
|
|
err := we.WaitForError()
|
|
// Due to how goroutines run, it is possible that either of those return first. This is an acceptable limitation
|
|
be.True(t, err == er1 || err == er2)
|
|
}
|
|
|
|
func TestWaitForErrorNoErr(t *testing.T) {
|
|
we := new(waiterr.WaitErr)
|
|
we.Go(func() error { return nil })
|
|
we.Go(func() error { return nil })
|
|
we.Go(func() error { return nil })
|
|
|
|
err := we.WaitForError()
|
|
be.Err(t, err, nil)
|
|
}
|