scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 139

test(proxy): cover credential database failures

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-07-21 07:05:03 +00:00
parent 523e55d197
commit f1960ef527
1 changed files with 66 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
tests/test_litellm/proxy/credential_endpoints/test_endpoints.py
View file

@ -454,6 +454,72 @@ async def test_update_credential_handles_database_lookup_failure(_connected_db):
_connected_db.update_by_name.assert_not_awaited()
@pytest.mark.asyncio
async def test_update_credential_handles_missing_database_client(monkeypatch):
import litellm.proxy.proxy_server as proxy_server
monkeypatch.setattr(proxy_server, "prisma_client", None)
result = await endpoints.update_credential(
request=MagicMock(),
fastapi_response=MagicMock(),
credential=UpdateCredentialItem(
credential_info={"access": {"global": True}},
),
credential_name="dest",
user_api_key_dict=_admin(),
)
assert result.code == "500"
@pytest.mark.asyncio
async def test_update_credential_handles_missing_database_credential(_connected_db):
_connected_db.find_by_name = AsyncMock(return_value=None)
_connected_db.update_by_name = AsyncMock()
result = await endpoints.update_credential(
request=MagicMock(),
fastapi_response=MagicMock(),
credential=UpdateCredentialItem(
credential_info={"access": {"global": True}},
),
credential_name="dest",
user_api_key_dict=_admin(),
)
assert result.code == "404"
_connected_db.find_by_name.assert_awaited_once_with("dest")
_connected_db.update_by_name.assert_not_awaited()
@pytest.mark.asyncio
async def test_update_credential_handles_database_write_failure(_connected_db):
db_error = ClientNotConnectedError()
existing = CredentialItem(
credential_name="dest",
credential_values={"langfuse_host": "h"},
credential_info=_LOGGING_INFO,
)
_connected_db.find_by_name = AsyncMock(return_value=existing)
_connected_db.update_by_name = AsyncMock(side_effect=db_error)
result = await endpoints.update_credential(
request=MagicMock(),
fastapi_response=MagicMock(),
credential=UpdateCredentialItem(
credential_info={"access": {"global": True}},
),
credential_name="dest",
user_api_key_dict=_admin(),
)
assert result.code == "500"
assert result.message == str(db_error)
_connected_db.find_by_name.assert_awaited_once_with("dest")
_connected_db.update_by_name.assert_awaited_once()
@pytest.mark.asyncio
async def test_provider_credential_patch_forbidden_for_non_admin(_connected_db, monkeypatch):
"""A team-admin (or any non-admin) cannot PATCH a non-logging credential.