TestGossipSubDiscoveryAfterBootstrap thread safety

libp2p · Jun 11, 2019 · e93015a · e93015a
1 parent 895db49
commit e93015a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/discovery_test.go b/discovery_test.go
@@ -231,7 +231,7 @@ func TestGossipSubDiscoveryAfterBootstrap(t *testing.T) {
 	time.Sleep(time.Second * 1)
 
 	for _, ps := range psubs {
-		pn := len(ps.peers)
+		pn := len(ps.ListPeers("foobar"))
 		if pn != partitionSize-1 {
 			t.Fatal("partitions not properly formed")
 		}
@@ -246,7 +246,7 @@ func TestGossipSubDiscoveryAfterBootstrap(t *testing.T) {
 
 	// verify the network is fully connected
 	for _, ps := range psubs {
-		pn := len(ps.peers)
+		pn := len(ps.ListPeers("foobar"))
 		if pn != numHosts-1 {
 			t.Fatal("network not fully connected")
 		}