Skip to content

Commit 21a9d6e

Browse files
committed
Adjusted test function to match upstream main.
1 parent ae6caf0 commit 21a9d6e

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

‎tests/server/mcpserver/test_server.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1585,17 +1585,17 @@ def prompt_fn() -> str:
15851585
# wraps `async with Client(...): with pytest.raises(...): await ...` as
15861586
# a 4th nesting level around a single `await` statement (3 levels of
15871587
# nesting is OK; 4 is not). So `caplog.set_level` avoids extra `with` layer.
1588-
caplog.set_level(logging.WARNING, logger="mcp.server.mcpserver.server")
1588+
caplog.set_level(logging.WARNING, logger="mcp.shared.jsonrpc_dispatcher")
15891589
async with Client(mcp, mode="legacy") as client:
15901590
with pytest.raises(MCPError):
15911591
await client.get_prompt("prompt_fn")
15921592

1593-
server_records = [r for r in caplog.records if r.name == "mcp.server.mcpserver.server"]
1594-
assert len(server_records) == 1
1593+
dispatcher_records = [r for r in caplog.records if r.name == "mcp.shared.jsonrpc_dispatcher"]
1594+
assert len(dispatcher_records) == 1
15951595

15961596
# Unexpected errors should have error log with exc_info.
1597-
assert server_records[0].levelno == logging.ERROR
1598-
assert server_records[0].exc_info is not None
1597+
assert dispatcher_records[0].levelno == logging.ERROR
1598+
assert dispatcher_records[0].exc_info is not None
15991599

16001600

16011601
async def test_resource_decorator_rfc6570_reserved_expansion():

0 commit comments

Comments
 (0)