fix type issues in tests with unbound variables

Author: dsp-ant

tests/client/test_session.py

@@ -32,6 +32,7 @@ async def test_client_session_initialize():
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](1)
 
     initialized_notification = None
+    result = None
 
     async def mock_server():
         nonlocal initialized_notification
@@ -239,6 +240,7 @@ async def test_client_session_version_negotiation_success():
     """Test successful version negotiation with supported version"""
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](1)
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](1)
+    result = None
 
     async def mock_server():
         session_message = await client_to_server_receive.receive()

tests/server/test_lowlevel_input_validation.py

@@ -21,7 +21,7 @@ async def run_tool_test(
     tools: list[Tool],
     call_tool_handler: Callable[[str, dict[str, Any]], Awaitable[list[TextContent]]],
     test_callback: Callable[[ClientSession], Awaitable[CallToolResult]],
-) -> CallToolResult:
+) -> CallToolResult | None:
     """Helper to run a tool test with minimal boilerplate.
 
     Args:
@@ -33,6 +33,7 @@ async def run_tool_test(
         The result of the tool call
     """
     server = Server("test")
+    result = None
 
     @server.list_tools()
     async def list_tools():

tests/server/test_lowlevel_output_validation.py

@@ -21,7 +21,7 @@ async def run_tool_test(
     tools: list[Tool],
     call_tool_handler: Callable[[str, dict[str, Any]], Awaitable[Any]],
     test_callback: Callable[[ClientSession], Awaitable[CallToolResult]],
-) -> CallToolResult:
+) -> CallToolResult | None:
     """Helper to run a tool test with minimal boilerplate.
 
     Args:
@@ -34,6 +34,8 @@ async def run_tool_test(
     """
     server = Server("test")
 
+    result = None
+
     @server.list_tools()
     async def list_tools():
         return tools

tests/server/test_lowlevel_tool_annotations.py

@@ -39,6 +39,7 @@ async def list_tools():
             )
         ]
 
+    tools_result = None
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](10)
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](10)
 

tests/shared/test_progress_notifications.py

@@ -217,6 +217,8 @@ async def test_progress_context_manager():
 
     server = Server(name="ProgressContextTestServer")
 
+    progress_token = None
+
     # Register progress handler
     @server.progress_notification()
     async def handle_progress(