feat: fix TestNewStreamCopy's data race

cloudwego · Dec 13, 2024 · 36cf3d5 · 36cf3d5
1 parent 5828188
commit 36cf3d5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/schema/stream_test.go b/schema/stream_test.go
@@ -321,7 +321,10 @@ func TestNewStreamCopy(t *testing.T) {
 
 		wgEven.Wait()
 		memo := copies[0].csr.parent.mem
-		assert.Equal(t, m/2, memo.closedNum)
+		memo.mu.Lock()
+		closedNum := memo.closedNum
+		memo.mu.Unlock()
+		assert.Equal(t, m/2, closedNum)
 
 		wg.Wait()
 		assert.Equal(t, m, memo.closedNum)