b49a0abf39
Some checks failed
Architecture Lint / Linter Tests (pull_request) Successful in 9s
PR Checklist / pr-checklist (pull_request) Failing after 1m17s
Smoke Test / smoke (pull_request) Failing after 7s
Validate Config / YAML Lint (pull_request) Failing after 6s
Validate Config / JSON Validate (pull_request) Successful in 6s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 8s
Validate Config / Shell Script Lint (pull_request) Successful in 15s
Validate Config / Cron Syntax Check (pull_request) Successful in 5s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 5s
Validate Config / Playbook Schema Validation (pull_request) Successful in 8s
Architecture Lint / Lint Repository (pull_request) Failing after 8s
96 lines
3.6 KiB
Python
96 lines
3.6 KiB
Python
"""Tests for scripts/self_healing.py safe CLI behavior."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import subprocess
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
REPO_ROOT = Path(__file__).parent.parent
|
|
spec = importlib.util.spec_from_file_location("self_healing", REPO_ROOT / "scripts" / "self_healing.py")
|
|
sh = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(sh)
|
|
|
|
|
|
class TestMainCli:
|
|
def test_help_exits_without_running_healer(self, monkeypatch, capsys):
|
|
healer_cls = MagicMock()
|
|
monkeypatch.setattr(sh, "SelfHealer", healer_cls)
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
sh.main(["--help"])
|
|
|
|
assert excinfo.value.code == 0
|
|
healer_cls.assert_not_called()
|
|
out = capsys.readouterr().out
|
|
assert "--execute" in out
|
|
assert "--help-safe" in out
|
|
|
|
def test_help_safe_exits_without_running_healer(self, monkeypatch, capsys):
|
|
healer_cls = MagicMock()
|
|
monkeypatch.setattr(sh, "SelfHealer", healer_cls)
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
sh.main(["--help-safe"])
|
|
|
|
assert excinfo.value.code == 0
|
|
healer_cls.assert_not_called()
|
|
out = capsys.readouterr().out
|
|
assert "DRY-RUN" in out
|
|
assert "--confirm-kill" in out
|
|
|
|
def test_default_invocation_runs_in_dry_run_mode(self, monkeypatch):
|
|
healer = MagicMock()
|
|
healer_cls = MagicMock(return_value=healer)
|
|
monkeypatch.setattr(sh, "SelfHealer", healer_cls)
|
|
|
|
sh.main([])
|
|
|
|
healer_cls.assert_called_once_with(dry_run=True, confirm_kill=False, yes=False)
|
|
healer.run.assert_called_once_with()
|
|
|
|
def test_execute_flag_disables_dry_run(self, monkeypatch):
|
|
healer = MagicMock()
|
|
healer_cls = MagicMock(return_value=healer)
|
|
monkeypatch.setattr(sh, "SelfHealer", healer_cls)
|
|
|
|
sh.main(["--execute", "--yes", "--confirm-kill"])
|
|
|
|
healer_cls.assert_called_once_with(dry_run=False, confirm_kill=True, yes=True)
|
|
healer.run.assert_called_once_with()
|
|
|
|
def test_real_default_dry_run_path_completes(self, monkeypatch, capsys):
|
|
class FakeExecutor:
|
|
def __init__(self):
|
|
self.calls = []
|
|
|
|
def run_script(self, host, command, *, local=False, timeout=None):
|
|
self.calls.append((host, command, local, timeout))
|
|
if command.startswith("df -h /"):
|
|
return subprocess.CompletedProcess(command, 0, stdout="42\n", stderr="")
|
|
if command.startswith("free -m"):
|
|
return subprocess.CompletedProcess(command, 0, stdout="12.5\n", stderr="")
|
|
if command.startswith("ps aux"):
|
|
return subprocess.CompletedProcess(command, 0, stdout="", stderr="")
|
|
raise AssertionError(f"unexpected command: {command}")
|
|
|
|
fake_executor = FakeExecutor()
|
|
monkeypatch.setattr(sh, "FLEET", {"mac": {"ip": "127.0.0.1", "port": 8080}})
|
|
monkeypatch.setattr(sh.requests, "get", lambda url, timeout: object())
|
|
monkeypatch.setattr(sh, "VerifiedSSHExecutor", lambda: fake_executor)
|
|
|
|
sh.main([])
|
|
|
|
out = capsys.readouterr().out
|
|
assert "Starting self-healing cycle (DRY-RUN mode)." in out
|
|
assert "Auditing mac..." in out
|
|
assert "Cycle complete." in out
|
|
assert fake_executor.calls == [
|
|
("127.0.0.1", "df -h / | tail -1 | awk '{print $5}' | sed 's/%//'", True, 15),
|
|
("127.0.0.1", "free -m | awk '/^Mem:/{print $3/$2 * 100}'", True, 15),
|
|
("127.0.0.1", "ps aux --sort=-%cpu | awk 'NR>1 && $3>80 {print $2, $11, $3}'", True, 15),
|
|
]
|