scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

Update tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py

Browse source 
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Ishaan Jaff 2026-03-04 21:19:18 -08:00 • committed by GitHub
parent 4a950c1975
commit 49fce3404b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/proxy/_experimental/mcp_server/test_byok_oauth_endpoints.py
View file

@ -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)