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)