Skip to content

Commit 0126d1b

Browse files
move unrelated changes to different CL
1 parent 8392b4a commit 0126d1b

File tree

1 file changed

+7
-16
lines changed

1 file changed

+7
-16
lines changed

tests/issues/test_188_concurrency.py

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -14,38 +14,29 @@
1414
@pytest.mark.anyio
1515
async def test_messages_are_executed_concurrently():
1616
server = FastMCP("test")
17-
call_timestamps = []
18-
17+
1918
@server.tool("sleep")
2019
async def sleep_tool():
21-
call_timestamps.append(("tool_start_time", anyio.current_time()))
2220
await anyio.sleep(_sleep_time_seconds)
23-
call_timestamps.append(("tool_end_time", anyio.current_time()))
2421
return "done"
2522

2623
@server.resource(_resource_name)
2724
async def slow_resource():
28-
call_timestamps.append(("resource_start_time", anyio.current_time()))
2925
await anyio.sleep(_sleep_time_seconds)
30-
call_timestamps.append(("resource_end_time", anyio.current_time()))
3126
return "slow"
3227

3328
async with create_session(server._mcp_server) as client_session:
29+
start_time = anyio.current_time()
3430
async with anyio.create_task_group() as tg:
3531
for _ in range(10):
3632
tg.start_soon(client_session.call_tool, "sleep")
3733
tg.start_soon(client_session.read_resource, AnyUrl(_resource_name))
3834

39-
active_calls = 0
40-
max_concurrent_calls = 0
41-
for call_type, _ in sorted(call_timestamps, key=lambda x: x[1]):
42-
if "start" in call_type:
43-
active_calls += 1
44-
max_concurrent_calls = max(max_concurrent_calls, active_calls)
45-
else:
46-
active_calls -= 1
47-
print(f"Max concurrent calls: {max_concurrent_calls}")
48-
assert max_concurrent_calls > 1, "No concurrent calls were executed"
35+
end_time = anyio.current_time()
36+
37+
duration = end_time - start_time
38+
assert duration < 6 * _sleep_time_seconds
39+
print(duration)
4940

5041

5142
def main():

0 commit comments

Comments
 (0)