scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

style(mcp): format the new connect sign-in preflight tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-27 02:18:11 +00:00
parent 0099b96ddc
commit e6a6e3ec35
1 changed files with 7 additions and 3 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_server.py
View file

@ -11249,7 +11249,9 @@ class TestConnectChallengeResolver:
)
authenticate: Final = (exc.value.headers or {}).get("WWW-Authenticate") or ""
assert authenticate.startswith(f'Bearer resource_metadata="/.well-known/oauth-protected-resource/mcp/{route_name}"')
assert authenticate.startswith(
f'Bearer resource_metadata="/.well-known/oauth-protected-resource/mcp/{route_name}"'
)
class TestConnectSignInPreflight:
@ -11289,7 +11291,8 @@ class TestConnectSignInPreflight:
server = _catalog_server()
guardrail = _CallerSignInGuardrail(
guardrail_name="sign-in-stub", preflight_result=Rejected("the Entra OBO exchange was rejected (AADSTS70002)")
guardrail_name="sign-in-stub",
preflight_result=Rejected("the Entra OBO exchange was rejected (AADSTS70002)"),
)
litellm.logging_callback_manager.add_litellm_callback(guardrail)
try:
@ -11312,7 +11315,8 @@ class TestConnectSignInPreflight:
server = _catalog_server()
guardrail = _CallerSignInGuardrail(
guardrail_name="sign-in-stub", preflight_result=Unavailable("the Entra token endpoint could not be reached", fail_open=False)
guardrail_name="sign-in-stub",
preflight_result=Unavailable("the Entra token endpoint could not be reached", fail_open=False),
)
litellm.logging_callback_manager.add_litellm_callback(guardrail)
try: