gopenfusion/util/util_test.go

36 lines
692 B
Go
Raw Normal View History

2023-12-02 02:13:53 +00:00
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
}