test: remove obsolete pragma and isolate session establishment in tests

Author: musi22

src/mcp/shared/_httpx_utils.py

@@ -59,7 +59,7 @@ def create_mcp_http_client(
     kwargs: dict[str, Any] = {"timeout": timeout}
     if headers is not None:
         kwargs["headers"] = headers
-    if auth is not None:  # pragma: no cover
+    if auth is not None:
         kwargs["auth"] = auth
     return httpx2.AsyncClient(**kwargs)
 

tests/client/test_session_group.py

@@ -565,9 +565,10 @@ def record_stream_client(*args: Any, **kwargs: Any) -> mock.AsyncMock:
             mock_session_cls.return_value = mock_session_cm
 
             group = ClientSessionGroup()
-            async with group:
-                await group.connect_to_server(params_a)
-                await group.connect_to_server(params_b)
+            async with contextlib.AsyncExitStack() as stack:
+                group._exit_stack = stack
+                await group._establish_session(params_a, ClientSessionParameters())
+                await group._establish_session(params_b, ClientSessionParameters())
 
             assert len(recorded_clients) == 2
             assert recorded_clients[0].auth is auth_a