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

test(test_proxy_startup.py): separate tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-08 17:58:37 +05:30 • committed by ishaan-jaff
parent a37b40cdb9
commit 834d3362d9
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_proxy_startup.py
View file

@ -22,7 +22,7 @@ from litellm.proxy.proxy_server import (
llm_model_list llm_model_list
) )
def test_proxy_gunicorn_startup(): def test_proxy_gunicorn_startup_direct_config():
""" """
gunicorn startup requires the config to be passed in via environment variables gunicorn startup requires the config to be passed in via environment variables
@ -36,6 +36,11 @@ def test_proxy_gunicorn_startup():
config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml" config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml"
os.environ["WORKER_CONFIG"] = config_fp os.environ["WORKER_CONFIG"] = config_fp
asyncio.run(startup_event()) asyncio.run(startup_event())
def test_proxy_gunicorn_startup_config_dict():
filepath = os.path.dirname(os.path.abspath(__file__))
# test with worker_config = config yaml
config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml"
# test with worker_config = dict # test with worker_config = dict
worker_config = {"config": config_fp} worker_config = {"config": config_fp}
os.environ["WORKER_CONFIG"] = json.dumps(worker_config) os.environ["WORKER_CONFIG"] = json.dumps(worker_config)