ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: test

Browse Source
This commit is contained in:
Yuta Saito 2025-09-30 07:25:32 +09:00
parent f1f58bd1d1
commit 5359a0d6a6
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/mcp_tests/test_mcp_client_unit.py
View File

@ -44,7 +44,6 @@ class TestMCPClientUnitTests:
headers = client._get_auth_headers()
assert headers == {
"Authorization": "Bearer test_token",
"MCP-Protocol-Version": "2025-06-18",
}
# Basic auth
@ -55,7 +54,6 @@ class TestMCPClientUnitTests:
headers = client._get_auth_headers()
assert headers == {
"Authorization": f"Basic {expected_encoded}",
"MCP-Protocol-Version": "2025-06-18",
}
# API key
@ -65,7 +63,6 @@ class TestMCPClientUnitTests:
headers = client._get_auth_headers()
assert headers == {
"X-API-Key": "api_key_123",
"MCP-Protocol-Version": "2025-06-18",
}
# Custom authorization header
@ -77,13 +74,12 @@ class TestMCPClientUnitTests:
headers = client._get_auth_headers()
assert headers == {
"Authorization": "Token custom_token",
"MCP-Protocol-Version": "2025-06-18",
}
# No auth
client = MCPClient("http://example.com")
headers = client._get_auth_headers()
assert headers == {"MCP-Protocol-Version": "2025-06-18"}
assert headers == {}
@pytest.mark.asyncio
@patch("litellm.experimental_mcp_client.client.streamablehttp_client")
@ -112,7 +108,6 @@ class TestMCPClientUnitTests:
call_args = mock_transport.call_args
assert call_args[1]["headers"] == {
"Authorization": "Bearer test_token",
"MCP-Protocol-Version": "2025-06-18",
}
# Verify session was initialized