nfelib/tests/cte/test_cte.py at 5612721d731518e1a0cd37b9d2542b138219c518 · akretion/nfelib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# Copyright (C) 2023 - TODAY Raphaël Valyi - Akretion

import os
from pathlib import Path
from unittest import TestCase

from xmldiff import main
from xsdata.formats.dataclass.parsers import XmlParser
from xsdata.formats.dataclass.serializers import XmlSerializer
from xsdata.formats.dataclass.serializers.config import SerializerConfig

from nfelib.cte.bindings import v4_0  # noqa: F401
from nfelib.cte.bindings.v4_0.cte_tipos_basico_v4_00 import Tcte, TcteOs, TcteSimp


class CTeTests(TestCase):
    def test_patched_xsdata_for_ibscsb(self):
        # see https://github.com/akretion/nfelib/pull/151
        # IBSCBS should be TtribCte, not str
        assert (
            str(Tcte.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Optional[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte]"
            # Python < 3.9:
            or str(Tcte.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Union[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte, NoneType]"
        )
        assert (
            str(TcteOs.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Optional[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte]"
            # Python < 3.9:
            or str(TcteOs.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Union[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte, NoneType]"
        )
        assert (
            str(TcteSimp.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Optional[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte]"
            # Python < 3.9:
            or str(TcteSimp.InfCte.Imp().__annotations__["IBSCBS"])
            == "typing.Union[nfelib.cte.bindings.v4_0.dfe_tipos_basicos_v1_00.TtribCte, NoneType]"
        )

    def test_in_out_cte(self):
        path = os.path.join("nfelib", "cte", "samples", "v4_0")
        for filename in os.listdir(path):
            input_file = os.path.join(path, filename)
            parser = XmlParser()
            obj = parser.from_path(Path(input_file))
            serializer = XmlSerializer(config=SerializerConfig(indent="  "))
            xml = serializer.render(
                obj=obj, ns_map={None: "http://www.portalfiscal.inf.br/cte"}
            )

            output_file = "tests/output_cte.xml"
            with open(output_file, "w") as f:
                f.write(xml)

            diff = main.diff_files(input_file, output_file)
            self.assertEqual(
                len(diff),
                0,
                f"Error {output_file} != {input_file}. "
                "Stopping tests here so you can compare XML files.",
            )
            if len(diff) != 0:
                break