"""Временный режим «период ОСВ из версии консолидации» (arm-core#103): чистые функции `domain.version_period`."""

import datetime
import uuid

import pytest

from src.domain.version_period import takes_version_period, with_version_period
from src.models.consolidation import Consolidation, ConsolidationVersion
from src.models.enums import ConsolidationStatus, DocumentStatus, DocumentType, PeriodQuarter
from src.models.files import Document
from src.services.consolidation_documents import fits

CONSOLIDATION = uuid.uuid4()
OTHER = uuid.uuid4()
MEMBER = uuid.uuid4()


@pytest.mark.parametrize(
    ("enabled", "document_type", "file_consolidation_id", "fits_by_header", "expected"),
    [
        (True, DocumentType.OSV, CONSOLIDATION, False, True),  # ОСВ «не того» периода загружена в эту консолидацию
        (True, DocumentType.OSV, CONSOLIDATION, True, False),  # период шапки и так подходит — склейка кварталов цела
        (False, DocumentType.OSV, CONSOLIDATION, False, False),  # режим выключен — период из шапки
        (True, DocumentType.OSV, OTHER, False, False),  # загружена в другую консолидацию
        (True, DocumentType.OSV, None, False, False),  # загружена с рабочего стола, без консолидации
        (True, DocumentType.BALANCE, CONSOLIDATION, False, False),  # ББ/ОФР — только период из шапки
        (True, DocumentType.INCOME, CONSOLIDATION, False, False),
    ],
)
def test_takes_version_period(
    enabled: bool,
    document_type: DocumentType,
    file_consolidation_id: uuid.UUID | None,
    fits_by_header: bool,
    expected: bool,
) -> None:
    assert (
        takes_version_period(
            enabled=enabled,
            document_type=document_type,
            file_consolidation_id=file_consolidation_id,
            consolidation_id=CONSOLIDATION,
            fits_by_header=fits_by_header,
        )
        is expected
    )


def _document(*, year: int | None = 2025, quarter: PeriodQuarter | None = PeriodQuarter.Q1) -> Document:
    return Document(
        id=uuid.uuid4(),
        legal_entity_id=MEMBER,
        document_type=DocumentType.OSV,
        osv_account_code="60",
        period_year=year,
        period_quarter=quarter,
        is_cumulative=False,
        status=DocumentStatus.CONFIRMED,
        confirmed_at=datetime.datetime(2026, 10, 12, tzinfo=datetime.UTC),
    )


def test_with_version_period_takes_year_and_quarter_of_version_as_cumulative() -> None:
    document = _document()

    view = with_version_period(document, year=2026, quarter=PeriodQuarter.Q3)

    assert (view.period_year, view.period_quarter, view.is_cumulative) == (2026, PeriodQuarter.Q3, True)
    assert (view.id, view.legal_entity_id, view.document_type, view.osv_account_code) == (
        document.id,
        MEMBER,
        DocumentType.OSV,
        "60",
    )
    assert (view.status, view.confirmed_at) == (document.status, document.confirmed_at)
    # первичка не мутируется: документ сохраняет период из шапки
    assert (document.period_year, document.period_quarter, document.is_cumulative) == (2025, PeriodQuarter.Q1, False)


@pytest.mark.parametrize("quarter", [None, PeriodQuarter.Q4])
def test_version_period_fits_version_whatever_header_period(quarter: PeriodQuarter | None) -> None:
    """Документ без квартала или за другой год/квартал в представлении версии подходит ей (если ЮЛ — член)."""
    consolidation = Consolidation(year=2026, quarter=PeriodQuarter.Q2)
    version = ConsolidationVersion(status=ConsolidationStatus.INITIATED)
    view = with_version_period(_document(year=2024, quarter=quarter), year=2026, quarter=PeriodQuarter.Q2)

    assert fits(view, version, consolidation, {MEMBER}) is True
    assert fits(view, version, consolidation, {uuid.uuid4()}) is False