scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 156

test(cli): verify --setup forwards --detailed_debug to the wizard

Browse source
This commit is contained in:
Devin AI 2026-07-11 16:32:11 +00:00
parent 58bdd3dbb3
commit f7e9ed3ee2
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/test_litellm/proxy/test_proxy_cli.py
View file

@ -1907,3 +1907,24 @@ class TestWorkerStartupHooks:
assert _dummy_hook_called is True, "First hook was not called"
assert _dummy_async_hook_called is True, "Second hook was not called"
@pytest.mark.parametrize(
"cli_args, expected_debug",
[
(["--setup"], False),
(["--setup", "--detailed_debug"], True),
],
)
def test_setup_flag_forwards_detailed_debug(cli_args, expected_debug):
"""`litellm --setup` must run the wizard, and --detailed_debug must reach it
so the wizard can enable debug logging (previously it returned first)."""
from click.testing import CliRunner
from litellm.proxy.proxy_cli import run_server
with patch("litellm.setup_wizard.run_setup_wizard") as mock_wizard:
result = CliRunner().invoke(run_server, cli_args)
assert result.exit_code == 0, f"exit_code={result.exit_code}, output={result.output}"
mock_wizard.assert_called_once_with(debug=expected_debug)