From 4a950c1975a469f8d505eff227b87134d1970deb Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 4 Mar 2026 21:19:02 -0800 Subject: [PATCH] Update tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> --- .../_experimental/mcp_server/test_byok_oauth_endpoints.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py index df4e092d9df..cc7f2778eec 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py @@ -508,8 +508,9 @@ async def test_check_byok_credential_has_credential(): mock_prisma = MagicMock() with patch( - "litellm.proxy._experimental.mcp_server.db.has_user_credential", - new=AsyncMock(return_value=True), + "litellm.proxy._experimental.mcp_server.db.get_user_credential", + new=AsyncMock(return_value="some-credential-value"), + ), patch("litellm.proxy.proxy_server.prisma_client", mock_prisma): ), patch("litellm.proxy.proxy_server.prisma_client", mock_prisma): # Should not raise await _check_byok_credential(server, user_auth)