from typing import Optional

import pytest

from scripts.audit.anomalies import (
    PropertyAudit,
    _effective_lodged_divergence,
    _implausible_lodged_score,
)


def _make_audit(
    *,
    lodged_sap: Optional[float],
    effective_sap: Optional[float],
    rebaseline_reason: str = "both",
) -> PropertyAudit:
    return PropertyAudit(
        property_id=1,
        uprn=None,
        portfolio_id=796,
        scenario_id=None,
        scenario_goal_band=None,
        lodged_sap=lodged_sap,
        lodged_band=None,
        effective_sap=effective_sap,
        effective_band=None,
        rebaseline_reason=rebaseline_reason,
        post_sap=None,
        post_band=None,
        cost_of_works=None,
        energy_bill_savings=None,
        energy_consumption_savings=None,
        solar_sap_points=None,
        solar_bill_savings=None,
        n_measures=0,
    )


class TestEffectiveLodgedDivergence:
    def test_silent_when_lodged_sap_below_floor(self) -> None:
        # Arrange — Class C exemplar: lodged=1, effective=76 (gap +75, but lodged is implausible upstream data)
        audit = _make_audit(lodged_sap=1.0, effective_sap=76.0)

        # Act
        result = _effective_lodged_divergence(audit)

        # Assert
        assert result is None

    def test_fires_when_lodged_sap_at_or_above_floor_with_large_gap(self) -> None:
        # Arrange — lodged=20 (band G, plausible), effective=65 (band D), gap +45
        audit = _make_audit(lodged_sap=20.0, effective_sap=65.0)

        # Act
        result = _effective_lodged_divergence(audit)

        # Assert
        assert result is not None
        assert "65" in result
        assert "20" in result


class TestImplausibleLodgedScore:
    def test_fires_when_lodged_sap_below_floor_and_gap_large(self) -> None:
        # Arrange — Class C exemplar: lodged=1, effective=76 (gap +75)
        audit = _make_audit(lodged_sap=1.0, effective_sap=76.0)

        # Act
        result = _implausible_lodged_score(audit)

        # Assert
        assert result is not None
        assert "1" in result

    def test_silent_when_gap_below_threshold(self) -> None:
        # Arrange — lodged=9, effective=5 (gap -4, below ±15): both low, not a divergence anomaly
        audit = _make_audit(lodged_sap=9.0, effective_sap=5.0)

        # Act
        result = _implausible_lodged_score(audit)

        # Assert
        assert result is None

    def test_silent_when_lodged_sap_at_or_above_floor(self) -> None:
        # Arrange — lodged=13 is on the floor boundary; effective=70 gives large gap
        audit = _make_audit(lodged_sap=13.0, effective_sap=70.0)

        # Act
        result = _implausible_lodged_score(audit)

        # Assert
        assert result is None