util: added test

This commit is contained in:
CPunch 2023-12-01 20:13:53 -06:00
parent 3445b852fd
commit e33b7c0556

35
util/util_test.go Normal file
View File

@ -0,0 +1,35 @@
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
}