mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
test_get_default_unvicorn_init_args
This commit is contained in:
parent
241c0c6d2a
commit
31a4cb65bf
1 changed files with 6 additions and 1 deletions
|
|
@ -103,7 +103,11 @@ class TestProxyInitializationHelpers:
|
|||
args = ProxyInitializationHelpers._get_default_unvicorn_init_args(
|
||||
"localhost", 8000
|
||||
)
|
||||
assert args["log_config"] is None
|
||||
# When json_logs is True, log_config should be set to the JSON log config dict
|
||||
assert args["log_config"] is not None
|
||||
assert isinstance(args["log_config"], dict)
|
||||
assert "version" in args["log_config"]
|
||||
assert "formatters" in args["log_config"]
|
||||
|
||||
# Test with keepalive_timeout
|
||||
args = ProxyInitializationHelpers._get_default_unvicorn_init_args(
|
||||
|
|
@ -218,6 +222,7 @@ class TestProxyInitializationHelpers:
|
|||
@patch("atexit.register") # 🔥 critical
|
||||
def test_skip_server_startup(self, mock_atexit_register, mock_uvicorn_run):
|
||||
from click.testing import CliRunner
|
||||
|
||||
from litellm.proxy.proxy_cli import run_server
|
||||
|
||||
runner = CliRunner()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue