From 35de2e9581dbc69c811b5ad8883293acfd50fdf4 Mon Sep 17 00:00:00 2001
From: nachogiljaldo <jose.jaldo@grafana.com>
Date: Wed, 3 Jul 2024 00:08:18 +0200
Subject: [PATCH] Fix data race on test.

---
 reader_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/reader_test.go b/reader_test.go
index 47893907..6e97026a 100644
--- 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{