phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) invalid key auth

Browse source
This commit is contained in:
ishaan-jaff 2024-01-16 17:40:32 -08:00
parent adc921f0be
commit 2c79b8fc1b
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_key_generate_dynamodb.py
View file

@ -31,6 +31,7 @@ from litellm.proxy.proxy_server import new_user, user_api_key_auth
from litellm.proxy._types import NewUserRequest, DynamoDBArgs from litellm.proxy._types import NewUserRequest, DynamoDBArgs
from litellm.proxy.utils import DBClient from litellm.proxy.utils import DBClient
from starlette.datastructures import URL
db_args = { db_args = {
"ssl_verify": False, "ssl_verify": False,
@ -63,7 +64,6 @@ def test_generate_and_call_with_valid_key():
generated_key = key.key generated_key = key.key
bearer_token = "Bearer " + generated_key bearer_token = "Bearer " + generated_key
from starlette.datastructures import URL
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})
request._url = URL(url="/chat/completions") request._url = URL(url="/chat/completions")
@ -90,16 +90,19 @@ def test_call_with_invalid_key():
bearer_token = "Bearer " + generated_key bearer_token = "Bearer " + generated_key
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})
request._url = URL(url="/chat/completions")
# use generated key to auth in # use generated key to auth in
result = await user_api_key_auth(request=request, api_key=bearer_token) result = await user_api_key_auth(request=request, api_key=bearer_token)
print("result from user auth with new key", result) pytest.fail(f"This should have failed!")
asyncio.run(test()) asyncio.run(test())
# result = user_auth(ValidRequest(key)) # result = user_auth(ValidRequest(key))
# assert result is True # assert result is True
except Exception as e: except Exception as e:
pytest.fail(f"An exception occurred - {str(e)}") print("Got Exception", e)
print(str(e))
pass
# def test_call_with_invalid_model(): # def test_call_with_invalid_model():