Add 'fiscal/' from commit '102ba84500dbd9655a0471288c3e278652593ca9'

git-subtree-dir: fiscal
git-subtree-mainline: d95165fc8f
git-subtree-split: 102ba84500

fiscal/central/tests/test_reference_endpoints.py

@@ -0,0 +1,53 @@
+from datetime import datetime, timezone
+
+from fastapi.testclient import TestClient
+from sqlalchemy import create_engine
+from sqlalchemy.orm import Session
+from sqlalchemy.pool import StaticPool
+
+from central.app.datasets.service import DatasetService
+from central.app.main import create_app
+from central.app.models import AnpProductCode, Base, CfopEntry, Dataset, FcpRate, NcmUtribEntry
+
+
+def _client_with_reference_data():
+    engine = create_engine("sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool)
+    Base.metadata.create_all(engine)
+    with Session(engine) as session:
+        dataset = Dataset(kind="CFOP", provider="p", sha256="a" * 64, version_label="v", act="a", source_url="u", status="ACTIVE", record_count=1)
+        session.add(dataset); session.flush()
+        session.add(CfopEntry(dataset_id=dataset.id, code="1101", title="Compra", description="desc"))
+
+        d2 = Dataset(kind="NCM_UTRIB", provider="p", sha256="b" * 64, version_label="v", act="a", source_url="u", status="ACTIVE", record_count=1)
+        session.add(d2); session.flush()
+        session.add(NcmUtribEntry(dataset_id=d2.id, ncm="01012100", utrib_abbreviation="UN", utrib_description="Unidade"))
+
+        d3 = Dataset(kind="ANP", provider="p", sha256="c" * 64, version_label="v", act="a", source_url="u", status="ACTIVE", record_count=1)
+        session.add(d3); session.flush()
+        session.add(AnpProductCode(dataset_id=d3.id, cprod_anp="110203073"))
+
+        d4 = Dataset(kind="FCP", provider="p", sha256="d" * 64, version_label="v", act="a", source_url="u", status="ACTIVE", record_count=1)
+        session.add(d4); session.flush()
+        session.add(FcpRate(dataset_id=d4.id, uf_code="29", uf="BA", uf_name="BAHIA", aliq_1="Fixo:2.00", observacao="obs"))
+        session.commit()
+    app = create_app(session_factory=lambda: Session(engine), bootstrap_token="test-token")
+    return TestClient(app)
+
+
+def test_reference_endpoints_found_and_not_found():
+    client = _client_with_reference_data()
+    headers = {"Authorization": "Bearer test-token"}
+
+    assert client.get("/v1/reference/cfop/1101", headers=headers).json()["found"] is True
+    assert client.get("/v1/reference/cfop/9999", headers=headers).json()["found"] is False
+
+    assert client.get("/v1/reference/ncm_utrib/01012100", headers=headers).json()["found"] is True
+    assert client.get("/v1/reference/ncm_utrib/99999999", headers=headers).json()["found"] is False
+
+    assert client.get("/v1/reference/anp/110203073", headers=headers).json()["found"] is True
+    assert client.get("/v1/reference/anp/000000000", headers=headers).json()["found"] is False
+
+    fcp = client.get("/v1/reference/fcp/BA", headers=headers).json()
+    assert fcp["found"] is True
+    assert fcp["aliq_1"] == "Fixo:2.00"
+    assert client.get("/v1/reference/fcp/ZZ", headers=headers).json()["found"] is False