97 lines
2.5 KiB
Python
97 lines
2.5 KiB
Python
|
"""Tests for worker notification helpers."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from uuid import UUID, uuid4
|
||
|
|
|
||
|
|
import asyncpg
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from app.repositories.incident import IncidentRepository
|
||
|
|
from worker.tasks.notifications import NotificationDispatch, prepare_notification_dispatches
|
||
|
|
|
||
|
|
|
||
|
|
pytestmark = pytest.mark.asyncio
|
||
|
|
|
||
|
|
|
||
|
|
async def _seed_incident(conn: asyncpg.Connection) -> tuple[UUID, UUID, UUID]:
|
||
|
|
org_id = uuid4()
|
||
|
|
service_id = uuid4()
|
||
|
|
incident_id = uuid4()
|
||
|
|
|
||
|
|
await conn.execute(
|
||
|
|
"INSERT INTO orgs (id, name, slug) VALUES ($1, $2, $3)",
|
||
|
|
org_id,
|
||
|
|
"Notif Org",
|
||
|
|
"notif-org",
|
||
|
|
)
|
||
|
|
await conn.execute(
|
||
|
|
"INSERT INTO services (id, org_id, name, slug) VALUES ($1, $2, $3, $4)",
|
||
|
|
service_id,
|
||
|
|
org_id,
|
||
|
|
"API",
|
||
|
|
"api",
|
||
|
|
)
|
||
|
|
|
||
|
|
repo = IncidentRepository(conn)
|
||
|
|
await repo.create(
|
||
|
|
incident_id=incident_id,
|
||
|
|
org_id=org_id,
|
||
|
|
service_id=service_id,
|
||
|
|
title="Outage",
|
||
|
|
description="",
|
||
|
|
severity="high",
|
||
|
|
)
|
||
|
|
|
||
|
|
return org_id, service_id, incident_id
|
||
|
|
|
||
|
|
|
||
|
|
async def test_prepare_notification_dispatches_creates_attempts(db_conn: asyncpg.Connection) -> None:
|
||
|
|
org_id, _service_id, incident_id = await _seed_incident(db_conn)
|
||
|
|
|
||
|
|
target_id = uuid4()
|
||
|
|
await db_conn.execute(
|
||
|
|
"""
|
||
|
|
INSERT INTO notification_targets (id, org_id, name, target_type, enabled)
|
||
|
|
VALUES ($1, $2, $3, $4, $5)
|
||
|
|
""",
|
||
|
|
target_id,
|
||
|
|
org_id,
|
||
|
|
"Primary Webhook",
|
||
|
|
"webhook",
|
||
|
|
True,
|
||
|
|
)
|
||
|
|
|
||
|
|
dispatches = await prepare_notification_dispatches(db_conn, incident_id=incident_id, org_id=org_id)
|
||
|
|
|
||
|
|
assert len(dispatches) == 1
|
||
|
|
dispatch = dispatches[0]
|
||
|
|
assert isinstance(dispatch, NotificationDispatch)
|
||
|
|
assert dispatch.target["name"] == "Primary Webhook"
|
||
|
|
|
||
|
|
attempt = await db_conn.fetchrow(
|
||
|
|
"SELECT status FROM notification_attempts WHERE id = $1",
|
||
|
|
dispatch.attempt_id,
|
||
|
|
)
|
||
|
|
assert attempt is not None and attempt["status"] == "pending"
|
||
|
|
|
||
|
|
|
||
|
|
async def test_prepare_notification_dispatches_skips_disabled_targets(db_conn: asyncpg.Connection) -> None:
|
||
|
|
org_id, _service_id, incident_id = await _seed_incident(db_conn)
|
||
|
|
|
||
|
|
await db_conn.execute(
|
||
|
|
"""
|
||
|
|
INSERT INTO notification_targets (id, org_id, name, target_type, enabled)
|
||
|
|
VALUES ($1, $2, $3, $4, $5)
|
||
|
|
""",
|
||
|
|
uuid4(),
|
||
|
|
org_id,
|
||
|
|
"Disabled",
|
||
|
|
"email",
|
||
|
|
False,
|
||
|
|
)
|
||
|
|
|
||
|
|
dispatches = await prepare_notification_dispatches(db_conn, incident_id=incident_id, org_id=org_id)
|
||
|
|
|
||
|
|
assert dispatches == []
|