test(e2e): replace fixed WS-readiness sleeps with polls in core e2e tests

test_e2e.py and test_parallel_tool_calls.py both used fixed asyncio.sleep()
calls (0.5s–0.8s) as synchronization barriers to wait for WebSocket
connections to establish. Under full-suite load these could fail or waste
wall-clock time.

test_publish_and_chat: poll conn._ws instead of sleep(0.5).
test_invoke_capability: poll pub.find_capability() which is the actual
  readiness condition (WS up + hub round-trip + connection-event processed).
test_parallel_tool_calls: poll conn._ws instead of sleep(0.8).

Also removed the pre-existing unused `os` import from test_e2e.py.

Author: Zawwarsami16

tests/test_e2e.py

@@ -5,7 +5,6 @@
 """
 
 import asyncio
-import os
 import socket
 import threading
 import time
@@ -88,7 +87,10 @@ def chat_handler(messages, options):
         hub_url=hub,
         capabilities={},  # no capabilities exposed
     )
-    await asyncio.sleep(0.5)
+    for _ in range(50):
+        if conn._ws is not None:
+            break
+        await asyncio.sleep(0.05)
 
     resp = await conn.chat(messages=[{"role": "user", "content": "hello"}])
     assert "got: hello" in resp.get("text", "")
@@ -117,18 +119,23 @@ def fake_capability(args):
             break
         await asyncio.sleep(0.1)
 
-    conn = connect(
+    connect(
         ai_name=pub.name,
         api_key=pub.api_key,
         hub_url=hub,
         capabilities={
             "test_op": ({"type": "object"}, fake_capability),
         },
     )
-    # wait for the connection-event to propagate
-    await asyncio.sleep(0.8)
+    # poll until the publisher receives the connection-event carrying the
+    # capability (WS up + hub round-trip + publisher WS message processed)
+    cid = None
+    for _ in range(60):
+        cid = pub.find_capability("test_op")
+        if cid is not None:
+            break
+        await asyncio.sleep(0.05)
 
-    cid = pub.find_capability("test_op")
     assert cid is not None, "publisher never saw the capability"
 
     result = await pub.invoke(cid, "test_op", {"hello": "world"})

tests/test_parallel_tool_calls.py

@@ -120,7 +120,10 @@ async def slow_b(_args):
             "slow_b": ({"type": "object"}, slow_b),
         },
     )
-    await asyncio.sleep(0.8)
+    for _ in range(60):
+        if conn._ws is not None:
+            break
+        await asyncio.sleep(0.05)
 
     t0 = time.monotonic()
     async with httpx.AsyncClient(timeout=15.0) as client: