mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 22:48:35 +00:00
Jonas Neubert
23 lines
586 B
Python
23 lines
586 B
Python
import pytest
|
|
|
|
from litellm.proxy.utils import _get_openapi_url
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"env_vars, expected_url",
|
|
[
|
|
({}, "/openapi.json"), # default case
|
|
({"NO_OPENAPI": "True"}, None), # OpenAPI disabled
|
|
],
|
|
)
|
|
def test_get_openapi_url(monkeypatch, env_vars, expected_url):
|
|
# Clear relevant environment variables
|
|
monkeypatch.delenv("NO_OPENAPI", raising=False)
|
|
|
|
# Set test environment variables
|
|
for key, value in env_vars.items():
|
|
monkeypatch.setenv(key, value)
|
|
|
|
result = _get_openapi_url()
|
|
assert result == expected_url
|