From 8ef4164209f77390d62841f94ab1388adb757697 Mon Sep 17 00:00:00 2001 From: Kevin Wan Date: Sun, 4 Jun 2023 23:20:58 +0800 Subject: [PATCH] chore: make test stable (#3317) --- core/queue/queue_test.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/core/queue/queue_test.go b/core/queue/queue_test.go index d11c228d..63ed78ef 100644 --- a/core/queue/queue_test.go +++ b/core/queue/queue_test.go @@ -2,6 +2,7 @@ package queue import ( "errors" + "math" "sync" "sync/atomic" "testing" @@ -39,7 +40,7 @@ func TestQueue(t *testing.T) { } func TestQueue_Broadcast(t *testing.T) { - producer := newMockedProducer(rounds) + producer := newMockedProducer(math.MaxInt32) consumer := newMockedConsumer() consumer.wait.Add(consumers) q := NewQueue(func() (Producer, error) { @@ -51,14 +52,14 @@ func TestQueue_Broadcast(t *testing.T) { q.SetName("mockqueue") q.SetNumConsumer(consumers) q.SetNumProducer(1) - q.Broadcast("message") go func() { - producer.wait.Wait() + time.Sleep(time.Millisecond * 100) q.Stop() }() - q.Start() + go q.Start() + time.Sleep(time.Millisecond * 50) + q.Broadcast("message") consumer.wait.Wait() - assert.Equal(t, int32(rounds), atomic.LoadInt32(&consumer.count)) assert.Equal(t, int32(consumers), atomic.LoadInt32(&consumer.events)) }