chore(home): mirror from kalinamall (9883e6a) with papatramp URLs

backend/tests/test_telegram_notify.py

@@ -0,0 +1,76 @@
+"""Tests for SAC Telegram notification helpers."""
+
+from datetime import datetime, timezone
+from unittest.mock import patch
+
+from app.models import Event, Problem
+from app.services import telegram_notify
+from app.services.notification_settings import TelegramConfig
+
+
+def _telegram_cfg(*, min_severity: str) -> TelegramConfig:
+    return TelegramConfig(
+        enabled=True,
+        bot_token="test-token",
+        chat_id="123",
+        min_severity=min_severity,
+        source="env",
+    )
+
+
+def test_notify_event_respects_min_severity():
+    sent: list[str] = []
+
+    def fake_send(message: str, **kwargs) -> None:
+        sent.append(message)
+
+    event = Event(
+        event_id="00000000-0000-4000-8000-000000000001",
+        host_id=1,
+        occurred_at=datetime(2026, 5, 28, 12, 0, tzinfo=timezone.utc),
+        category="auth",
+        type="rdp.login.success",
+        severity="info",
+        title="ok",
+        summary="info event",
+        payload={},
+    )
+
+    cfg = _telegram_cfg(min_severity="warning")
+    with patch.object(telegram_notify, "get_effective_telegram_config", return_value=cfg):
+        with patch.object(telegram_notify, "send_telegram_text", side_effect=fake_send):
+            telegram_notify.notify_event(event)
+    assert sent == []
+
+    event.severity = "warning"
+    with patch.object(telegram_notify, "get_effective_telegram_config", return_value=cfg):
+        with patch.object(telegram_notify, "send_telegram_text", side_effect=fake_send):
+            telegram_notify.notify_event(event)
+    assert len(sent) == 1
+
+
+def test_notify_problem_respects_min_severity():
+    sent: list[str] = []
+
+    def fake_send(message: str, **kwargs) -> None:
+        sent.append(message)
+
+    problem = Problem(
+        title="brute",
+        summary="burst",
+        severity="warning",
+        status="open",
+        fingerprint="fp1",
+    )
+
+    cfg = _telegram_cfg(min_severity="high")
+    with patch.object(telegram_notify, "get_effective_telegram_config", return_value=cfg):
+        with patch.object(telegram_notify, "send_telegram_text", side_effect=fake_send):
+            telegram_notify.notify_problem(problem)
+    assert sent == []
+
+    problem.severity = "high"
+    with patch.object(telegram_notify, "get_effective_telegram_config", return_value=cfg):
+        with patch.object(telegram_notify, "send_telegram_text", side_effect=fake_send):
+            telegram_notify.notify_problem(problem)
+    assert len(sent) == 1