From e6a6e3ec359fb0e81a4fd223b593c29386883f8c Mon Sep 17 00:00:00 2001 From: yucheng Date: Sun, 27 Sep 2026 02:18:11 +0000 Subject: [PATCH] style(mcp): format the new connect sign-in preflight tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/_experimental/mcp_server/test_mcp_server.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index bfaa1767456..d642c5de2c8 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -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: