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

(fix) test proxy

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 22:11:11 -08:00
parent d3bbf5bdff
commit 5e7b9658e3
2 changed files with 18 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_proxy_custom_auth.py
View file

@ -25,11 +25,14 @@ from litellm.proxy.proxy_server import router, save_worker_config, initialize #
# Make sure the fixture returns TestClient(app)
@pytest.fixture(scope="function")
def client():
from litellm.proxy.proxy_server import cleanup_router_config_variables
cleanup_router_config_variables()
filepath = os.path.dirname(os.path.abspath(__file__))
config_fp = f"{filepath}/test_configs/test_config_custom_auth.yaml"
# initialize can get run in parallel, it sets specific variables for the fast api app, sinc eit gets run in parallel different tests use the wrong variables
initialize(config=config_fp)
app = FastAPI()
initialize(config=config_fp)
app.include_router(router) # Include your router in the test app
return TestClient(app)