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

fix test_key_name_null

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-16 21:37:31 -07:00
parent 5393930701
commit 860d97c342
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_key_generate_prisma.py
View file

@ -1587,11 +1587,12 @@ async def test_key_name_null(prisma_client):
""" """
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client) setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234") setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
setattr(litellm.proxy.proxy_server, "general_settings", {"allow_user_auth": False}) os.environ["DISABLE_KEY_NAME"] = "True"
await litellm.proxy.proxy_server.prisma_client.connect() await litellm.proxy.proxy_server.prisma_client.connect()
try: try:
request = GenerateKeyRequest() request = GenerateKeyRequest()
key = await generate_key_fn(request) key = await generate_key_fn(request)
print("generated key=", key)
generated_key = key.key generated_key = key.key
result = await info_key_fn(key=generated_key) result = await info_key_fn(key=generated_key)
print("result from info_key_fn", result) print("result from info_key_fn", result)
@ -1599,6 +1600,8 @@ async def test_key_name_null(prisma_client):
except Exception as e: except Exception as e:
print("Got Exception", e) print("Got Exception", e)
pytest.fail(f"Got exception {e}") pytest.fail(f"Got exception {e}")
finally:
os.environ["DISABLE_KEY_NAME"] = "False"
@pytest.mark.asyncio() @pytest.mark.asyncio()