43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
def test_auto_heal_blocks_unsafe_ssh_commands():
|
|
from services.auto_heal_service import AutoHealService
|
|
|
|
svc = AutoHealService()
|
|
|
|
assert svc._is_allowed_ssh_argv(["docker", "ps"]) is True
|
|
assert svc._is_allowed_ssh_argv(["df", "-h"]) is True
|
|
assert svc._is_allowed_ssh_argv(["docker", "restart", "momo-pro-system"]) is False
|
|
assert svc._is_allowed_ssh_argv(["docker", "logs", "momo-db"]) is False
|
|
assert svc._is_allowed_ssh_argv(["bash", "-lc", "whoami"]) is False
|
|
|
|
|
|
def test_auto_heal_code_fix_writes_audit(monkeypatch):
|
|
from services.auto_heal_service import AutoHealResult, AutoHealService
|
|
|
|
svc = AutoHealService()
|
|
calls = []
|
|
monkeypatch.setattr(
|
|
svc,
|
|
"_handle_code_fix",
|
|
lambda playbook, context: AutoHealResult(True, "CODE_FIX", "fixed"),
|
|
)
|
|
monkeypatch.setattr(
|
|
svc,
|
|
"_alert_and_store",
|
|
lambda playbook, context, result, duration_ms=0.0: calls.append((playbook, context, result, duration_ms)),
|
|
)
|
|
monkeypatch.setattr("services.auto_heal_service._find_playbook", lambda error_type: {
|
|
"id": 99,
|
|
"name": "Code fix",
|
|
"action_type": "CODE_FIX",
|
|
"action_params": {},
|
|
"max_retries": 1,
|
|
"cooldown_min": 0,
|
|
})
|
|
monkeypatch.setattr(svc, "_ensure_incident", lambda error_type, context: 123)
|
|
|
|
result = svc.handle_exception("code_exception", {"target_file": "services/example.py"})
|
|
|
|
assert result.success is True
|
|
assert calls
|
|
assert calls[0][2].action == "CODE_FIX"
|