AlexeerCT
06b3b20fd8
- Introduced a new fixture, `test_tenant`, to allocate ephemeral tenant and company IDs for tests, reducing conflicts with real data. - Updated various test functions to use the new `test_tenant` fixture, ensuring consistent tenant ID usage across tests. - Enhanced the `ensure_tenant_company` function to accept dynamic tenant and company IDs, improving flexibility in test scenarios. - Adjusted database interaction in tests to utilize the ephemeral IDs, streamlining the setup process and enhancing isolation.
134 lines
6.1 KiB
Python
134 lines
6.1 KiB
Python
from decimal import Decimal
|
|
|
|
import pytest
|
|
|
|
from api.v1.modules.a24.balance_movements.models import BalanceMovement, MovementType
|
|
from api.v1.modules.a76.invoices.exports.process import main_process as export_main
|
|
from api.v1.modules.a76.invoices.imports.process import main_process as import_main
|
|
from core.exceptions import ValidationException
|
|
from tests.fixtures.builders import (
|
|
EphemeralOrgIds,
|
|
create_business_catalogs,
|
|
create_export_invoice_with_line,
|
|
create_import_invoice_with_line,
|
|
ensure_reference_data,
|
|
)
|
|
|
|
|
|
def _patch_import_pipeline(monkeypatch, import_line):
|
|
monkeypatch.setattr(import_main, "pre_validators", lambda *args, **kwargs: [import_line])
|
|
monkeypatch.setattr(import_main, "review_classes", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_exchange_rate", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_weights_kgs", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_weights_lbs", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "_validate_sisimp_limits", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "_validate_lines", lambda *args, **kwargs: ([], {}))
|
|
|
|
|
|
def _patch_import_pipeline_without_prevalidators(monkeypatch):
|
|
monkeypatch.setattr(import_main, "review_classes", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_exchange_rate", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_weights_kgs", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "review_weights_lbs", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "_validate_sisimp_limits", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(import_main, "_validate_lines", lambda *args, **kwargs: ([], {}))
|
|
|
|
|
|
def _patch_export_pipeline(monkeypatch, export_line):
|
|
monkeypatch.setattr(export_main, "pre_validators", lambda *args, **kwargs: [export_line])
|
|
monkeypatch.setattr(export_main, "review_class", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(export_main, "review_exchange_rate", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(export_main, "assign_values", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(export_main, "review_qty_vs_weight", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(export_main, "review_unit_cost", lambda *args, **kwargs: None)
|
|
monkeypatch.setattr(export_main, "review_qty_series", lambda *args, **kwargs: None)
|
|
|
|
|
|
def _prepare_inventory(client, db_session, monkeypatch, test_tenant: EphemeralOrgIds):
|
|
tid, cid = test_tenant.tenant_id, test_tenant.company_id
|
|
ensure_reference_data(db_session)
|
|
catalogs = create_business_catalogs(db_session, tenant_id=tid, company_id=cid)
|
|
import_invoice, import_line = create_import_invoice_with_line(
|
|
db_session, tid, cid, catalogs, invoice_type="TEM", invoice_number="IMP-EXP-01", qty=Decimal("10")
|
|
)
|
|
_patch_import_pipeline(monkeypatch, import_line)
|
|
resp = client.post(f"/api/v1/a76/invoices/{import_invoice.id}/process?company_id={cid}")
|
|
assert resp.status_code == 200
|
|
return catalogs, import_invoice, import_line
|
|
|
|
|
|
def test_process_export_endpoint_consumes_existing_balances(client, db_session, monkeypatch, test_tenant):
|
|
catalogs, import_invoice, import_line = _prepare_inventory(
|
|
client, db_session, monkeypatch, test_tenant
|
|
)
|
|
tid, cid = test_tenant.tenant_id, test_tenant.company_id
|
|
export_invoice, export_line = create_export_invoice_with_line(
|
|
db_session, tid, cid, catalogs, import_invoice, import_line, qty=Decimal("3")
|
|
)
|
|
_patch_export_pipeline(monkeypatch, export_line)
|
|
|
|
resp = client.post(f"/api/v1/a76/invoices/{export_invoice.id}/process?company_id={cid}")
|
|
assert resp.status_code == 200
|
|
|
|
consumptions = (
|
|
db_session.query(BalanceMovement)
|
|
.filter(
|
|
BalanceMovement.source_invoice_id == export_invoice.id,
|
|
BalanceMovement.movement_type == MovementType.CONSUMPTION,
|
|
)
|
|
.all()
|
|
)
|
|
assert consumptions
|
|
|
|
|
|
def test_process_export_prevents_negative_balance(client, db_session, monkeypatch, test_tenant):
|
|
catalogs, import_invoice, import_line = _prepare_inventory(
|
|
client, db_session, monkeypatch, test_tenant
|
|
)
|
|
tid, cid = test_tenant.tenant_id, test_tenant.company_id
|
|
export_invoice, export_line = create_export_invoice_with_line(
|
|
db_session, tid, cid, catalogs, import_invoice, import_line, qty=Decimal("99")
|
|
)
|
|
_patch_export_pipeline(monkeypatch, export_line)
|
|
|
|
with pytest.raises(ValidationException):
|
|
client.post(f"/api/v1/a76/invoices/{export_invoice.id}/process?company_id={cid}")
|
|
|
|
# Debe no crear consumos al existir insuficiencia de saldo.
|
|
consumptions = (
|
|
db_session.query(BalanceMovement)
|
|
.filter(
|
|
BalanceMovement.source_invoice_id == export_invoice.id,
|
|
BalanceMovement.movement_type == MovementType.CONSUMPTION,
|
|
)
|
|
.all()
|
|
)
|
|
assert consumptions == []
|
|
|
|
|
|
def test_process_endpoint_prevents_double_processing_import(client, db_session, monkeypatch, test_tenant):
|
|
tid, cid = test_tenant.tenant_id, test_tenant.company_id
|
|
ensure_reference_data(db_session)
|
|
catalogs = create_business_catalogs(db_session, tenant_id=tid, company_id=cid)
|
|
invoice, line = create_import_invoice_with_line(
|
|
db_session, tid, cid, catalogs, invoice_type="TEM", invoice_number="IMP-DOUBLE-01", qty=Decimal("5")
|
|
)
|
|
_patch_import_pipeline_without_prevalidators(monkeypatch)
|
|
|
|
first = client.post(f"/api/v1/a76/invoices/{invoice.id}/process?company_id={cid}")
|
|
assert first.status_code == 200
|
|
with pytest.raises(ValidationException):
|
|
client.post(f"/api/v1/a76/invoices/{invoice.id}/process?company_id={cid}")
|
|
|
|
# No debe duplicar entradas para la misma factura/línea procesada.
|
|
entries = (
|
|
db_session.query(BalanceMovement)
|
|
.filter(
|
|
BalanceMovement.import_invoice_id == invoice.id,
|
|
BalanceMovement.import_item_line_id == line.id,
|
|
BalanceMovement.movement_type == MovementType.ENTRY,
|
|
)
|
|
.all()
|
|
)
|
|
assert len(entries) == 1
|