fix: test + formatting

Author: lorenzocesconetto

src/mcp/shared/session.py

@@ -7,20 +7,31 @@
 
 import anyio
 import httpx
-from anyio.streams.memory import (MemoryObjectReceiveStream,
-                                  MemoryObjectSendStream)
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from pydantic import BaseModel, ValidationError
 from typing_extensions import Self
 
 from mcp.shared.exceptions import McpError
-from mcp.shared.message import (MessageMetadata, ServerMessageMetadata,
-                                SessionMessage)
-from mcp.types import (CONNECTION_CLOSED, INVALID_PARAMS,
-                       CancelledNotification, ClientNotification,
-                       ClientRequest, ClientResult, ErrorData, JSONRPCError,
-                       JSONRPCMessage, JSONRPCNotification, JSONRPCRequest,
-                       JSONRPCResponse, ProgressNotification, RequestParams,
-                       ServerNotification, ServerRequest, ServerResult)
+from mcp.shared.message import MessageMetadata, ServerMessageMetadata, SessionMessage
+from mcp.types import (
+    CONNECTION_CLOSED,
+    INVALID_PARAMS,
+    CancelledNotification,
+    ClientNotification,
+    ClientRequest,
+    ClientResult,
+    ErrorData,
+    JSONRPCError,
+    JSONRPCMessage,
+    JSONRPCNotification,
+    JSONRPCRequest,
+    JSONRPCResponse,
+    ProgressNotification,
+    RequestParams,
+    ServerNotification,
+    ServerRequest,
+    ServerResult,
+)
 
 SendRequestT = TypeVar("SendRequestT", ClientRequest, ServerRequest)
 SendResultT = TypeVar("SendResultT", ClientResult, ServerResult)

tests/shared/test_progress_notifications.py

@@ -335,18 +335,14 @@ def mock_log_error(msg, *args):
         logged_errors.append(msg % args if args else msg)
 
     # Create a progress callback that raises an exception
-    async def failing_progress_callback(
-        progress: float, total: float | None, message: str | None
-    ) -> None:
+    async def failing_progress_callback(progress: float, total: float | None, message: str | None) -> None:
         raise ValueError("Progress callback failed!")
 
     # Create a server with a tool that sends progress notifications
     server = Server(name="TestProgressServer")
 
     @server.call_tool()
-    async def handle_call_tool(
-        name: str, arguments: dict | None
-    ) -> list[types.TextContent]:
+    async def handle_call_tool(name: str, arguments: dict | None) -> list[types.TextContent]:
         if name == "progress_tool":
             # Send a progress notification
             await server.request_context.session.send_progress_notification(
@@ -376,9 +372,7 @@ async def handle_list_tools() -> list[types.Tool]:
                 types.ClientRequest(
                     types.CallToolRequest(
                         method="tools/call",
-                        params=types.CallToolRequestParams(
-                            name="progress_tool", arguments={}
-                        ),
+                        params=types.CallToolRequestParams(name="progress_tool", arguments={}),
                     )
                 ),
                 types.CallToolResult,
@@ -388,12 +382,9 @@ async def handle_list_tools() -> list[types.Tool]:
             # Verify the request completed successfully despite the callback failure
             assert len(result.content) == 1
             content = result.content[0]
-            assert isinstance(content, TextContent)
+            assert isinstance(content, types.TextContent)
             assert content.text == "progress_result"
 
             # Check that a warning was logged for the progress callback exception
             assert len(logged_errors) > 0
-            assert any(
-                "Progress callback raised an exception" in warning
-                for warning in logged_errors
-            )
+            assert any("Progress callback raised an exception" in warning for warning in logged_errors)