From 05d811db900dfcfd975687e17c2d3a43ef65518a Mon Sep 17 00:00:00 2001
From: green <xgreenx9999@gmail.com>
Date: Sat, 5 Oct 2024 00:32:02 +0200
Subject: [PATCH] Remove flakiness from `test_peer_info` test

---
 tests/tests/node_info.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/tests/node_info.rs b/tests/tests/node_info.rs
index f84b767e4df..4dda549bb86 100644
--- a/tests/tests/node_info.rs
+++ b/tests/tests/node_info.rs
@@ -92,8 +92,18 @@ async fn test_peer_info() {
     // This is just a mock of what we should be able to do with GQL API.
     let client = producer.node.bound_address;
     let client = FuelClient::from(client);
-    let peers = client.connected_peers_info().await.unwrap();
-    assert_eq!(peers.len(), 2);
+    let mut peers;
+
+    // It takes some time before all validators are connected.
+    loop {
+        peers = client.connected_peers_info().await.unwrap();
+
+        if peers.len() == 2 {
+            break;
+        }
+        tokio::time::sleep(Duration::from_secs(1)).await;
+    }
+
     let info = peers
         .iter()
         .find(|info| info.id.to_string() == validator_peer_id.to_base58())