fix(tests): re-attempt connection to NATS

Signed-off-by: Victor Adossi <[email protected]>
wasmCloud · Oct 4, 2024 · 96325ff · 96325ff
1 parent 3f1f0a9
commit 96325ff
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tests/e2e.rs b/tests/e2e.rs
@@ -94,10 +94,18 @@ impl ClientInfo {
             .kill_on_drop(true)
             .spawn()
             .expect("Unable to watch docker logs");
+
         // Connect to NATS
-        let client = async_nats::connect("127.0.0.1:4222")
-            .await
-            .expect("Unable to connect to nats");
+        let client = tokio::time::timeout(Duration::from_secs(3), async {
+            loop {
+                if let Ok(client) = async_nats::connect("127.0.0.1:4222").await {
+                    return client;
+                }
+                tokio::time::sleep(Duration::from_millis(250)).await;
+            }
+        })
+        .await
+        .expect("timed out while creating NATS client");
 
         ClientInfo {
             client,