Remove Tasks (SEP-1686) from the SDK (#2714)

tests/server/mcpserver/test_server.py

@@ -11,7 +11,6 @@ from starlette.routing import Mount, Route
 
 from mcp.client import Client
 from mcp.server.context import ServerRequestContext
-from mcp.server.experimental.request_context import Experimental
 from mcp.server.mcpserver import Context, MCPServer
 from mcp.server.mcpserver.exceptions import ToolError
 from mcp.server.mcpserver.prompts.base import Message, UserMessage
@@ -1502,7 +1501,6 @@ async def test_report_progress_passes_related_request_id():
         session=mock_session,
         meta={"progress_token": "tok-1"},
         lifespan_context=None,
-        experimental=Experimental(),
     )
 
     ctx = Context(request_context=request_context, mcp_server=MagicMock())

tests/server/test_session.py

@@ -76,6 +76,49 @@ async def test_server_session_initialize():
     assert received_initialized
 
 
+@pytest.mark.anyio
+async def test_check_client_capability():
+    """check_client_capability reflects the capabilities sent by the client at initialize."""
+    server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](1)
+    client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](1)
+
+    initialized = anyio.Event()
+
+    async def list_roots_callback(context: Any) -> types.ListRootsResult:  # pragma: no cover
+        return types.ListRootsResult(roots=[])
+
+    async def run_server(server_session: ServerSession):
+        async for message in server_session.incoming_messages:  # pragma: no branch
+            if isinstance(message, ClientNotification) and isinstance(
+                message, InitializedNotification
+            ):  # pragma: no branch
+                initialized.set()
+                return
+
+    async with (
+        ServerSession(
+            client_to_server_receive,
+            server_to_client_send,
+            InitializationOptions(server_name="mcp", server_version="0.1.0", capabilities=ServerCapabilities()),
+        ) as server_session,
+        ClientSession(
+            server_to_client_receive,
+            client_to_server_send,
+            list_roots_callback=list_roots_callback,
+        ) as client_session,
+        anyio.create_task_group() as tg,
+    ):
+        tg.start_soon(run_server, server_session)
+        await client_session.initialize()
+        with anyio.fail_after(5):
+            await initialized.wait()
+
+        # ClientSession advertises roots when a list_roots_callback is provided.
+        assert server_session.check_client_capability(types.ClientCapabilities(roots=types.RootsCapability()))
+        # ClientSession does not advertise sampling without a sampling_callback.
+        assert not server_session.check_client_capability(types.ClientCapabilities(sampling=types.SamplingCapability()))
+
+
 @pytest.mark.anyio
 async def test_server_capabilities():
     notification_options = NotificationOptions()