fix: data race in test

Signed-off-by: gfanton <8671905+gfanton@users.noreply.github.com>
berty · Mar 1, 2022 · 40ddaff · 40ddaff
1 parent 3c890ed
commit 40ddaff
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/replicate_automatically_test.go b/tests/replicate_automatically_test.go
@@ -237,6 +237,7 @@ func TestReplicateAutomatically(t *testing.T) {
 
 		infinity := -1
 		go func() {
+			defer cancel()
 			for event := range sub2.Out() {
 				switch e := event.(type) {
 				case stores.EventReplicateProgress:
@@ -264,7 +265,7 @@ func TestReplicateAutomatically(t *testing.T) {
 						require.Equal(t, string(result1[i].GetValue()), string(result2[i].GetValue()))
 					}
 
-					cancel()
+					return
 				}
 			}
 		}()
@@ -275,6 +276,7 @@ func TestReplicateAutomatically(t *testing.T) {
 		}
 
 		<-subCtx.Done()
+
 		require.True(t, hasAllResults)
 	})
 }