scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

test(responses): cover store failure and missing hook paths for background responses

Browse source
This commit is contained in:
Devin AI 2026-07-10 15:24:18 +00:00
parent 170d348fb0
commit 6781dd6008
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/response_api_endpoints/test_endpoints.py
View file

@ -834,3 +834,69 @@ class TestStoreBackgroundResponseInManagedObjects:
)
managed_files_obj.store_unified_object_id.assert_not_awaited()
@pytest.mark.asyncio
async def test_store_failure_is_swallowed(self):
import sys
import types
from litellm.proxy.response_api_endpoints.endpoints import (
_store_background_response_in_managed_objects,
)
fake_module = types.ModuleType(
"litellm_enterprise.proxy.hooks.managed_files"
)
fake_module._PROXY_LiteLLMManagedFiles = type("_FakeManagedFiles", (), {})
managed_files_obj = MagicMock()
managed_files_obj.store_unified_object_id = AsyncMock(
side_effect=RuntimeError("db down")
)
proxy_logging_obj = MagicMock()
proxy_logging_obj.get_proxy_hook = MagicMock(return_value=managed_files_obj)
with patch.dict(
sys.modules,
{"litellm_enterprise.proxy.hooks.managed_files": fake_module},
):
# Storage failures must not propagate out of the endpoint.
await _store_background_response_in_managed_objects(
response=self._response(),
proxy_logging_obj=proxy_logging_obj,
llm_router=MagicMock(),
user_api_key_dict=MagicMock(),
)
managed_files_obj.store_unified_object_id.assert_awaited_once()
@pytest.mark.asyncio
async def test_noop_when_hook_missing(self):
import sys
import types
from litellm.proxy.response_api_endpoints.endpoints import (
_store_background_response_in_managed_objects,
)
fake_module = types.ModuleType(
"litellm_enterprise.proxy.hooks.managed_files"
)
fake_module._PROXY_LiteLLMManagedFiles = type("_FakeManagedFiles", (), {})
proxy_logging_obj = MagicMock()
proxy_logging_obj.get_proxy_hook = MagicMock(return_value=None)
with patch.dict(
sys.modules,
{"litellm_enterprise.proxy.hooks.managed_files": fake_module},
):
await _store_background_response_in_managed_objects(
response=self._response(),
proxy_logging_obj=proxy_logging_obj,
llm_router=MagicMock(),
user_api_key_dict=MagicMock(),
)
proxy_logging_obj.get_proxy_hook.assert_called_once_with("managed_files")