Skip to content

Commit e34d254

Browse files
author
Jianke LIN
committed
fix: make streamable-http tests pass CI
1 parent f6ce6bd commit e34d254

3 files changed

Lines changed: 3 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
@@ -487,7 +487,7 @@ async def _handle_reconnection(
487487
# Stream ended again without response - reconnect again (reset attempt counter)
488488
logger.info("SSE stream disconnected, reconnecting...")
489489
await self._handle_reconnection(ctx, reconnect_last_event_id, reconnect_retry_ms, 0)
490-
except Exception as e: # pragma: no cover
490+
except Exception as e:
491491
logger.debug(f"Reconnection failed: {e}")
492492
# Try to reconnect again if we still have an event ID
493493
await self._handle_reconnection(ctx, last_event_id, retry_interval_ms, attempt + 1)

tests/client/test_streamable_http.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
127127
await transport._handle_sse_response(response, ctx)
128128
message = await read_stream.receive()
129129

130+
assert isinstance(message, SessionMessage)
130131
assert isinstance(message.message, JSONRPCError)
131132
assert message.message.id == 1
132133
assert message.message.error.code == CONNECTION_CLOSED

tests/interaction/transports/test_hosting_resume.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ async def record_request(request: httpx.Request) -> None:
346346
):
347347
manager_ref = manager
348348
with anyio.fail_after(5): # pragma: no branch
349-
async with (
349+
async with ( # pragma: no branch
350350
streamable_http_client(f"{BASE_URL}/mcp", http_client=http, terminate_on_close=False) as (r, w),
351351
ClientSession(r, w) as session,
352352
anyio.create_task_group() as tg,

0 commit comments

Comments
 (0)