diff --git a/tests/test_litellm/proxy/management_endpoints/test_delete_verification_tokens_failed.py b/tests/test_litellm/proxy/management_endpoints/test_delete_verification_tokens_failed.py index 63e584e49bc..388d865de42 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_delete_verification_tokens_failed.py +++ b/tests/test_litellm/proxy/management_endpoints/test_delete_verification_tokens_failed.py @@ -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", diff --git a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py index 653592fc13f..2bafe0fcecc 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py @@ -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 = (