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

fix(test): make cache mocks await-capable for key-delete identity invalidation

Browse source 
delete_verification_tokens now awaits async_delete_cache (via
invalidate_identity_for_token). Tests that patched user_api_key_cache as a
bare MagicMock raised 'MagicMock can't be used in await expression'. Set
async_delete_cache to AsyncMock in the affected key-deletion tests.
This commit is contained in:
Yassin Kortam 2026-06-08 18:22:56 -07:00
parent 80fcb674cf
commit a91448c69c
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/proxy/management_endpoints/test_delete_verification_tokens_failed.py
View file

@ -26,7 +26,6 @@ from litellm.proxy.management_endpoints.key_management_endpoints import (
delete_verification_tokens,
)
# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------
@ -95,6 +94,7 @@ async def test_delete_all_tokens_admin_returns_empty_failed_tokens(monkeypatch):
mock_cache = MagicMock()
mock_cache.delete_cache = MagicMock()
mock_cache.async_delete_cache = AsyncMock()
monkeypatch.setattr(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",
@ -133,6 +133,7 @@ async def test_delete_tokens_non_admin_all_succeed_returns_empty_failed_tokens(
mock_cache = MagicMock()
mock_cache.delete_cache = MagicMock()
mock_cache.async_delete_cache = AsyncMock()
monkeypatch.setattr(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",
@ -184,6 +185,7 @@ async def test_delete_tokens_non_admin_token_not_in_db_returns_failed_tokens(
mock_cache = MagicMock()
mock_cache.delete_cache = MagicMock()
mock_cache.async_delete_cache = AsyncMock()
monkeypatch.setattr(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",
@ -235,6 +237,7 @@ async def test_delete_tokens_admin_partial_db_failure_returns_failed_tokens(
mock_cache = MagicMock()
mock_cache.delete_cache = MagicMock()
mock_cache.async_delete_cache = AsyncMock()
monkeypatch.setattr(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",

3
tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py
View file

@ -4137,6 +4137,9 @@ async def test_delete_verification_tokens_persists_deleted_keys(monkeypatch):
# Mock cache delete_cache method
mock_user_api_key_cache.delete_cache = MagicMock()
# `delete_verification_tokens` now also invalidates the identity cache,
# which awaits `async_delete_cache`.
mock_user_api_key_cache.async_delete_cache = AsyncMock()
mock_create_many = AsyncMock()
mock_prisma_client.db.litellm_deletedverificationtoken.create_many = (