Skip to content

Commit 8b7f1cd

Browse files
committed
simplify test code
1 parent 45ac52a commit 8b7f1cd

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

tests/shared/test_session.py

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from mcp.shared.exceptions import McpError
1010
from mcp.shared.memory import create_connected_server_and_client_session
1111
from mcp.types import (
12-
ClientRequest,
1312
EmptyResult,
1413
)
1514

@@ -88,20 +87,10 @@ async def handle_list_tools() -> list[types.Tool]:
8887

8988
return server
9089

91-
async def make_request(client_session):
90+
async def make_request(client_session: ClientSession):
9291
nonlocal ev_cancelled
9392
try:
94-
await client_session.send_request(
95-
ClientRequest(
96-
types.CallToolRequest(
97-
method="tools/call",
98-
params=types.CallToolRequestParams(
99-
name="slow_tool", arguments={}
100-
),
101-
)
102-
),
103-
types.CallToolResult,
104-
)
93+
await client_session.call_tool("slow_tool")
10594
pytest.fail("Request should have been cancelled")
10695
except McpError as e:
10796
# Expected - request was cancelled

0 commit comments

Comments
 (0)