chore: import upstream snapshot with attribution
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Has been cancelled
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,85 @@
|
||||
import anyio
|
||||
import pytest
|
||||
|
||||
from mcp import Client
|
||||
from mcp.server.mcpserver import MCPServer
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_messages_are_executed_concurrently_tools():
|
||||
server = MCPServer("test")
|
||||
event = anyio.Event()
|
||||
tool_started = anyio.Event()
|
||||
call_order: list[str] = []
|
||||
|
||||
@server.tool("sleep")
|
||||
async def sleep_tool():
|
||||
call_order.append("waiting_for_event")
|
||||
tool_started.set()
|
||||
await event.wait()
|
||||
call_order.append("tool_end")
|
||||
return "done"
|
||||
|
||||
@server.tool("trigger")
|
||||
async def trigger():
|
||||
# Wait for tool to start before setting the event
|
||||
await tool_started.wait()
|
||||
call_order.append("trigger_started")
|
||||
event.set()
|
||||
call_order.append("trigger_end")
|
||||
return "slow"
|
||||
|
||||
async with Client(server) as client_session:
|
||||
# First tool will wait on event, second will set it
|
||||
async with anyio.create_task_group() as tg:
|
||||
# Start the tool first (it will wait on event)
|
||||
tg.start_soon(client_session.call_tool, "sleep")
|
||||
# Then the trigger tool will set the event to allow the first tool to continue
|
||||
await client_session.call_tool("trigger")
|
||||
|
||||
# Verify that both ran concurrently
|
||||
assert call_order == [
|
||||
"waiting_for_event",
|
||||
"trigger_started",
|
||||
"trigger_end",
|
||||
"tool_end",
|
||||
], f"Expected concurrent execution, but got: {call_order}"
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_messages_are_executed_concurrently_tools_and_resources():
|
||||
server = MCPServer("test")
|
||||
event = anyio.Event()
|
||||
tool_started = anyio.Event()
|
||||
call_order: list[str] = []
|
||||
|
||||
@server.tool("sleep")
|
||||
async def sleep_tool():
|
||||
call_order.append("waiting_for_event")
|
||||
tool_started.set()
|
||||
await event.wait()
|
||||
call_order.append("tool_end")
|
||||
return "done"
|
||||
|
||||
@server.resource("slow://slow_resource")
|
||||
async def slow_resource():
|
||||
# Wait for tool to start before setting the event
|
||||
await tool_started.wait()
|
||||
event.set()
|
||||
call_order.append("resource_end")
|
||||
return "slow"
|
||||
|
||||
async with Client(server) as client_session:
|
||||
# First tool will wait on event, second will set it
|
||||
async with anyio.create_task_group() as tg:
|
||||
# Start the tool first (it will wait on event)
|
||||
tg.start_soon(client_session.call_tool, "sleep")
|
||||
# Then the resource (it will set the event)
|
||||
tg.start_soon(client_session.read_resource, "slow://slow_resource")
|
||||
|
||||
# Verify that both ran concurrently
|
||||
assert call_order == [
|
||||
"waiting_for_event",
|
||||
"resource_end",
|
||||
"tool_end",
|
||||
], f"Expected concurrent execution, but got: {call_order}"
|
||||
Reference in New Issue
Block a user