"""PRD §5.4 FRFM042 — payment failure flow (category + audit + auto-queue)."""
from __future__ import annotations

from datetime import timedelta
from decimal import Decimal
from unittest.mock import patch

import pytest
from django.utils import timezone

from apps.core.audit.models import AuditLog
from apps.core.audit.tasks import persist_audit_log
from apps.rims.finance.models import (
    Budget,
    BudgetLine,
    DisbursementExecution,
    DisbursementRequest,
    PaymentRetry,
)
from apps.rims.finance.services import mark_execution_failed
from apps.rims.grants.models import Application, Award, GrantCall


@pytest.fixture(autouse=True)
def _sync_audit():
    with patch("apps.core.audit.tasks.write_audit_log.delay", side_effect=lambda **kw: persist_audit_log(**kw)):
        with patch(
            "apps.core.audit.mixins.transaction.on_commit",
            side_effect=lambda fn: fn(),
        ):
            yield


def _execution(applicant_user, institution, finance_user) -> DisbursementExecution:
    call = GrantCall.objects.create(
        title=f"6E-fail call {timezone.now().timestamp()}",
        slug=f"6e-fail-{int(timezone.now().timestamp() * 1000) % 10_000_000}",
        call_type="grant",
        opens_at=timezone.now() - timedelta(days=200),
        closes_at=timezone.now() + timedelta(days=10),
        status=GrantCall.Status.PUBLISHED,
    )
    app = Application.objects.create(call=call, applicant=applicant_user, institution=institution)
    award = Award.objects.create(
        application=app,
        amount=Decimal("5000"),
        currency="USD",
        awarded_at=timezone.now(),
        project_end_date=timezone.now().date() + timedelta(days=365),
    )
    budget = Budget.objects.create(award=award, name="Failure budget", currency="USD")
    BudgetLine.objects.create(budget=budget, label="L", amount=Decimal("5000"))
    request = DisbursementRequest.objects.create(
        budget=budget,
        amount=Decimal("500"),
        requested_by=finance_user,
        status=DisbursementRequest.Status.APPROVED,
    )
    return DisbursementExecution.objects.create(
        request=request,
        status=DisbursementExecution.Status.SCHEDULED,
    )


@pytest.mark.django_db
def test_mark_failed_sets_category_emits_audit_and_queues_retry(
    applicant_user, institution, finance_user
):
    execution = _execution(applicant_user, institution, finance_user)
    retry = mark_execution_failed(
        execution,
        actor=finance_user,
        failure_category=DisbursementExecution.FailureCategory.NSF,
        failure_reason="Insufficient funds at source bank",
    )
    execution.refresh_from_db()
    assert execution.status == DisbursementExecution.Status.FAILED
    assert execution.failure_category == "nsf"
    assert execution.failure_reason.startswith("Insufficient funds")
    assert retry.status == PaymentRetry.Status.QUEUED
    assert AuditLog.objects.filter(action="PAYMENT_EXECUTION_FAILED").exists()
    assert AuditLog.objects.filter(action="PAYMENT_RETRY_QUEUED").exists()


@pytest.mark.django_db
def test_mark_failed_requires_category(applicant_user, institution, finance_user):
    execution = _execution(applicant_user, institution, finance_user)
    with pytest.raises(Exception):
        mark_execution_failed(
            execution, actor=finance_user, failure_category="", failure_reason="x"
        )