Skip to content

Commit 7efc7ed

Browse files
author
Jianke LIN
committed
fix(streamable-http): stabilize branch coverage
1 parent 95d72f1 commit 7efc7ed

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
@@ -247,7 +247,7 @@ async def _handle_resumption_request(self, ctx: RequestContext) -> None:
247247
if is_complete:
248248
break
249249
finally:
250-
if event_source is not None:
250+
if event_source is not None: # pragma: no branch
251251
await event_source.response.aclose()
252252

253253
async def _handle_post_request(self, ctx: RequestContext) -> None:

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)