Johan LEROY
e50921c907
Consultation des alertes de consommation, filtrable par site_id et severity a l'identique du contrat GET /alerts de l'API Mock. Reprend le gabarit endpoints -> services -> repositories -> models pose par sites, sur la table alert deja creee par la revision Alembic e6d2026091501. Generalise aussi le garde-fou OpenAPI du 403 (ROUTES_A_ROLE) au-dela du seul tag users, pour que l'ajout d'alerts a la liste des routes protegees par role soit reellement verifie. Closes #59
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from datetime import UTC, datetime
|
|
|
|
from app.models.energy import Alert
|
|
from app.services.alert import AlertService
|
|
|
|
|
|
def alert(
|
|
alert_id: int = 1,
|
|
site_id: str = "site-1",
|
|
severity: str = "high",
|
|
) -> Alert:
|
|
return Alert(
|
|
alert_id=alert_id,
|
|
source_alert_id=f"ALR-{alert_id}",
|
|
site_id=site_id,
|
|
source="enervision",
|
|
timestamp=datetime(2026, 9, 16, tzinfo=UTC),
|
|
type="threshold",
|
|
severity=severity,
|
|
message="Dépassement du seuil configuré",
|
|
value=812.5,
|
|
threshold=720.0,
|
|
metric="consumption_kw",
|
|
prediction_id=None,
|
|
raw_data={},
|
|
)
|
|
|
|
|
|
class FakeRepository:
|
|
def __init__(self, alerts: list[Alert]) -> None:
|
|
self._alerts = alerts
|
|
self.appels: list[tuple[str | None, str | None]] = []
|
|
|
|
async def list_all(
|
|
self, *, site_id: str | None = None, severity: str | None = None
|
|
) -> list[Alert]:
|
|
self.appels.append((site_id, severity))
|
|
return self._alerts
|
|
|
|
|
|
async def test_list_all_returns_the_repository_alerts() -> None:
|
|
service = AlertService(alerts=FakeRepository([alert(1), alert(2)]))
|
|
|
|
alertes = await service.list_all()
|
|
|
|
assert [a.alert_id for a in alertes] == [1, 2]
|
|
|
|
|
|
async def test_list_all_relays_the_filters_to_the_repository() -> None:
|
|
depot = FakeRepository([])
|
|
service = AlertService(alerts=depot)
|
|
|
|
await service.list_all(site_id="site-1", severity="critical")
|
|
|
|
assert depot.appels == [("site-1", "critical")]
|