fix race in tests

gartnera · gartnera · commit 184b601a122f · 2024-11-15T08:33:12.000-08:00
diff --git a/p2p/discovery_test.go b/p2p/discovery_test.go
@@ -84,8 +84,10 @@ func TestDiscovery(t *testing.T) {
 	assert.Equal(t, 4, len(comm3.host.Peerstore().Peers()))
 	assert.Equal(t, 4, len(comm4.host.Peerstore().Peers()))
 
+	comm.discovery.mu.Lock()
 	assert.Equal(t, 3, len(comm.discovery.knownPeers))
 	for peer, knownPeers := range comm.discovery.knownPeers {
 		assert.LessOrEqual(t, len(knownPeers.Addrs), 4, "%s has more than 4 addresses (%d)?", peer.String(), len(knownPeers.Addrs))
 	}
+	comm.discovery.mu.Unlock()
 }

Original file line number	Diff line number	Diff line change
`@@ -84,8 +84,10 @@ func TestDiscovery(t *testing.T) {`
`84`	`84`	`assert.Equal(t, 4, len(comm3.host.Peerstore().Peers()))`
`85`	`85`	`assert.Equal(t, 4, len(comm4.host.Peerstore().Peers()))`
`86`	`86`
	`87`	`+ comm.discovery.mu.Lock()`
`87`	`88`	`assert.Equal(t, 3, len(comm.discovery.knownPeers))`
`88`	`89`	`for peer, knownPeers := range comm.discovery.knownPeers {`
`89`	`90`	`assert.LessOrEqual(t, len(knownPeers.Addrs), 4, "%s has more than 4 addresses (%d)?", peer.String(), len(knownPeers.Addrs))`
`90`	`91`	`}`
	`92`	`+ comm.discovery.mu.Unlock()`
`91`	`93`	`}`