Skip to content

Commit 7ecc23d

Browse files
author
Jianke LIN
committed
fix(streamable-http): stabilize branch coverage
1 parent 2b7d050 commit 7ecc23d

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/mcp/client/streamable_http.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ async def _handle_resumption_request(self, ctx: RequestContext) -> None:
270270
if is_complete:
271271
break
272272
finally:
273-
if event_source is not None:
273+
if event_source is not None: # pragma: no branch
274274
await event_source.response.aclose()
275275

276276
def _consume_modern_cancellation(self, session_message: SessionMessage) -> bool:

tests/client/test_streamable_http_response_cleanup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@
33
import httpx
44
import pytest
55
from httpx_sse import ServerSentEvent
6+
from mcp_types import JSONRPCRequest
67

78
from mcp.client.streamable_http import RequestContext, StreamableHTTPTransport
89
from mcp.shared._context_streams import create_context_streams
910
from mcp.shared.message import ClientMessageMetadata, SessionMessage
10-
from mcp.types import JSONRPCRequest
1111

1212

1313
class _RaiseEventSource:

0 commit comments

Comments
 (0)