From f7f33e10cc18b4a20542d0208aeaccb91ad64f99 Mon Sep 17 00:00:00 2001
From: nisdas <nishdas93@gmail.com>
Date: Thu, 10 Sep 2020 17:19:19 +0800
Subject: [PATCH] satisfy race detector

---
 gossipsub_test.go | 37 +++++++++++++++++++++++++------------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/gossipsub_test.go b/gossipsub_test.go
index 458589b..11f9ed3 100644
--- a/gossipsub_test.go
+++ b/gossipsub_test.go
@@ -1584,12 +1584,21 @@ func TestGossipsubMultipleGraftTopics(t *testing.T) {
 	firstPeer := hosts[0].ID()
 	secondPeer := hosts[1].ID()
 
+	p2Sub := psubs[1]
 	p1Router := psubs[0].rt.(*GossipSubRouter)
 	p2Router := psubs[1].rt.(*GossipSubRouter)
-	// Add topics to second peer
-	p2Router.mesh[firstTopic] = map[peer.ID]struct{}{}
-	p2Router.mesh[secondTopic] = map[peer.ID]struct{}{}
-	p2Router.mesh[thirdTopic] = map[peer.ID]struct{}{}
+
+	finChan := make(chan struct{})
+
+	p2Sub.eval <- func() {
+		// Add topics to second peer
+		p2Router.mesh[firstTopic] = map[peer.ID]struct{}{}
+		p2Router.mesh[secondTopic] = map[peer.ID]struct{}{}
+		p2Router.mesh[thirdTopic] = map[peer.ID]struct{}{}
+
+		finChan <- struct{}{}
+	}
+	<-finChan
 
 	// Send multiple GRAFT messages to second peer from
 	// 1st peer
@@ -1599,15 +1608,19 @@ func TestGossipsubMultipleGraftTopics(t *testing.T) {
 
 	time.Sleep(time.Second * 1)
 
-	if _, ok := p2Router.mesh[firstTopic][firstPeer]; !ok {
-		t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", firstTopic)
-	}
-	if _, ok := p2Router.mesh[secondTopic][firstPeer]; !ok {
-		t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", secondTopic)
-	}
-	if _, ok := p2Router.mesh[thirdTopic][firstPeer]; !ok {
-		t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", thirdTopic)
+	p2Sub.eval <- func() {
+		if _, ok := p2Router.mesh[firstTopic][firstPeer]; !ok {
+			t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", firstTopic)
+		}
+		if _, ok := p2Router.mesh[secondTopic][firstPeer]; !ok {
+			t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", secondTopic)
+		}
+		if _, ok := p2Router.mesh[thirdTopic][firstPeer]; !ok {
+			t.Errorf("First peer wasnt added to mesh of the second peer for the topic %s", thirdTopic)
+		}
+		finChan <- struct{}{}
 	}
+	<-finChan
 }
 
 func TestGossipsubOpportunisticGrafting(t *testing.T) {