fix flaky TestEndToEndSimConnect holepunching test

libp2p · Sep 19, 2021 · 9709f3c · 9709f3c
1 parent a5f982f
commit 9709f3c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/p2p/protocol/holepunch/coordination_test.go b/p2p/protocol/holepunch/coordination_test.go
@@ -102,8 +102,8 @@ func TestEndToEndSimConnect(t *testing.T) {
 	ensureDirectConn(t, h1, h2)
 	// ensure no hole-punching streams are open on either side
 	ensureNoHolePunchingStream(t, h1, h2)
-	events := tr.getEvents()
-	require.Len(t, events, 3)
+	var events []*holepunch.Event
+	require.Eventually(t, func() bool { events = tr.getEvents(); return len(events) == 3 }, time.Second, 10*time.Millisecond)
 	require.Equal(t, events[0].Type, holepunch.StartHolePunchEvtT)
 	require.Equal(t, events[1].Type, holepunch.HolePunchAttemptEvtT)
 	require.Equal(t, events[2].Type, holepunch.EndHolePunchEvtT)