mirror of
https://github.com/CPunch/gopenfusion.git
synced 2024-09-19 18:58:20 +00:00
36 lines
692 B
Go
36 lines
692 B
Go
package util_test
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/CPunch/gopenfusion/util"
|
|
"github.com/matryer/is"
|
|
)
|
|
|
|
func TestWaitWithTimeout(t *testing.T) {
|
|
is := is.New(t)
|
|
wg := &sync.WaitGroup{}
|
|
go func() {
|
|
time.Sleep(2 * time.Second)
|
|
wg.Done()
|
|
}()
|
|
|
|
wg.Add(1)
|
|
is.True(!util.WaitWithTimeout(wg, 1)) // timeout should occur
|
|
is.True(util.WaitWithTimeout(wg, 2)) // timeout shouldn't occur
|
|
}
|
|
|
|
func TestSelectWithTimeout(t *testing.T) {
|
|
is := is.New(t)
|
|
ch := make(chan struct{})
|
|
go func() {
|
|
time.Sleep(2 * time.Second)
|
|
close(ch)
|
|
}()
|
|
|
|
is.True(!util.SelectWithTimeout(ch, 1)) // timeout should occur
|
|
is.True(util.SelectWithTimeout(ch, 2)) // timeout shouldn't occur
|
|
}
|