fix go test TestSkipEnqueue error, move queue.Run (#2209)

This commit is contained in:
chenpengdev 2018-03-19 19:46:54 +08:00 committed by Manuel Alejandro de Brito Fontes
parent 94deb3a01a
commit f4caa13b28

View file

@ -137,8 +137,6 @@ func TestSkipEnqueue(t *testing.T) {
atomic.StoreUint32(&sr, 0) atomic.StoreUint32(&sr, 0)
q := NewCustomTaskQueue(mockSynFn, mockKeyFn) q := NewCustomTaskQueue(mockSynFn, mockKeyFn)
stopCh := make(chan struct{}) stopCh := make(chan struct{})
// run queue
go q.Run(time.Second, stopCh)
// mock object whichi will be enqueue // mock object whichi will be enqueue
mo := mockEnqueueObj{ mo := mockEnqueueObj{
k: "testKey", k: "testKey",
@ -148,6 +146,8 @@ func TestSkipEnqueue(t *testing.T) {
q.Enqueue(mo) q.Enqueue(mo)
q.Enqueue(mo) q.Enqueue(mo)
q.Enqueue(mo) q.Enqueue(mo)
// run queue
go q.Run(time.Second, stopCh)
// wait for 'mockSynFn' // wait for 'mockSynFn'
time.Sleep(time.Millisecond * 10) time.Sleep(time.Millisecond * 10)
if atomic.LoadUint32(&sr) != 1 { if atomic.LoadUint32(&sr) != 1 {