Skip to content

Commit 4b64646

Browse files
author
Jianke LIN
committed
test(streamable-http): cover closed read stream cleanup
1 parent 8bf41bc commit 4b64646

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
@@ -100,6 +100,26 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
100100
assert message.message.error.code == CONNECTION_CLOSED
101101

102102

103+
@pytest.mark.anyio
104+
async def test_sse_response_disconnect_ignores_closed_read_stream() -> None:
105+
transport = StreamableHTTPTransport("http://example.com/mcp")
106+
async with httpx.AsyncClient() as client:
107+
read_stream_writer, read_stream = create_context_streams[SessionMessage | Exception](1)
108+
request = JSONRPCRequest(jsonrpc="2.0", id=1, method="tools/call", params={"name": "noop", "arguments": {}})
109+
ctx = RequestContext(
110+
client=client,
111+
session_id=None,
112+
session_message=SessionMessage(request),
113+
metadata=None,
114+
read_stream_writer=read_stream_writer,
115+
)
116+
response = httpx.Response(200, headers={"content-type": "text/event-stream"}, content=b"")
117+
118+
async with read_stream_writer, read_stream:
119+
await read_stream.aclose()
120+
await transport._handle_sse_response(response, ctx)
121+
122+
103123
@pytest.mark.anyio
104124
async def test_post_request_merges_per_message_metadata_headers() -> None:
105125
"""`ClientMessageMetadata.headers` on a `SessionMessage` are merged into the outgoing POST headers

0 commit comments

Comments
 (0)