from __future__ import annotations

from contextlib import asynccontextmanager
from types import SimpleNamespace

import pytest

from src.models.incident import Severity
from src.services import incident_memory as incident_memory_module
from src.services.incident_memory import (
    IncidentDbAdapter,
    _derive_incident_alert_metadata,
)


def test_signal_worker_incident_metadata_uses_signal_alert_name() -> None:
    incident = SimpleNamespace(
        incident_id="INC-TEST",
        severity=Severity.P2,
        signals=[
            SimpleNamespace(
                alert_name="HostErrorLogFlood",
                severity=Severity.P2,
                source="journal",
                labels={"error_count": "29"},
                annotations={"summary": "29 ERROR log entries in last 5min"},
                fingerprint="b000a87cfe4bc658",
            )
        ],
    )

    metadata = _derive_incident_alert_metadata(incident)

    assert metadata["alertname"] == "HostErrorLogFlood"
    assert metadata["alert_category"] == "host_resource"
    assert metadata["notification_type"] == "TYPE-3"
    assert metadata["description"] == "29 ERROR log entries in last 5min"
    assert metadata["actor"] == "journal"


@pytest.mark.asyncio
async def test_incident_db_adapter_uses_explicit_project_context(monkeypatch) -> None:
    project_ids: list[str | None] = []

    class FakeResult:
        def scalar_one_or_none(self):
            return None

    class FakeDb:
        async def execute(self, _statement):
            return FakeResult()

    @asynccontextmanager
    async def fake_get_db_context(project_id=None):
        project_ids.append(project_id)
        yield FakeDb()

    monkeypatch.setattr(
        incident_memory_module,
        "get_db_context",
        fake_get_db_context,
    )

    result = await IncidentDbAdapter(project_id="awoooi").load("INC-NOT-FOUND")

    assert result is None
    assert project_ids == ["awoooi"]