phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-04 18:13:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

updated the test script

Browse source
This commit is contained in:
Omar Abdelwahab 2025-11-04 15:36:09 -08:00
parent c911e9a3c1
commit 5c5f6f7e65
1 changed files with 4 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/integration/responses/test_mcp_authentication.py
View file

@ -33,10 +33,7 @@ def test_mcp_authorization_bearer(compat_client, text_model_id):
"type": "mcp", "type": "mcp",
"server_label": "auth-mcp", "server_label": "auth-mcp",
"server_url": "<FILLED_BY_TEST_RUNNER>", "server_url": "<FILLED_BY_TEST_RUNNER>",
"authorization": { "authorization": f"Bearer {test_token}",
"type": "bearer",
"token": test_token,
},
} }
], ],
mcp_server_info, mcp_server_info,
@ -73,10 +70,7 @@ def test_mcp_authorization_different_token(compat_client, text_model_id):
"type": "mcp", "type": "mcp",
"server_label": "auth2-mcp", "server_label": "auth2-mcp",
"server_url": "<FILLED_BY_TEST_RUNNER>", "server_url": "<FILLED_BY_TEST_RUNNER>",
"authorization": { "authorization": f"Bearer {test_token}",
"type": "bearer",
"token": test_token,
},
} }
], ],
mcp_server_info, mcp_server_info,
@ -98,7 +92,7 @@ def test_mcp_authorization_different_token(compat_client, text_model_id):
def test_mcp_authorization_fallback_to_headers(compat_client, text_model_id): def test_mcp_authorization_fallback_to_headers(compat_client, text_model_id):
"""Test that authorization parameter doesn't override existing headers.""" """Test that authorization parameter doesn't override existing Authorization header."""
if not isinstance(compat_client, LlamaStackAsLibraryClient): if not isinstance(compat_client, LlamaStackAsLibraryClient):
pytest.skip("in-process MCP server is only supported in library client") pytest.skip("in-process MCP server is only supported in library client")
@ -112,10 +106,7 @@ def test_mcp_authorization_fallback_to_headers(compat_client, text_model_id):
"server_label": "headers-mcp", "server_label": "headers-mcp",
"server_url": "<FILLED_BY_TEST_RUNNER>", "server_url": "<FILLED_BY_TEST_RUNNER>",
"headers": {"Authorization": f"Bearer {test_token}"}, "headers": {"Authorization": f"Bearer {test_token}"},
"authorization": { "authorization": "Bearer should-not-override",
"type": "bearer",
"token": "should-not-override",
},
} }
], ],
mcp_server_info, mcp_server_info,