modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests: add lowest dependency tests (#1067)

Browse source 
Co-authored-by: ihrpr <inna@anthropic.com>
This commit is contained in:
Marcelo Trylesinski 2025-07-06 09:38:56 +02:00 • committed by GitHub
commit 3fa7bc7b61
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 473 additions and 499 deletions
Download patch file Download diff file Expand all files Collapse all files

68
tests/issues/test_188_concurrency.py
View file

@ -1,60 +1,46 @@
import anyio
import pytest
from pydantic import AnyUrl
from mcp.server.fastmcp import FastMCP
from mcp.shared.memory import (
create_connected_server_and_client_session as create_session,
)
_sleep_time_seconds = 0.01
_resource_name = "slow://slow_resource"
from mcp.shared.memory import create_connected_server_and_client_session as create_session
@pytest.mark.anyio
async def test_messages_are_executed_concurrently():
server = FastMCP("test")
call_timestamps = []
event = anyio.Event()
tool_started = anyio.Event()
call_order = []
@server.tool("sleep")
async def sleep_tool():
call_timestamps.append(("tool_start_time", anyio.current_time()))
await anyio.sleep(_sleep_time_seconds)
call_timestamps.append(("tool_end_time", anyio.current_time()))
call_order.append("waiting_for_event")
tool_started.set()
await event.wait()
call_order.append("tool_end")
return "done"
@server.resource(_resource_name)
async def slow_resource():
call_timestamps.append(("resource_start_time", anyio.current_time()))
await anyio.sleep(_sleep_time_seconds)
call_timestamps.append(("resource_end_time", anyio.current_time()))
@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 create_session(server._mcp_server) as client_session:
# First tool will wait on event, second will set it
async with anyio.create_task_group() as tg:
for _ in range(10):
tg.start_soon(client_session.call_tool, "sleep")
tg.start_soon(client_session.read_resource, AnyUrl(_resource_name))
# 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")
active_calls = 0
max_concurrent_calls = 0
for call_type, _ in sorted(call_timestamps, key=lambda x: x[1]):
if "start" in call_type:
active_calls += 1
max_concurrent_calls = max(max_concurrent_calls, active_calls)
else:
active_calls -= 1
print(f"Max concurrent calls: {max_concurrent_calls}")
assert max_concurrent_calls > 1, "No concurrent calls were executed"
def main():
anyio.run(test_messages_are_executed_concurrently)
if __name__ == "__main__":
import logging
logging.basicConfig(level=logging.DEBUG)
main()
# 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}"