feat: fix fanout test data race

apache · Oct 22, 2022 · 1373cb0 · 1373cb0
1 parent d2b0e7c
commit 1373cb0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/util/fanout/fanout_test.go b/pkg/util/fanout/fanout_test.go
@@ -19,19 +19,27 @@ package fanout
 
 import (
 	"context"
+	"sync"
 	"testing"
 	"time"
 )
 
 func TestFanout_Do(t *testing.T) {
 	ca := New("cache", WithWorker(1), WithBuffer(1024))
 	var run bool
+	var mtx sync.Mutex
+
 	ca.Do(context.Background(), func(c context.Context) {
+		mtx.Lock()
 		run = true
+		mtx.Unlock()
 		panic("error")
 	})
+
 	time.Sleep(time.Millisecond * 50)
 	t.Log("not panic")
+	mtx.Lock()
+	defer mtx.Unlock()
 	if !run {
 		t.Fatal("expect run be true")
 	}