Skip to content

Commit 1beb093

Browse files
fix flaky TestAutoRelay test
1 parent c126f37 commit 1beb093

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

p2p/host/autorelay/autorelay_test.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,12 @@ type mockRoutingTable struct {
4040
peers map[peer.ID]peer.AddrInfo
4141
}
4242

43+
func (t *mockRoutingTable) NumPeers() int {
44+
t.mx.Lock()
45+
defer t.mx.Unlock()
46+
return len(t.peers)
47+
}
48+
4349
type mockRouting struct {
4450
h host.Host
4551
tab *mockRoutingTable
@@ -175,6 +181,7 @@ func testAutoRelay(t *testing.T, useRelayv2 bool) {
175181
require.NoError(t, err)
176182
relayDiscovery := discovery.NewRoutingDiscovery(relayRouting)
177183
autorelay.Advertise(context.Background(), relayDiscovery)
184+
require.Eventually(t, func() bool { return mtab.NumPeers() > 0 }, time.Second, 10*time.Millisecond)
178185

179186
// the client hosts
180187
h1, err := libp2p.New(libp2p.EnableRelay())

0 commit comments

Comments
 (0)