diff --git a/tests/server/mcpserver/test_url_elicitation.py b/tests/server/mcpserver/test_url_elicitation.py index 1311bd672..c077eb21f 100644 --- a/tests/server/mcpserver/test_url_elicitation.py +++ b/tests/server/mcpserver/test_url_elicitation.py @@ -1,6 +1,5 @@ """Test URL mode elicitation feature (SEP 1036).""" -import anyio import pytest from pydantic import BaseModel, Field @@ -216,10 +215,8 @@ async def elicitation_callback(context: RequestContext[ClientSession], params: E assert isinstance(result.content[0], TextContent) assert result.content[0].text == "Elicitation completed" - # Give time for notification to be processed - await anyio.sleep(0.1) - - # Verify the notification was sent + # Verify the notification was sent (flag is set inside the tool handler + # which completes before call_tool returns, so no wait needed) assert notification_sent