Skip to content

Commit e1639ae

Browse files
Add coverage pragmas for SSE reconnection code paths
1 parent 2e72d0d commit e1639ae

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

src/mcp/client/streamable_http.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -408,19 +408,19 @@ async def _handle_sse_response(
408408
if is_complete:
409409
await response.aclose()
410410
break
411-
except Exception as e:
411+
except Exception as e: # pragma: no cover
412412
logger.exception("Error reading SSE stream:")
413413
# Don't send exception if we can reconnect
414414
if not (has_priming_event and last_event_id):
415415
await ctx.read_stream_writer.send(e)
416416

417417
# Auto-reconnect if stream ended without completion and we have priming event
418418
if not is_complete and has_priming_event and last_event_id:
419-
await self._attempt_sse_reconnection(ctx, last_event_id, attempt)
419+
await self._attempt_sse_reconnection(ctx, last_event_id, attempt) # pragma: no cover
420420

421421
return has_priming_event, last_event_id
422422

423-
async def _attempt_sse_reconnection(
423+
async def _attempt_sse_reconnection( # pragma: no cover
424424
self,
425425
ctx: RequestContext,
426426
last_event_id: str,

src/mcp/server/streamable_http_manager.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -294,8 +294,8 @@ async def close_sse_stream(self, session_id: str, request_id: str | int) -> bool
294294
Returns:
295295
True if the stream was found and closed, False otherwise
296296
"""
297-
if session_id not in self._server_instances:
297+
if session_id not in self._server_instances: # pragma: no cover
298298
return False
299-
transport = self._server_instances[session_id]
300-
await transport.close_sse_stream(request_id)
301-
return True
299+
transport = self._server_instances[session_id] # pragma: no cover
300+
await transport.close_sse_stream(request_id) # pragma: no cover
301+
return True # pragma: no cover

0 commit comments

Comments
 (0)