Fix possible concurrent map write in test

ipni · Dec 22, 2023 · 208037b · 208037b
1 parent fe44e6c
commit 208037b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dagsync/subscriber_test.go b/dagsync/subscriber_test.go
@@ -617,8 +617,11 @@ func TestMaxAsyncSyncs(t *testing.T) {
 	dstHost := test.MkTestHost(t)
 	dstLnkS, blocked := test.MkBlockedLinkSystem(dstStore)
 	blocksSeenByHook := make(map[cid.Cid]struct{})
+	bhMutex := new(sync.Mutex)
 	blockHook := func(p peer.ID, c cid.Cid, _ dagsync.SegmentSyncActions) {
+		bhMutex.Lock()
 		blocksSeenByHook[c] = struct{}{}
+		bhMutex.Unlock()
 	}
 
 	sub, err := dagsync.NewSubscriber(dstHost, dstStore, dstLnkS, testTopic,