scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(cli): pin the up alias port forwarding and the removed-settings stop message

Browse source
This commit is contained in:
mateo-berri 2026-09-17 15:55:50 -07:00
parent 4371ddb620
commit d3963e5d63
1 changed files with 25 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/test_litellm/proxy/client/cli/autoroute/test_commands.py
View file

@ -3,6 +3,7 @@ import socket
import stat
from typing import Optional
import pytest
import yaml
from click.testing import CliRunner
@ -389,9 +390,15 @@ class TestStartCommand:
written_config = yaml.safe_load(config_path.read_text())
assert written_config["general_settings"]["master_key"] == "fresh-minted-key"
def test_port_override_reaches_settings_launch_and_pid_record(self, monkeypatch, tmp_path):
"""A --port override must flow to every consumer of the port; a hardcoded default in any
one of them would leave the patched settings pointing somewhere the proxy is not."""
@pytest.mark.parametrize(
("command", "leading_args"),
[(start, []), (autoroute_group, ["start"]), (autoroute_group, ["up"])],
ids=["start", "group start", "deprecated up alias"],
)
def test_port_override_reaches_settings_launch_and_pid_record(self, monkeypatch, tmp_path, command, leading_args):
"""A --port override must flow to every consumer of the port, through the deprecated `up`
alias too; a hardcoded default in any one of them would leave the patched settings pointing
somewhere the proxy is not."""
config_path, _log_path, claude_settings_path, _backup_path, pid_record_path = _patch_paths(
monkeypatch, tmp_path
)
@ -420,7 +427,7 @@ class TestStartCommand:
monkeypatch.setattr("threading.Event.wait", fake_wait)
result = self.runner.invoke(start, ["--port", "6111"])
result = self.runner.invoke(command, [*leading_args, "--port", "6111"])
assert result.exit_code == 0, result.output
assert captured["env"]["ANTHROPIC_BASE_URL"] == "http://127.0.0.1:6111"
@ -501,6 +508,20 @@ class TestStopCommand:
assert not backup_path.exists()
assert json.loads(claude_settings_path.read_text()) == original_settings
def test_removes_settings_that_did_not_exist_before_start(self, monkeypatch, tmp_path):
_config_path, _log_path, claude_settings_path, backup_path, _pid_record_path = _patch_paths(
monkeypatch, tmp_path
)
write_backup(ClaudeBackupRecord(existed=False, content=None), backup_path)
claude_settings_path.write_text(json.dumps({"env": {"ANTHROPIC_AUTH_TOKEN": "fixed-master-key"}}))
result = self.runner.invoke(stop)
assert result.exit_code == 0, result.output
assert f"Removed {claude_settings_path} (it did not exist before `lite autoroute start`)." in result.output
assert not claude_settings_path.exists()
assert not backup_path.exists()
def test_is_a_clean_no_op_when_nothing_is_running_and_no_backup_exists(self, monkeypatch, tmp_path):
_config_path, _log_path, claude_settings_path, _backup_path, _pid_record_path = _patch_paths(
monkeypatch, tmp_path