mirror of
https://github.com/CPunch/gopenfusion.git
synced 2024-11-12 19:20:06 +00:00
util: added test
This commit is contained in:
parent
3445b852fd
commit
e33b7c0556
35
util/util_test.go
Normal file
35
util/util_test.go
Normal 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
|
||||
}
|
Loading…
Reference in New Issue
Block a user