fixes ci

Author: asheshvidyut

tests/server/fastmcp/test_elicitation.py

@@ -10,6 +10,7 @@
 from mcp.client.session import ElicitationFnT
 from mcp.client.transport_session import ClientTransportSession
 from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSession
 from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.context import RequestContext
 from mcp.types import ElicitRequestParams, ElicitResult, TextContent

tests/server/test_cancel_handling.py

@@ -55,8 +55,6 @@ async def handle_call_tool(name: str, arguments: dict[str, Any] | None) -> list[
 
     async with Client(server) as client:
         # First request (will be cancelled)
-        assert isinstance(client, ClientSession)
-
         async def first_request():
             try:
                 await client.session.send_request(

tests/shared/test_progress_notifications.py

@@ -369,6 +369,7 @@ async def handle_list_tools() -> list[types.Tool]:
     # Test with mocked logging
     with patch("mcp.shared.session.logging.exception", side_effect=mock_log_exception):
         async with Client(server) as client:
+            # Call tool with a failing progress callback
             result = await client.call_tool(
                 "progress_tool",
                 arguments={},

tests/shared/test_session.py

@@ -23,26 +23,12 @@
 )
 
 
-@pytest.fixture
-def mcp_server() -> Server:
-    return Server(name="test server")
-
-
-@pytest.fixture
-async def client_connected_to_server(
-    mcp_server: Server,
-) -> AsyncGenerator[ClientTransportSession, None]:
-    async with create_connected_server_and_client_session(mcp_server) as client_session:
-        yield client_session
-
-
 @pytest.mark.anyio
 async def test_in_flight_requests_cleared_after_completion():
     """Verify that _in_flight is empty after all requests complete."""
     # Send a request and wait for response
     server = Server(name="test server")
     async with Client(server) as client:
-        # Send a request and wait for response
         response = await client.send_ping()
         assert isinstance(response, EmptyResult)