From 49fce3404b1a713318c1f24a7067829347f1093e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 4 Mar 2026 21:19:18 -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 cc7f2778eec..a7391666cda 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 @@ -475,8 +475,9 @@ async def test_check_byok_credential_missing_credential(): mock_prisma = MagicMock() with patch( - "litellm.proxy._experimental.mcp_server.db.has_user_credential", - new=AsyncMock(return_value=False), + "litellm.proxy._experimental.mcp_server.db.get_user_credential", + new=AsyncMock(return_value=None), + ), patch("litellm.proxy.proxy_server.prisma_client", mock_prisma): ), patch("litellm.proxy.proxy_server.prisma_client", mock_prisma): with pytest.raises(HTTPException) as exc_info: await _check_byok_credential(server, user_auth)