test: wait for uvicorn server to start completly

opentensor · Jan 22, 2025 · d5753db · d5753db
1 parent 0958209
commit d5753db
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/unit_tests/test_axon.py b/tests/unit_tests/test_axon.py
@@ -16,6 +16,7 @@
 # DEALINGS IN THE SOFTWARE.
 
 
+import asyncio
 import contextlib
 import re
 import threading
@@ -816,6 +817,9 @@ async def lifespan(app):
 
     server_started.wait()
 
+    while not server.started or server_stopped.is_set():
+        await asyncio.sleep(1)
+
     assert server.is_running is True
 
     async with aiohttp.ClientSession(