scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(mcp): add subject_token param to fake_create_mcp_client stubs in hook header tests

Browse source
This commit is contained in:
Ishaan Jaffer 2026-05-01 11:57:16 -07:00
parent 75b4f089f1
commit e7d863cd27
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py
View file

@ -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()