From 3dd3a4dba42d059a50d86b44ae146d5ce5902610 Mon Sep 17 00:00:00 2001
From: Michael Vlach <vlach.michael@gmail.com>
Date: Sat, 1 Jun 2024 08:06:49 +0200
Subject: [PATCH] Update test_server.rs

---
 agdb_server/tests/test_server.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/agdb_server/tests/test_server.rs b/agdb_server/tests/test_server.rs
index 340c9e8d8..bc7d42221 100644
--- a/agdb_server/tests/test_server.rs
+++ b/agdb_server/tests/test_server.rs
@@ -222,3 +222,15 @@ impl Drop for TestServer {
         }
     }
 }
+
+pub async fn wait_for_ready(api: &AgdbApi<ReqwestClient>) -> anyhow::Result<()> {
+    for _ in 0..RETRY_ATTEMPS {
+        if api.status().await.is_ok() {
+            return Ok(());
+        }
+
+        std::thread::sleep(RETRY_TIMEOUT);
+    }
+
+    anyhow::bail!("Server not ready")
+}