CRM_AGENTES_CARGA/backend/tests/test_expedientes.py

"""Pruebas de la entidad expediente y de su enganche con la solicitud de servicio."""

import pytest
from sqlalchemy.exc import IntegrityError

from api.v1.modules.crm.expedientes import service as expedientes_service
from api.v1.modules.crm.expedientes.models import Expediente
from api.v1.modules.crm.service_requests import service as sr_service
from api.v1.modules.crm.service_requests.dto import ServiceRequestCreate
from tests.conftest import COMPANY_ID, TENANT_ID

OTRO_TENANT = 99
OTRA_COMPANY = 77


def _crear_solicitud(db, **kwargs) -> object:
    payload = ServiceRequestCreate(operation_type=kwargs.pop("operation_type", "importacion"), **kwargs)
    return sr_service.create_service_request(db, payload, TENANT_ID, COMPANY_ID, "user-1")


def test_una_solicitud_nueva_nace_con_su_expediente_y_folio(db):
    solicitud = _crear_solicitud(db)

    expediente = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)
    assert expediente is not None
    assert expediente.folio.startswith("EXP")
    assert expediente.sequence == 1
    assert expediente.status == "abierto"
    assert expediente.efc_link_state == "PENDING"


def test_el_expediente_nace_con_su_storage_token_ya_asignado(db):
    """El token es la carpeta de MinIO en EFC y es INMUTABLE: se fija al nacer, no al enlazar.

    Si se asignara al momento de crear el provisional en EFC, un documento subido antes de que EFC
    conteste no sabría bajo qué prefijo va.
    """
    solicitud = _crear_solicitud(db)
    expediente = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)
    assert expediente.efc_storage_token == f"CRM-{COMPANY_ID}-{expediente.folio}"


def test_ensure_es_idempotente(db):
    solicitud = _crear_solicitud(db)
    primero = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    segundo = expedientes_service.ensure_expediente(db, solicitud.id, TENANT_ID, COMPANY_ID, "user-1")
    tercero = expedientes_service.ensure_expediente(db, solicitud.id, TENANT_ID, COMPANY_ID, "user-1")

    assert primero.id == segundo.id == tercero.id
    assert primero.folio == segundo.folio == tercero.folio
    assert len(expedientes_service.list_expedientes(db, TENANT_ID, COMPANY_ID)) == 1


def test_dos_solicitudes_reciben_folios_consecutivos(db):
    a = _crear_solicitud(db)
    b = _crear_solicitud(db)
    exp_a = expedientes_service.find_by_service_request(db, a.id, TENANT_ID, COMPANY_ID)
    exp_b = expedientes_service.find_by_service_request(db, b.id, TENANT_ID, COMPANY_ID)
    assert exp_b.sequence == exp_a.sequence + 1
    assert exp_a.folio != exp_b.folio


def test_dos_expedientes_con_el_mismo_folio_en_el_mismo_tenant_y_company_revientan(db):
    """``uq_crm_expedientes_folio`` es la red de seguridad si el contador se corrompe.

    Un folio duplicado tiene que fallar ruidosamente en vez de mezclar dos hilos documentales.
    """
    solicitud = _crear_solicitud(db)
    original = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    duplicado = Expediente(
        folio=original.folio,
        period_year=original.period_year,
        period_month=original.period_month,
        sequence=original.sequence + 1,  # distinta secuencia: el que debe romper es el folio
        status="abierto",
        efc_link_state="PENDING",
        tenant_id=TENANT_ID,
        company_id=COMPANY_ID,
    )
    db.add(duplicado)
    with pytest.raises(IntegrityError):
        db.commit()
    db.rollback()


def test_dos_expedientes_con_la_misma_secuencia_del_mes_revientan(db):
    """``uq_crm_expedientes_periodo_seq``: el consecutivo es un constraint real, no un parse."""
    solicitud = _crear_solicitud(db)
    original = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    duplicado = Expediente(
        folio=original.folio + "-BIS",  # distinto folio: el que debe romper es (año, mes, seq)
        period_year=original.period_year,
        period_month=original.period_month,
        sequence=original.sequence,
        status="abierto",
        efc_link_state="PENDING",
        tenant_id=TENANT_ID,
        company_id=COMPANY_ID,
    )
    db.add(duplicado)
    with pytest.raises(IntegrityError):
        db.commit()
    db.rollback()


def test_el_mismo_folio_en_otra_company_si_puede_existir(db):
    """El folio es único por ``(tenant, company)``, no globalmente.

    Por eso el ``storage_token`` que viaja a EFC lleva el company_id: allá sí comparten organización.
    """
    solicitud = _crear_solicitud(db)
    original = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    gemelo = Expediente(
        folio=original.folio,
        period_year=original.period_year,
        period_month=original.period_month,
        sequence=original.sequence,
        status="abierto",
        efc_link_state="PENDING",
        tenant_id=TENANT_ID,
        company_id=OTRA_COMPANY,
    )
    db.add(gemelo)
    db.commit()
    assert gemelo.id is not None


def test_un_expediente_no_se_ve_desde_otro_tenant_ni_otra_company(db):
    solicitud = _crear_solicitud(db)
    expediente = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    assert expedientes_service.list_expedientes(db, OTRO_TENANT, COMPANY_ID) == []
    assert expedientes_service.list_expedientes(db, TENANT_ID, OTRA_COMPANY) == []

    from fastapi import HTTPException

    with pytest.raises(HTTPException) as exc:
        expedientes_service.get_expediente(db, expediente.id, OTRO_TENANT, COMPANY_ID)
    assert exc.value.status_code == 404


def test_completar_guarda_la_data_aduanera_y_cambia_el_estado(db):
    from datetime import date

    from api.v1.modules.crm.expedientes.dto import ExpedienteCompleteInput

    solicitud = _crear_solicitud(db)
    expediente = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)

    # Datos dummy: patente 0000, aduana 000, pedimento 0000-0000000, RFC XAXX010101000.
    completado = expedientes_service.complete_expediente(
        db,
        expediente.id,
        ExpedienteCompleteInput(
            patente="0000",
            aduana="000",
            numero_pedimento="0000000",
            anio=2026,
            clave_pedimento="A1",
            regimen="IMD",
            fecha_pago=date(2026, 8, 1),
            rfc_importador="XAXX010101000",
        ),
        TENANT_ID,
        COMPANY_ID,
        "user-1",
    )

    assert completado.status == "completado"
    assert completado.patente == "0000"
    assert completado.aduana == "000"
    assert completado.rfc_importador == "XAXX010101000"
    # El folio y el token NO cambian al completar: es lo que permite que ningún archivo se mueva.
    assert completado.folio == expediente.folio
    assert completado.efc_storage_token == expediente.efc_storage_token


def test_completar_dos_veces_da_409(db):
    from api.v1.modules.crm.expedientes.dto import ExpedienteCompleteInput
    from fastapi import HTTPException

    solicitud = _crear_solicitud(db)
    expediente = expedientes_service.find_by_service_request(db, solicitud.id, TENANT_ID, COMPANY_ID)
    payload = ExpedienteCompleteInput(patente="0000", aduana="000", numero_pedimento="0000000", anio=2026)

    expedientes_service.complete_expediente(db, expediente.id, payload, TENANT_ID, COMPANY_ID)
    with pytest.raises(HTTPException) as exc:
        expedientes_service.complete_expediente(db, expediente.id, payload, TENANT_ID, COMPANY_ID)
    assert exc.value.status_code == 409