scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

test(proxy): assert the hydration ran in the failure-path test

Browse source 
test_add_deployment_survives_a_non_llm_hydration_failure passed with the fix
reverted. Reverting stops _init_non_llm_objects_in_db from being called at
all, so nothing raised and the test proved nothing.

Record that the override ran and assert it. Both added tests now fail with
litellm/proxy/proxy_server.py reverted to the base.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
Yash Raj Pandey 2026-09-07 16:46:28 -04:00
parent fa73c52f9f
commit 03e4ab7b1c
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/test_proxy_server.py
View file

@ -1104,6 +1104,7 @@ async def test_add_deployment_survives_a_non_llm_hydration_failure():
def __init__(self):
super().__init__()
self.reconcile_ran = False
self.hydration_attempted = False
async def _get_models_from_db(self, prisma_client):
return []
@ -1113,6 +1114,7 @@ async def test_add_deployment_survives_a_non_llm_hydration_failure():
return frozenset()
async def _init_non_llm_objects_in_db(self, prisma_client):
self.hydration_attempted = True
raise RuntimeError("guardrail table unavailable")
proxy_config = FailingHydrationProxyConfig()
@ -1123,6 +1125,7 @@ async def test_add_deployment_survives_a_non_llm_hydration_failure():
)
assert proxy_config.reconcile_ran is True
assert proxy_config.hydration_attempted is True
assert outcome.still_desired == frozenset()