first commit

testing/api/utils.py

@@ -0,0 +1,27 @@
+import multiprocessing
+import time
+
+import requests
+
+from atroposlib.cli.run_api import main as run_api_main
+
+
+def check_api_running() -> bool:
+    try:
+        data = requests.get("http://localhost:8000/info")
+        return data.status_code == 200
+    except requests.exceptions.ConnectionError:
+        return False
+
+
+def launch_api_for_testing(max_wait_for_api: int = 10) -> multiprocessing.Process:
+    api_proc = multiprocessing.Process(target=run_api_main)
+    api_proc.start()
+    counter = 0
+    while not check_api_running():
+        time.sleep(1)
+        counter += 1
+        if counter > max_wait_for_api:
+            raise TimeoutError("API server did not start in time.")
+    print("API server started for testing.")
+    return api_proc