scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test_async_log_cache_hit_on_callbacks

Browse source
This commit is contained in:
Ishaan Jaff 2025-09-08 17:15:53 -07:00
parent 4af67aacb6
commit c7f9be6803
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/local_testing/test_caching_handler.py
View file

@ -134,6 +134,7 @@ async def test_async_log_cache_hit_on_callbacks():
mock_logging_obj = MagicMock()
mock_logging_obj.async_success_handler = AsyncMock()
mock_logging_obj.success_handler = MagicMock()
mock_logging_obj.handle_sync_success_callbacks_for_async_calls = MagicMock()
cached_result = "Mocked cached result"
start_time = datetime.now()
@ -156,14 +157,14 @@ async def test_async_log_cache_hit_on_callbacks():
# Assertions
mock_logging_obj.async_success_handler.assert_called_once_with(
cached_result, start_time, end_time, cache_hit
result=cached_result, start_time=start_time, end_time=end_time, cache_hit=cache_hit
)
# Wait for the thread to complete
await asyncio.sleep(0.5)
mock_logging_obj.success_handler.assert_called_once_with(
cached_result, start_time, end_time, cache_hit
mock_logging_obj.handle_sync_success_callbacks_for_async_calls.assert_called_once_with(
result=cached_result, start_time=start_time, end_time=end_time, cache_hit=cache_hit
)