Update test_server.rs

agnesoft · Jun 1, 2024 · 3dd3a4d · 3dd3a4d
1 parent 0d6aace
commit 3dd3a4d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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")
+}