scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: add missing patch test assertions for database_failure and no_prisma_client scenarios

Browse source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Harshit28j 2026-03-16 17:22:14 +05:30
parent 8d4b5ed571
commit 69f5784875
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/guardrails/test_guardrail_endpoints.py
View file

@ -951,6 +951,10 @@ async def test_patch_guardrail_endpoint(
assert mock_guardrail_registry.update_guardrail_in_db.call_count == 2
# Verify rollback attempted in-memory re-sync with old config
assert mock_in_memory_handler.sync_guardrail_from_db.call_count == 2
elif scenario == "database_failure":
assert "Database error" in str(exc_info.value.detail)
elif scenario == "no_prisma_client":
assert "Prisma client not initialized" in str(exc_info.value.detail)
else:
result = await patch_guardrail(
"test-guardrail-id", MOCK_PATCH_REQUEST, user_api_key_dict=MOCK_ADMIN_USER