Fix data race on test.

segmentio · Jul 2, 2024 · 35de2e9 · 35de2e9
1 parent f8eddfc
commit 35de2e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/reader_test.go b/reader_test.go
@@ -923,6 +923,8 @@ func TestAssignmentListener(t *testing.T) {
 	defer r.Close()
 
 	assert.Eventually(t, func() bool {
+		lock.Lock()
+		defer lock.Unlock()
 		return reflect.DeepEqual(assignments, [][]GroupMemberTopic{
 			{
 				GroupMemberTopic{