tests: add timeouts

opentensor · Feb 3, 2025 · e6d00ad · e6d00ad
1 parent 321d359
commit e6d00ad
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/unit_tests/test_axon.py b/tests/unit_tests/test_axon.py
@@ -815,10 +815,11 @@ async def lifespan(app):
     )
     server.start()
 
-    server_started.wait()
+    server_started.wait(3.0)
 
-    while not (server.started or server_stopped.is_set()):
-        await asyncio.sleep(1)
+    async with asyncio.timeout(3.0):
+        while not (server.started or server_stopped.is_set()):
+            await asyncio.sleep(1.0)
 
     assert server.is_running is True