Clean up test code: remove useless comments and replace sleeps with events

- Remove redundant '# Should not raise' comments in test_capabilities.py
- Remove redundant '# No handler' comments in test_server_task_context.py
- Replace arbitrary sleeps with deterministic event-based synchronization
  in test_task_result_handler.py (poll for wait events before proceeding)

Author: maxisbey

tests/experimental/tasks/server/test_server_task_context.py

@@ -228,7 +228,7 @@ async def test_elicit_raises_without_handler() -> None:
         store=store,
         session=mock_session,
         queue=queue,
-        handler=None,  # No handler
+        handler=None,
     )
 
     with pytest.raises(RuntimeError, match="handler is required"):
@@ -251,7 +251,7 @@ async def test_elicit_url_raises_without_handler() -> None:
         store=store,
         session=mock_session,
         queue=queue,
-        handler=None,  # No handler
+        handler=None,
     )
 
     with pytest.raises(RuntimeError, match="handler is required for elicit_url"):
@@ -278,7 +278,7 @@ async def test_create_message_raises_without_handler() -> None:
         store=store,
         session=mock_session,
         queue=queue,
-        handler=None,  # No handler
+        handler=None,
     )
 
     with pytest.raises(RuntimeError, match="handler is required"):
@@ -662,7 +662,7 @@ async def test_elicit_as_task_raises_without_handler() -> None:
         store=store,
         session=mock_session,
         queue=queue,
-        handler=None,  # No handler
+        handler=None,
     )
 
     with pytest.raises(RuntimeError, match="handler is required for elicit_as_task"):
@@ -697,7 +697,7 @@ async def test_create_message_as_task_raises_without_handler() -> None:
         store=store,
         session=mock_session,
         queue=queue,
-        handler=None,  # No handler
+        handler=None,
     )
 
     with pytest.raises(RuntimeError, match="handler is required for create_message_as_task"):

tests/experimental/tasks/server/test_task_result_handler.py

@@ -151,7 +151,10 @@ async def run_handle() -> None:
 
     async with anyio.create_task_group() as tg:
         tg.start_soon(run_handle)
-        await anyio.sleep(0.05)
+
+        # Wait for handler to start waiting (event gets created when wait starts)
+        while task.taskId not in store._update_events:
+            await anyio.sleep(0)
 
         await store.store_result(task.taskId, CallToolResult(content=[TextContent(type="text", text="Done")]))
         await store.update_task(task.taskId, status="completed")
@@ -303,7 +306,9 @@ async def failing_wait(task_id: str) -> None:
 
     # Queue a message to unblock the race via the queue path
     async def enqueue_later() -> None:
-        await anyio.sleep(0.01)
+        # Wait for queue to start waiting (event gets created when wait starts)
+        while task.taskId not in queue._events:
+            await anyio.sleep(0)
         await queue.enqueue(
             task.taskId,
             QueuedMessage(
@@ -338,7 +343,9 @@ async def failing_wait(task_id: str) -> None:
 
     # Update the store to unblock the race via the store path
     async def update_later() -> None:
-        await anyio.sleep(0.01)
+        # Wait for store to start waiting (event gets created when wait starts)
+        while task.taskId not in store._update_events:
+            await anyio.sleep(0)
         await store.update_task(task.taskId, status="completed")
 
     async with anyio.create_task_group() as tg:

tests/experimental/tasks/test_capabilities.py

@@ -252,7 +252,7 @@ def test_passes_when_present(self) -> None:
                 )
             )
         )
-        require_task_augmented_elicitation(caps)  # Should not raise
+        require_task_augmented_elicitation(caps)
 
 
 class TestRequireTaskAugmentedSampling:
@@ -280,4 +280,4 @@ def test_passes_when_present(self) -> None:
                 )
             )
         )
-        require_task_augmented_sampling(caps)  # Should not raise
+        require_task_augmented_sampling(caps)