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 85

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:02 -08:00 • committed by GitHub
parent 2f24a7ebf2
commit 4a950c1975
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

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