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

(test) proxy - pytest correct usage

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 21:30:02 -08:00
parent 3e7604420e
commit f07da60593
2 changed files with 21 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_proxy_custom_auth.py
View file

@ -23,18 +23,19 @@ from litellm.proxy.proxy_server import router, save_worker_config, initialize #
# Here you create a fixture that will be used by your tests
# Make sure the fixture returns TestClient(app)
def get_client(config_fp):
@pytest.fixture(scope="function")
def client():
filepath = os.path.dirname(os.path.abspath(__file__))
config_fp = f"{filepath}/test_configs/{config_fp}"
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()
app.include_router(router) # Include your router in the test app
return TestClient(app)
def test_custom_auth():
def test_custom_auth(client):
try:
client = get_client(config_fp="test_config_custom_auth.yaml")
# Your test data
test_data = {
"model": "openai-model",