test: satisfy streamable http coverage checks

Author: Genmin

src/mcp/client/session_group.py

@@ -147,9 +147,9 @@ def __init__(
         self._session_exit_stacks = {}
         self._component_name_hook = component_name_hook
 
-    async def __aenter__(self) -> Self:  # pragma: no cover
+    async def __aenter__(self) -> Self:
         # Enter the exit stack only if we created it ourselves
-        if self._owns_exit_stack:
+        if self._owns_exit_stack:  # pragma: no branch
             await self._exit_stack.__aenter__()
         return self
 
@@ -158,22 +158,22 @@ async def __aexit__(
         _exc_type: type[BaseException] | None,
         _exc_val: BaseException | None,
         _exc_tb: TracebackType | None,
-    ) -> bool | None:  # pragma: no cover
+    ) -> bool | None:
         """Closes session exit stacks and main exit stack upon completion."""
 
         # Only close the main exit stack if we created it
-        if self._owns_exit_stack:
+        if self._owns_exit_stack:  # pragma: no branch
             await self._exit_stack.aclose()
 
         # Concurrently close session stacks.
         async with anyio.create_task_group() as tg:
             for exit_stack in self._session_exit_stacks.values():
-                tg.start_soon(exit_stack.aclose)
+                tg.start_soon(exit_stack.aclose)  # pragma: no cover
 
     @property
     def sessions(self) -> list[mcp.ClientSession]:
         """Returns the list of sessions being managed."""
-        return list(self._sessions.keys())  # pragma: no cover
+        return list(self._sessions.keys())
 
     @property
     def prompts(self) -> dict[str, types.Prompt]:
@@ -323,7 +323,7 @@ async def _establish_session(
             await self._exit_stack.enter_async_context(session_stack)
 
             return result.server_info, session
-        except Exception:  # pragma: no cover
+        except Exception:
             # If anything during this setup fails, ensure the session-specific
             # stack is closed.
             await session_stack.aclose()

tests/client/test_notification_response.py

@@ -162,7 +162,9 @@ async def raise_connect_error(request: httpx.Request) -> httpx.Response:
     async with httpx.AsyncClient(transport=httpx.MockTransport(raise_connect_error)) as client:
         async with streamable_http_client("http://localhost/mcp", http_client=client) as (read_stream, write_stream):
             async with ClientSession(read_stream, write_stream) as session:  # pragma: no branch
-                with pytest.raises(MCPError, match="Transport error: All connection attempts failed"):
+                with pytest.raises(
+                    MCPError, match="Transport error: All connection attempts failed"
+                ):  # pragma: no branch
                     await session.initialize()
 
 

tests/client/test_session_group.py

@@ -402,7 +402,7 @@ def mock_client_factory(
     group = ClientSessionGroup()
     async with group:
         with mock.patch("mcp.client.session_group.create_mcp_http_client", side_effect=mock_client_factory):
-            with pytest.raises(MCPError, match="Transport error: All connection attempts failed"):
+            with pytest.raises(MCPError, match="Transport error: All connection attempts failed"):  # pragma: no branch
                 await group.connect_to_server(StreamableHttpParameters(url="http://localhost:3001/mcp/"))
 
     assert group.sessions == []