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

(ci/cd) proxy:test_add_new_key

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 22:53:03 +05:30
parent 40aaac69cc
commit ae54e6d8b0
1 changed files with 4 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_proxy_server_keys.py
View file

@ -52,13 +52,6 @@ save_worker_config(
save=False, save=False,
use_queue=False, use_queue=False,
) )
app = FastAPI()
app.include_router(router) # Include your router in the test app
@app.on_event("startup")
async def wrapper_startup_event():
await startup_event()
import asyncio import asyncio
@ -87,8 +80,10 @@ def client():
cleanup_router_config_variables() # rest proxy before test cleanup_router_config_variables() # rest proxy before test
asyncio.run(initialize(config=config_fp, debug=True)) asyncio.run(initialize(config=config_fp, debug=True))
with TestClient(app) as client: app = FastAPI()
yield client app.include_router(router) # Include your router in the test app
return TestClient(app)
def test_add_new_key(client): def test_add_new_key(client):