attempt to fix flakiness

Author: tac0turtle

pkg/p2p/client_test.go

@@ -341,9 +341,34 @@ func TestClientInfoMethods(t *testing.T) {
 	})
 
 	t.Run("GetPeers", func(t *testing.T) {
+		// Wait for peer discovery to find the peers
+		expectedPeerIDs := []peer.ID{client1.host.ID(), client2.host.ID()}
+		
+		err := waitForCondition(2*time.Second, func() bool {
+			peers, err := client0.GetPeers()
+			if err != nil {
+				return false
+			}
+			
+			actualPeerIDs := make(map[peer.ID]bool)
+			for _, p := range peers {
+				actualPeerIDs[p.ID] = true
+			}
+			
+			// Check if all expected peers are found
+			for _, expectedID := range expectedPeerIDs {
+				if !actualPeerIDs[expectedID] {
+					return false
+				}
+			}
+			return true
+		})
+		
+		require.NoError(err, "Timed out waiting for GetPeers to discover all peers")
+		
+		// Now verify the peers are as expected
 		peers, err := client0.GetPeers()
 		assert.NoError(err)
-		expectedPeerIDs := []peer.ID{client1.host.ID(), client2.host.ID()}
 		actualPeerIDs := make([]peer.ID, len(peers))
 		for i, p := range peers {
 			actualPeerIDs[i] = p.ID