98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
|
"""Unit tests for autoscalerctl CLI argument and display behavior."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from nix_builder_autoscaler import cli
|
||
|
|
from nix_builder_autoscaler.cli import _parse_args, _print_status_summary
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_args_without_command_prints_help_and_exits_zero(
|
||
|
|
capsys: pytest.CaptureFixture[str],
|
||
|
|
) -> None:
|
||
|
|
with pytest.raises(SystemExit) as exc:
|
||
|
|
_parse_args([])
|
||
|
|
assert exc.value.code == 0
|
||
|
|
captured = capsys.readouterr()
|
||
|
|
assert "Autoscaler CLI" in captured.out
|
||
|
|
assert "status" in captured.out
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_args_json_status() -> None:
|
||
|
|
args = _parse_args(["--json", "status"])
|
||
|
|
assert args.command == "status"
|
||
|
|
assert args.json is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_args_bulk_commands() -> None:
|
||
|
|
assert _parse_args(["drain-all"]).command == "drain-all"
|
||
|
|
assert _parse_args(["unquarantine-all"]).command == "unquarantine-all"
|
||
|
|
|
||
|
|
|
||
|
|
def test_print_status_summary_renders_metrics_table(capsys: pytest.CaptureFixture[str]) -> None:
|
||
|
|
_print_status_summary(
|
||
|
|
{
|
||
|
|
"slots": {
|
||
|
|
"total": 4,
|
||
|
|
"ready": 1,
|
||
|
|
"launching": 1,
|
||
|
|
"booting": 1,
|
||
|
|
"binding": 0,
|
||
|
|
"terminating": 0,
|
||
|
|
"empty": 1,
|
||
|
|
"error": 0,
|
||
|
|
},
|
||
|
|
"reservations": {"pending": 2, "ready": 1, "failed": 0},
|
||
|
|
"ec2": {"api_ok": True},
|
||
|
|
"haproxy": {"socket_ok": True},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
assert "metric" in out
|
||
|
|
assert "slots.total" in out
|
||
|
|
assert "reservations.pending" in out
|
||
|
|
assert "haproxy.socket_ok" in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_bulk_drain_only_targets_ready_slots(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
def _fake_request(socket_path: str, method: str, path: str, body=None): # noqa: ANN001
|
||
|
|
assert socket_path == "/tmp/sock"
|
||
|
|
if method == "GET" and path == "/v1/slots":
|
||
|
|
return 200, [
|
||
|
|
{"slot_id": "slot001", "state": "ready"},
|
||
|
|
{"slot_id": "slot002", "state": "booting"},
|
||
|
|
]
|
||
|
|
if method == "POST" and path == "/v1/admin/drain" and body == {"slot_id": "slot001"}:
|
||
|
|
return 200, {"state": "draining"}
|
||
|
|
raise AssertionError(f"unexpected request: {method} {path} {body}")
|
||
|
|
|
||
|
|
monkeypatch.setattr(cli, "_uds_request", _fake_request)
|
||
|
|
summary = cli._bulk_slot_action("/tmp/sock", "drain")
|
||
|
|
assert summary["matched"] == 1
|
||
|
|
assert summary["attempted"] == 1
|
||
|
|
assert summary["succeeded"] == 1
|
||
|
|
assert summary["failed"] == 0
|
||
|
|
assert summary["skipped"] == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_bulk_unquarantine_only_targets_error_slots(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
def _fake_request(socket_path: str, method: str, path: str, body=None): # noqa: ANN001
|
||
|
|
assert socket_path == "/tmp/sock"
|
||
|
|
if method == "GET" and path == "/v1/slots":
|
||
|
|
return 200, [
|
||
|
|
{"slot_id": "slot001", "state": "error"},
|
||
|
|
{"slot_id": "slot002", "state": "ready"},
|
||
|
|
]
|
||
|
|
if method == "POST" and path == "/v1/admin/unquarantine" and body == {"slot_id": "slot001"}:
|
||
|
|
return 200, {"state": "empty"}
|
||
|
|
raise AssertionError(f"unexpected request: {method} {path} {body}")
|
||
|
|
|
||
|
|
monkeypatch.setattr(cli, "_uds_request", _fake_request)
|
||
|
|
summary = cli._bulk_slot_action("/tmp/sock", "unquarantine")
|
||
|
|
assert summary["matched"] == 1
|
||
|
|
assert summary["attempted"] == 1
|
||
|
|
assert summary["succeeded"] == 1
|
||
|
|
assert summary["failed"] == 0
|
||
|
|
assert summary["skipped"] == 1
|