fix flaky TestBackoffConnector test

libp2p · Feb 8, 2022 · fafaab0 · fafaab0
1 parent ddc3e2b
commit fafaab0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/p2p/discovery/backoff/backoffconnector_test.go b/p2p/discovery/backoff/backoffconnector_test.go
@@ -77,19 +77,20 @@ func TestBackoffConnector(t *testing.T) {
 	require.NoError(t, err)
 
 	bc.Connect(context.Background(), loadCh(hosts))
-	require.Eventually(t, func() bool { return len(primary.Network().Conns()) == len(hosts)-1 }, 3*time.Second, 10*time.Millisecond)
+	require.Eventually(t, func() bool { return len(primary.Network().Peers()) == len(hosts)-1 }, 3*time.Second, 10*time.Millisecond)
 
+	time.Sleep(100 * time.Millisecond) // give connection attempts time to complete (relevant when using multiple transports)
 	for _, c := range primary.Network().Conns() {
 		c.Close()
 	}
-	require.Eventually(t, func() bool { return len(primary.Network().Conns()) == 0 }, 3*time.Second, 10*time.Millisecond)
+	require.Eventually(t, func() bool { return len(primary.Network().Peers()) == 0 }, 3*time.Second, 10*time.Millisecond)
 
 	bc.Connect(context.Background(), loadCh(hosts))
 	require.Empty(t, primary.Network().Conns(), "shouldn't be connected to any peers")
 
 	time.Sleep(time.Millisecond * 500)
 	bc.Connect(context.Background(), loadCh(hosts))
-	require.Eventually(t, func() bool { return len(primary.Network().Conns()) == len(hosts)-2 }, 3*time.Second, 10*time.Millisecond)
+	require.Eventually(t, func() bool { return len(primary.Network().Peers()) == len(hosts)-2 }, 3*time.Second, 10*time.Millisecond)
 	// make sure we actually don't connect to host 1 any more
 	time.Sleep(100 * time.Millisecond)
 	require.Len(t, primary.Network().Conns(), len(hosts)-2, "wrong number of connections")