fix(test): add auth context mocks to test_streamable_http_mcp_handler_mock

Same fix as test_sse_mcp_handler_mock — the handler calls
extract_mcp_auth_context() before reaching session_manager.handle_request.
Without mocking auth, the handler catches the auth exception and returns
an error response, so handle_request is never called.

Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-02-28 23:50:06 +00:00
parent 863d07011f
commit 92c3c7e9e6
34 changed files with 16 additions and 0 deletions

View file

@ -395,6 +395,7 @@ async def test_mcp_http_transport_tool_not_found():
@pytest.mark.asyncio
async def test_streamable_http_mcp_handler_mock():
"""Test the streamable HTTP MCP handler functionality"""
from litellm.proxy._types import UserAPIKeyAuth
# Mock the session manager and its methods
mock_session_manager = AsyncMock()
@ -413,12 +414,27 @@ async def test_streamable_http_mcp_handler_mock():
mock_receive = AsyncMock()
mock_send = AsyncMock()
# Mock auth context to bypass authentication
mock_auth_result = (
UserAPIKeyAuth(),
None,
None,
{},
{},
[],
)
with patch(
"litellm.proxy._experimental.mcp_server.server._SESSION_MANAGERS_INITIALIZED",
True,
), patch(
"litellm.proxy._experimental.mcp_server.server.session_manager",
mock_session_manager,
), patch(
"litellm.proxy._experimental.mcp_server.server.extract_mcp_auth_context",
new=AsyncMock(return_value=mock_auth_result),
), patch(
"litellm.proxy._experimental.mcp_server.server.set_auth_context",
):
from litellm.proxy._experimental.mcp_server.server import (
handle_streamable_http_mcp,