fix: add type annotations and coverage pragma to tests

Github-Issue: #1283

Author: felixweinberger

tests/issues/test_1283_idle_session_cleanup.py

@@ -16,17 +16,19 @@
 """
 
 import time
+from collections.abc import Callable, Coroutine
+from typing import Any
 
 import anyio
 import pytest
-from starlette.types import Message
+from starlette.types import Message, Scope
 
 from mcp.server.lowlevel import Server
 from mcp.server.streamable_http import MCP_SESSION_ID_HEADER, StreamableHTTPServerTransport
 from mcp.server.streamable_http_manager import StreamableHTTPSessionManager
 
 
-def _make_scope() -> dict:
+def _make_scope() -> Scope:
     return {
         "type": "http",
         "method": "POST",
@@ -39,7 +41,7 @@ async def _mock_receive() -> Message:  # pragma: no cover
     return {"type": "http.request", "body": b"", "more_body": False}
 
 
-def _make_send(sent: list[Message]):
+def _make_send(sent: list[Message]) -> Callable[[Message], Coroutine[Any, Any, None]]:
     async def mock_send(message: Message) -> None:
         sent.append(message)
 
@@ -52,13 +54,15 @@ def _extract_session_id(sent_messages: list[Message]) -> str:
             for name, value in msg.get("headers", []):
                 if name.decode().lower() == MCP_SESSION_ID_HEADER.lower():
                     return value.decode()
-    raise AssertionError("Session ID not found in response headers")
+    raise AssertionError("Session ID not found in response headers")  # pragma: no cover
 
 
-def _make_blocking_run(stop_event: anyio.Event):
+def _make_blocking_run(
+    stop_event: anyio.Event,
+) -> Callable[..., Coroutine[Any, Any, None]]:
     """Create a mock app.run that blocks until stop_event is set."""
 
-    async def blocking_run(*args, **kwargs):  # type: ignore[no-untyped-def]
+    async def blocking_run(*args: Any, **kwargs: Any) -> None:
         await stop_event.wait()
 
     return blocking_run
@@ -277,7 +281,7 @@ async def test_run_server_exits_promptly_after_terminate():
 
     original_run = app.run
 
-    async def instrumented_run(*args, **kwargs):  # type: ignore[no-untyped-def]
+    async def instrumented_run(*args: Any, **kwargs: Any) -> None:
         try:
             await original_run(*args, **kwargs)
         finally:
@@ -360,7 +364,7 @@ async def test_run_server_finally_block_runs_after_terminate():
     lifecycle_events: list[str] = []
     original_run = app.run
 
-    async def instrumented_run(*args, **kwargs):  # type: ignore[no-untyped-def]
+    async def instrumented_run(*args: Any, **kwargs: Any) -> None:
         lifecycle_events.append("run_entered")
         try:
             await original_run(*args, **kwargs)