From ae95e4c3f9a4c1d5c43a3daee218aadf685711a5 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 2 Feb 2024 19:57:55 -0800 Subject: [PATCH] (test) user_api_key_auth_cache --- litellm/tests/test_key_generate_prisma.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index f75d20b67..140d638e8 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -119,6 +119,20 @@ def test_generate_and_call_with_valid_key(prisma_client): in user_api_key_cache.in_memory_cache.cache_dict ) + cached_value = user_api_key_cache.in_memory_cache.cache_dict[ + hash_token(generated_key) + ] + + print("cached value=", cached_value) + print("cached token", cached_value.token) + + value_from_prisma = valid_token = await prisma_client.get_data( + token=generated_key, + ) + print("token from prisma", value_from_prisma) + + assert value_from_prisma.token == cached_value.token + request = Request(scope={"type": "http"}) request._url = URL(url="/chat/completions")