From 180e72d15d0afc613c56a4060caea37edc7147ff Mon Sep 17 00:00:00 2001 From: joereyna Date: Thu, 12 Mar 2026 12:22:18 -0700 Subject: [PATCH] fix: add oauth2_flow=client_credentials to MCPServer test helper --- .../proxy/_experimental/mcp_server/test_oauth2_token_cache.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_oauth2_token_cache.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_oauth2_token_cache.py index 55735dca98..02b4ba6c99 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_oauth2_token_cache.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_oauth2_token_cache.py @@ -29,6 +29,7 @@ def _server(**overrides) -> MCPServer: client_id="cid", client_secret="csec", token_url="https://auth.example.com/token", + oauth2_flow="client_credentials", ) defaults.update(overrides) return MCPServer(**defaults) @@ -103,6 +104,7 @@ async def test_falls_back_to_static_token(): client_id=None, client_secret=None, token_url=None, + oauth2_flow=None, authentication_token="static-tok-xyz", ) result = await resolve_mcp_auth(server) @@ -115,7 +117,7 @@ def test_needs_user_oauth_token_property(): assert _server().needs_user_oauth_token is False # OAuth2 without credentials → needs per-user token - assert _server(client_id=None, client_secret=None, token_url=None).needs_user_oauth_token is True + assert _server(client_id=None, client_secret=None, token_url=None, oauth2_flow=None).needs_user_oauth_token is True # Non-OAuth2 → never needs user OAuth token assert _server(auth_type=MCPAuth.bearer_token).needs_user_oauth_token is False