from datetime import timedelta
from decimal import Decimal

import pytest
from django.urls import reverse
from django.utils import timezone

from apps.rims.finance.models import Budget, BudgetLine, DisbursementRequest
from apps.rims.grants.models import Application, Award, GrantCall
from apps.rims.grants.services import award_application, shortlist_application, submit_application


@pytest.mark.django_db
def test_award_closeout_blocked_by_pending_finance_workflow(
    client, grants_manager_user, applicant_user, institution
):
    call = GrantCall.objects.create(
        title="Closeout finance gate",
        slug=f"closeout-finance-{int(timezone.now().timestamp())}",
        opens_at=timezone.now() - timedelta(days=1),
        closes_at=timezone.now() + timedelta(days=15),
        status=GrantCall.Status.PUBLISHED,
    )
    app = Application.objects.create(call=call, applicant=applicant_user, institution=institution)
    submit_application(app)
    app = Application.objects.get(pk=app.pk)
    shortlist_application(app)
    award_application(app, Decimal("1500"), timezone.now().date() + timedelta(days=180), narrative="")
    award = Application.objects.get(pk=app.pk).award
    budget = Budget.objects.create(award=award, name="Closeout budget")
    BudgetLine.objects.create(budget=budget, label="Line", amount=Decimal("1200"))
    DisbursementRequest.objects.create(
        budget=budget,
        amount=Decimal("100"),
        status=DisbursementRequest.Status.SUBMITTED,
    )
    client.force_login(grants_manager_user)
    url = reverse("rims_grants:award_closeout", kwargs={"pk": award.pk})
    resp = client.post(
        url,
        {
            "narrative": "Final report narrative",
            "co_done__final_narrative": "on",
            "co_done__financial_reconciliation": "on",
            "co_done__no_outstanding_flags": "on",
        },
        follow=True,
    )
    assert resp.status_code == 200
    award.refresh_from_db()
    assert award.status == Award.Status.ACTIVE