From e7d863cd27f628a8459d950e5dab3627f88b0507 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 1 May 2026 11:57:16 -0700 Subject: [PATCH] test(mcp): add subject_token param to fake_create_mcp_client stubs in hook header tests --- .../mcp_server/test_mcp_hook_extra_headers.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py index 649a08e8744..18b2d5cce22 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py @@ -549,7 +549,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -589,7 +589,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -635,7 +635,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -691,7 +691,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -739,7 +739,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock()