fix(tests): re-attempt connection to NATS

Signed-off-by: Victor Adossi <[email protected]>
wasmCloud · Oct 4, 2024 · d4df73e · d4df73e
1 parent 22dd2c0
commit d4df73e
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,