Skip to content

Commit f69d39e

Browse files
author
Jianke LIN
committed
test(streamable-http): cover closed read stream cleanup
1 parent 8650d5c commit f69d39e

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

tests/client/test_streamable_http.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,26 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
8585
assert message.message.error.code == CONNECTION_CLOSED
8686

8787

88+
@pytest.mark.anyio
89+
async def test_sse_response_disconnect_ignores_closed_read_stream() -> None:
90+
transport = StreamableHTTPTransport("http://example.com/mcp")
91+
async with httpx.AsyncClient() as client:
92+
read_stream_writer, read_stream = create_context_streams[SessionMessage | Exception](1)
93+
request = JSONRPCRequest(jsonrpc="2.0", id=1, method="tools/call", params={"name": "noop", "arguments": {}})
94+
ctx = RequestContext(
95+
client=client,
96+
session_id=None,
97+
session_message=SessionMessage(request),
98+
metadata=None,
99+
read_stream_writer=read_stream_writer,
100+
)
101+
response = httpx.Response(200, headers={"content-type": "text/event-stream"}, content=b"")
102+
103+
async with read_stream_writer, read_stream:
104+
await read_stream.aclose()
105+
await transport._handle_sse_response(response, ctx)
106+
107+
88108
@pytest.mark.anyio
89109
async def test_post_request_merges_per_message_metadata_headers() -> None:
90110
"""`ClientMessageMetadata.headers` on a `SessionMessage` are merged into the outgoing POST headers

0 commit comments

Comments
 (0)