From 7918f41aca7a2e4192308e9cbe405752cecc39ba Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 12 Jul 2024 18:45:01 -0700 Subject: [PATCH] test expired key raises correct exception --- litellm/tests/test_key_generate_prisma.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 48e3689dd..0d1f488e1 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -941,6 +941,8 @@ def test_generate_and_call_with_expired_key(prisma_client): print("Got Exception", e) print(e.message) assert "Authentication Error" in e.message + assert e.type == ProxyErrorTypes.expired_key + pass