Files
FabledSteward/tests/test_alert_ansible.py
T
bvandeusen 4771d17f6d
CI / lint (push) Successful in 2s
CI / unit (push) Successful in 8s
CI / integration (push) Failing after 2m16s
feat(alerts): run an Ansible playbook on alert firing (task 250)
Rebuilds the deleted NUT/UPS automation as a general alert action: any metric
can drive a playbook run on transition-to-firing.

- models/alerts.py + migration 0014: AlertRule.ansible_action (JSON, admin-only,
  reuses the #546 param shape); AlertEvent.ansible_run_id links a firing event to
  the run it triggered
- core/alerts.py: pure alert_extra_vars() injects steward_alert_* context; on
  ('firing', event) with an action set, schedule _run_ansible_action (deferred
  after commit, same pattern as notifications) — fires once per transition,
  consecutive_failures_required is the debounce; system-triggered AnsibleRun
- alerts/routes.py: admin-only parse/validate of the action (source must be
  configured, playbook must exist); operators keep editing rules, action preserved
- rules_form.html: admin-only 'On firing -> run a playbook' section
- tests: unit for alert_extra_vars; integration drives record_metric to firing
  and asserts a system AnsibleRun ran the playbook with the injected var and the
  event linked to it

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-02 14:22:29 -04:00

31 lines
1.2 KiB
Python

"""Unit tests for alert -> Ansible action context vars (task 250)."""
from steward.core.alerts import alert_extra_vars
from steward.models.alerts import AlertOperator, AlertRule
def test_alert_extra_vars_describes_the_firing():
rule = AlertRule(
name="UPS low battery", source_module="snmp", resource_name="ups",
metric_name="battery_status", operator=AlertOperator.lt, threshold=3.0,
created_by="u",
)
ev = alert_extra_vars(rule, 2.0)
assert "steward_alert_rule=UPS low battery" in ev
assert "steward_alert_source=snmp" in ev
assert "steward_alert_resource=ups" in ev
assert "steward_alert_metric=battery_status" in ev
assert "steward_alert_value=2.0" in ev
assert "steward_alert_threshold=3.0" in ev
assert "steward_alert_operator=<" in ev
def test_alert_extra_vars_are_key_value_strings():
rule = AlertRule(
name="r", source_module="host_agent", resource_name="srv1",
metric_name="cpu_pct", operator=AlertOperator.gt, threshold=90.0,
created_by="u",
)
ev = alert_extra_vars(rule, 95.5)
assert all("=" in item for item in ev)
assert all(item.startswith("steward_alert_") for item in ev)