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

fix test: configure async_delete_cache as AsyncMock and assert cleanup called

Browse source
This commit is contained in:
Ishaan Jaffer 2026-03-06 16:26:58 -08:00
parent faa8e29cd9
commit 579078f468
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/management_endpoints/test_ui_sso.py
View file

@ -3801,6 +3801,7 @@ class TestPKCEFunctionality:
# Cache returns an integer — unexpected format
mock_cache = MagicMock()
mock_cache.async_get_cache = AsyncMock(return_value=12345)
mock_cache.async_delete_cache = AsyncMock()
mock_request = MagicMock(spec=Request)
mock_request.query_params = {"state": "bad_format_state"}
@ -3917,6 +3918,7 @@ class TestPKCEFunctionality:
# Cache returns an integer — unexpected format
mock_cache = MagicMock()
mock_cache.async_get_cache = AsyncMock(return_value=12345)
mock_cache.async_delete_cache = AsyncMock()
mock_request = MagicMock(spec=Request)
mock_request.query_params = {"state": "bad_format_non_strict"}
@ -3948,6 +3950,8 @@ class TestPKCEFunctionality:
for r in caplog.records
if r.levelno >= logging.WARNING
), f"Expected a format/cache warning. Records: {[r.message for r in caplog.records]}"
# Verify cleanup was attempted for the corrupt/stale cache entry
mock_cache.async_delete_cache.assert_called_once()
@pytest.mark.asyncio
async def test_pkce_legacy_string_cache_format_backward_compat(self):