118 lines
6.4 KiB
Python
118 lines
6.4 KiB
Python
import pytest
|
|
|
|
from gti_fiscal_agent.company_context import CompanyContextError, config_from_env, mask_cnpj, preflight_report, resolve
|
|
from gti_fiscal_agent.firebird_reader import parse_company_context
|
|
|
|
CN_A, CN_B, CN_C = "11111111000111", "22222222000122", "33333333000133"
|
|
|
|
|
|
def co(code, cnpj, uf="BA", crt="1", optante="S"):
|
|
return {"code": code, "cnpj": cnpj, "uf": uf, "crt": crt, "optante": optante, "name": "X"}
|
|
|
|
|
|
def ev(companies, emissor=None, docs=None):
|
|
return {"c000004": companies, "emissor": emissor if emissor is not None else [], "nfce_keys": docs or [], "nf_by_filial": {}}
|
|
|
|
|
|
def emis(cnpj=CN_A, uf="BA", crt="1"):
|
|
return {"cnpj": cnpj, "uf": uf, "crt": crt}
|
|
|
|
|
|
DOC_A = [{"cuf": "29", "cnpj": CN_A, "n": 100}]
|
|
|
|
|
|
def test_single_company_all_sources_agree_is_auto_resolved():
|
|
r = resolve(ev([co("000001", CN_A)], [emis()], DOC_A))
|
|
assert r["status"] == "PASS" and r["classification"] == "AUTO_RESOLVED" and r["company_uf"] == "BA" and r["company_crt"] == "1" and r["confidence"] == "HIGH"
|
|
assert "DOCUMENT_KEYS" in r["source_of_truth"]["uf"]
|
|
|
|
|
|
def test_multiple_rows_resolved_only_through_emitter_identity_never_first_row():
|
|
rows = [co("000001", CN_B, "SP", None, "N"), co("000005", CN_A), co("000007", CN_C, "BA", None, "N")]
|
|
r = resolve(ev(rows, [emis(CN_A)], DOC_A))
|
|
assert r["status"] == "PASS" and r["company_code"] == "000005" and r["source_of_truth"]["company"] == "EMISSOR_CNPJ_MATCH"
|
|
|
|
|
|
def test_multiple_rows_without_disambiguation_requires_config():
|
|
rows = [co("000001", CN_A), co("000002", CN_B)]
|
|
r = resolve(ev(rows, []))
|
|
assert r["status"] == "FAIL" and r["classification"] == "CONFIG_REQUIRED" and r["company_code"] is None
|
|
assert resolve(ev(rows, [emis(CN_C)]))["classification"] == "CONFIG_REQUIRED" # emitter matches none
|
|
ok = resolve(ev(rows, [], []), {"company_code": "000002"})
|
|
assert ok["status"] == "PASS" and ok["company_code"] == "000002" and ok["classification"] == "CONFIGURED"
|
|
|
|
|
|
def test_configured_company_must_exist_and_agree_with_emitter():
|
|
rows = [co("000001", CN_A), co("000002", CN_B)]
|
|
assert resolve(ev(rows, [emis(CN_A)]), {"company_cnpj": "99999999999999"})["classification"] == "CONFIG_REQUIRED"
|
|
r = resolve(ev(rows, [emis(CN_A)], DOC_A), {"company_code": "000002"})
|
|
assert r["status"] == "FAIL" and r["classification"] == "CONFLICT" # selected company differs from the emitter: unexplained
|
|
|
|
|
|
def test_documents_only_from_another_cnpj_conflict_but_mixed_history_is_a_warning():
|
|
r = resolve(ev([co("000001", CN_A)], [emis()], [{"cuf": "29", "cnpj": CN_B, "n": 10}]))
|
|
assert r["status"] == "FAIL" and r["classification"] == "CONFLICT"
|
|
r = resolve(ev([co("000001", CN_A)], [emis()], [{"cuf": "29", "cnpj": CN_A, "n": 10}, {"cuf": "29", "cnpj": CN_B, "n": 1}]))
|
|
assert r["status"] == "PASS" and any("HISTORICAL_NFCE_KEYS_FROM_OTHER_CNPJ" in w for w in r["warnings"])
|
|
|
|
|
|
def test_uf_conflict_and_unknown():
|
|
assert resolve(ev([co("1", CN_A, "BA")], [emis(uf="SP")]))["classification"] == "CONFLICT"
|
|
assert resolve(ev([co("1", CN_A, "BA")], [emis()], [{"cuf": "35", "cnpj": CN_A, "n": 5}]))["classification"] == "CONFLICT" # key says SP
|
|
assert resolve(ev([co("1", CN_A, "")], [emis(uf="")]))["classification"] == "UNRESOLVED"
|
|
|
|
|
|
def test_crt_conflict_needs_explicit_source_and_records_it():
|
|
base = ev([co("1", CN_A, crt="3")], [emis(crt="1")], DOC_A)
|
|
r = resolve(base)
|
|
assert r["status"] == "FAIL" and r["classification"] == "CONFLICT" and r["crt_candidates"] == {"EMISSOR": "1", "C000004": "3"}
|
|
ok = resolve(base, {"crt_source": "EMISSOR"})
|
|
assert ok["status"] == "PASS" and ok["company_crt"] == "1" and ok["classification"] == "CONFIGURED" and ok["confidence"] == "MEDIUM"
|
|
assert resolve(base, {"crt_source": "C000004"})["company_crt"] == "3"
|
|
assert resolve(base, {"crt_source_invalid": True})["classification"] == "CONFIG_REQUIRED"
|
|
|
|
|
|
def test_crt_unknown_fails_and_single_source_is_flagged():
|
|
assert resolve(ev([co("1", CN_A, crt=None)], [emis(crt="")]))["classification"] == "UNRESOLVED"
|
|
r = resolve(ev([co("1", CN_A, crt=None)], [emis(crt="1")]))
|
|
assert r["status"] == "PASS" and "CRT_HAS_A_SINGLE_SOURCE" in r["warnings"]
|
|
|
|
|
|
def test_optante_inconsistency_is_a_warning_not_a_guess():
|
|
r = resolve(ev([co("1", CN_A, crt="3", optante="S")], [emis(crt="3")]))
|
|
assert r["status"] == "PASS" and any("OPTANTE_SIMPLES" in w for w in r["warnings"])
|
|
|
|
|
|
def test_full_cnpj_is_never_exposed():
|
|
r = resolve(ev([co("1", CN_A)], [emis()], DOC_A))
|
|
dump = str(preflight_report(r)) + str(r)
|
|
assert CN_A not in dump and mask_cnpj(CN_A) == "11**********11"
|
|
|
|
|
|
def test_read_company_context_fails_closed(monkeypatch):
|
|
from gti_fiscal_agent import company_context as cc, firebird_reader as fr
|
|
monkeypatch.setattr(cc, "preflight_company_context", lambda env: resolve(ev([co("1", CN_A), co("2", CN_B)], [])))
|
|
with pytest.raises(CompanyContextError):
|
|
fr.read_company_context({})
|
|
monkeypatch.setattr(cc, "preflight_company_context", lambda env: resolve(ev([co("1", CN_A)], [emis()], DOC_A)))
|
|
assert fr.read_company_context({})["uf"] == "BA"
|
|
|
|
|
|
def test_config_from_env_and_legacy_parser_unchanged():
|
|
assert config_from_env({"COMPANY_CODE": " 000005 ", "COMPANY_CNPJ": "11.111.111/0001-11", "COMPANY_CRT_SOURCE": "emissor"}) == {"company_code": "000005", "company_cnpj": "11111111000111", "crt_source": "EMISSOR", "crt_source_invalid": False, "uf_override": None}
|
|
assert config_from_env({"COMPANY_CRT_SOURCE": "x"})["crt_source_invalid"] is True
|
|
assert parse_company_context([("ba", "1")]) == {"uf": "BA", "crt": "1"}
|
|
|
|
|
|
def test_auto_crt_source_and_uf_override_semantics():
|
|
assert config_from_env({"COMPANY_CRT_SOURCE": "AUTO"}) ["crt_source"] is None and config_from_env({"COMPANY_CRT_SOURCE": "AUTO"})["crt_source_invalid"] is False
|
|
conflict = ev([co("1", CN_A, crt="3")], [emis(crt="1")], DOC_A)
|
|
assert resolve(conflict, config_from_env({"COMPANY_CRT_SOURCE": "AUTO"}))["status"] == "FAIL" # AUTO never hides a CRT conflict
|
|
good = ev([co("1", CN_A)], [emis()], DOC_A)
|
|
assert resolve(good, {"uf_override": "BA"})["status"] == "PASS"
|
|
r = resolve(good, {"uf_override": "SP"})
|
|
assert r["status"] == "FAIL" and r["classification"] == "CONFLICT" # override never replaces the resolver silently
|
|
blank = ev([co("1", CN_A, uf="")], [emis(uf="")])
|
|
r = resolve(blank, {"uf_override": "BA"})
|
|
assert r["status"] == "PASS" and "UF_FROM_EXPLICIT_OVERRIDE" in r["warnings"]
|