phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: improve debugging for test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 19:40:49 -07:00
parent dce7c459f8
commit 4ba576724c
4 changed files with 19 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_key_generate_prisma.py
View file

@ -303,6 +303,7 @@ def test_call_with_invalid_key(prisma_client):
def test_call_with_invalid_model(prisma_client):
litellm.set_verbose = True
# 3. Make a call to a key with an invalid model - expect to fail
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
@ -326,6 +327,11 @@ def test_call_with_invalid_model(prisma_client):
request.body = return_body
# use generated key to auth in
print(
"Bearer token being sent to user_api_key_auth() - {}".format(
bearer_token
)
)
result = await user_api_key_auth(request=request, api_key=bearer_token)
pytest.fail(f"This should have failed!. IT's an invalid model")