scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

fix(tests): clear app.openapi_schema in TestSwaggerChatCompletions setup/teardown

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 11:27:39 -08:00
parent 8bf5601ceb
commit 58a9f56f83
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/proxy/test_swagger_chat_completions.py
View file

@ -17,6 +17,12 @@ from litellm.proxy.proxy_server import app
class TestSwaggerChatCompletions:
"""Test suite for validating /chat/completions schema in Swagger documentation."""
def setup_method(self):
app.openapi_schema = None
def teardown_method(self):
app.openapi_schema = None
@pytest.fixture
def client(self):
"""FastAPI test client for the proxy server."""
@ -315,7 +321,8 @@ class TestSwaggerChatCompletions:
This ensures Swagger UI works correctly with reverse proxies and subpath deployments.
"""
from unittest.mock import patch
from litellm.proxy.proxy_server import get_openapi_schema, custom_openapi, app
from litellm.proxy.proxy_server import app, custom_openapi, get_openapi_schema
# Test cases: (server_root_path, expected_servers_url)
# Note: empty string is falsy in Python, so servers won't be set