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

Get baseline 100% clean coverage (#1553)

Browse source
This commit is contained in:
Max Isbey 2025-11-11 13:09:32 +00:00 • committed by GitHub
commit 89e9c43acf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1032 additions and 1084 deletions
Download patch file Download diff file Expand all files Collapse all files

36
tests/server/test_session.py
View file

@ -34,7 +34,7 @@ async def test_server_session_initialize():
client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](1)
# Create a message handler to catch exceptions
async def message_handler(
async def message_handler( # pragma: no cover
message: RequestResponder[types.ServerRequest, types.ClientResult] | types.ServerNotification | Exception,
) -> None:
if isinstance(message, Exception):
@ -54,11 +54,13 @@ async def test_server_session_initialize():
capabilities=ServerCapabilities(),
),
) as server_session:
async for message in server_session.incoming_messages:
if isinstance(message, Exception):
async for message in server_session.incoming_messages: # pragma: no branch
if isinstance(message, Exception): # pragma: no cover
raise message
if isinstance(message, ClientNotification) and isinstance(message.root, InitializedNotification):
if isinstance(message, ClientNotification) and isinstance(
message.root, InitializedNotification
): # pragma: no branch
received_initialized = True
return
@ -74,7 +76,7 @@ async def test_server_session_initialize():
tg.start_soon(run_server)
await client_session.initialize()
except anyio.ClosedResourceError:
except anyio.ClosedResourceError: # pragma: no cover
pass
assert received_initialized
@ -94,7 +96,7 @@ async def test_server_capabilities():
# Add a prompts handler
@server.list_prompts()
async def list_prompts() -> list[Prompt]:
async def list_prompts() -> list[Prompt]: # pragma: no cover
return []
caps = server.get_capabilities(notification_options, experimental_capabilities)
@ -104,7 +106,7 @@ async def test_server_capabilities():
# Add a resources handler
@server.list_resources()
async def list_resources() -> list[Resource]:
async def list_resources() -> list[Resource]: # pragma: no cover
return []
caps = server.get_capabilities(notification_options, experimental_capabilities)
@ -114,7 +116,7 @@ async def test_server_capabilities():
# Add a complete handler
@server.completion()
async def complete(
async def complete( # pragma: no cover
ref: PromptReference | ResourceTemplateReference,
argument: CompletionArgument,
context: CompletionContext | None,
@ -150,11 +152,13 @@ async def test_server_session_initialize_with_older_protocol_version():
capabilities=ServerCapabilities(),
),
) as server_session:
async for message in server_session.incoming_messages:
if isinstance(message, Exception):
async for message in server_session.incoming_messages: # pragma: no branch
if isinstance(message, Exception): # pragma: no cover
raise message
if isinstance(message, types.ClientNotification) and isinstance(message.root, InitializedNotification):
if isinstance(message, types.ClientNotification) and isinstance(
message.root, InitializedNotification
): # pragma: no branch
received_initialized = True
return
@ -234,12 +238,14 @@ async def test_ping_request_before_initialization():
capabilities=ServerCapabilities(),
),
) as server_session:
async for message in server_session.incoming_messages:
if isinstance(message, Exception):
async for message in server_session.incoming_messages: # pragma: no branch
if isinstance(message, Exception): # pragma: no cover
raise message
# We should receive a ping request before initialization
if isinstance(message, RequestResponder) and isinstance(message.request.root, types.PingRequest):
if isinstance(message, RequestResponder) and isinstance(
message.request.root, types.PingRequest
): # pragma: no branch
# Respond to the ping
with message:
await message.respond(types.ServerResult(types.EmptyResult()))
@ -323,7 +329,7 @@ async def test_other_requests_blocked_before_initialization():
# Wait for the error response
error_message = await server_to_client_receive.receive()
if isinstance(error_message.message.root, types.JSONRPCError):
if isinstance(error_message.message.root, types.JSONRPCError): # pragma: no branch
error_response_received = True
error_code = error_message.message.root.error.code