Skip to content

Commit

Permalink
[TEMP] l10n_br_mdfe_spec: enable test
Browse files Browse the repository at this point in the history
  • Loading branch information
@marcelsavegnago
marcelsavegnago committed Oct 20, 2024
1 parent d9bcf20 commit 86c8bbe
Showing 1 changed file with 145 additions and 156 deletions.
301 changes: 145 additions & 156 deletions l10n_br_mdfe_spec/tests/test_mdfe_import.py
View file
Original file line number Diff line number Diff line change
@@ -1,156 +1,145 @@
# # Copyright 2020 Akretion - Raphael Valyi <[email protected]>
# # License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl-3.0.en.html).

# import re
# from datetime import datetime

# import nfelib
# import pkg_resources
# from nfelib.mdfe.bindings.v3_0.mdfe_v3_00 import Mdfe
# from xsdata.formats.dataclass.parsers import XmlParser

# from odoo import api
# from odoo.tests import TransactionCase

# from ..models import spec_mixin

# tz_datetime = re.compile(r".*[-+]0[0-9]:00$")


# @api.model
# def build_fake(self, node, create=False):
# attrs = self.build_attrs_fake(node, create_m2o=True)
# return self.new(attrs)


# # flake8: noqa: C901
# @api.model
# def build_attrs_fake(self, node, create_m2o=False):
# """
# Similar to build_attrs from spec_driven_model but simpler: assuming
# generated abstract mixins are not injected into concrete Odoo models.
# """
# fields = self.fields_get()
# vals = self.default_get(fields.keys())

# spec_prefix = "mdfe30"
# for fname, fspec in node.__dataclass_fields__.items():
# if fname == "any_element":
# continue
# value = getattr(node, fname)
# if value is None:
# continue
# key = "%s%s" % (
# spec_prefix,
# fspec.metadata.get("name", fname),
# )
# if (
# fspec.type == str or not any(["." in str(i) for i in fspec.type.__args__])
# ) and not str(fspec.type).startswith("typing.List"):
# # SimpleType
# if fields[key]["type"] == "datetime":
# if "T" in value:
# if tz_datetime.match(value):
# old_value = value
# value = old_value[:19]
# # TODO see python3/pysped/xml_sped/base.py#L692
# value = datetime.strptime(value, "%Y-%m-%dT%H:%M:%S")
# vals[key] = value

# else:
# if hasattr(fspec.type.__args__[0], "__name__"):
# binding_type = fspec.type.__args__[0].__name__
# else:
# binding_type = fspec.type.__args__[0].__forward_arg__

# # ComplexType
# if fields.get(key) and fields[key].get("related"):
# key = fields[key]["related"][0]
# comodel_name = fields[key]["relation"]
# else:
# clean_type = binding_type.lower() # TODO double check
# comodel_name = "%s.%s.%s" % (
# self._schema_name,
# self._schema_version.replace(".", "")[0:2],
# clean_type.split(".")[-1],
# )
# comodel = self.env.get(comodel_name)
# if comodel is None: # example skip ICMS100 class
# continue

# if not str(fspec.type).startswith("typing.List"):
# # m2o
# new_value = comodel.build_attrs_fake(
# value,
# create_m2o=create_m2o,
# )
# if new_value is None:
# continue
# if comodel._name == self._name: # stacked m2o
# vals.update(new_value)
# else:
# vals[key] = self.match_or_create_m2o_fake(
# comodel, new_value, create_m2o
# )
# else: # if attr.get_container() == 1:
# # o2m
# lines = []
# for line in [li for li in value if li]:
# line_vals = comodel.build_attrs_fake(line, create_m2o=create_m2o)
# lines.append((0, 0, line_vals))
# vals[key] = lines

# for k, v in fields.items():
# if (
# v.get("related") is not None
# and len(v["related"]) == 1
# and vals.get(k) is not None
# ):
# vals[v["related"][0]] = vals.get(k)

# return vals


# @api.model
# def match_or_create_m2o_fake(self, comodel, new_value, create_m2o=False):
# return comodel.new(new_value).id


# # spec_models.CteSpecMixin._update_cache = _update_cache
# spec_mixin.MdfeSpecMixin.build_fake = build_fake
# spec_mixin.MdfeSpecMixin.build_attrs_fake = build_attrs_fake
# spec_mixin.MdfeSpecMixin.match_or_create_m2o_fake = match_or_create_m2o_fake


# class MdfeImportTest(TransactionCase):
# def test_import_mdfe(self):
# res_items = (
# "mdfe",
# "samples",
# "v3_0",
# "41190876676436000167580010000500001000437558-mdfe.xml",
# )
# resource_path = "/".join(res_items)
# nfe_stream = pkg_resources.resource_stream(nfelib.__name__, resource_path)
# parser = XmlParser()
# binding = parser.from_string(nfe_stream.read().decode(), Mdfe)
# inf_mdfe = self.env["mdfe.30.tmdfe_infmdfe"].build_fake(
# binding.infMDFe, create=False
# )
# self.assertEqual(inf_mdfe.mdfe30_emit.mdfe30_CNPJ, "76676436000167")

# def test_import_mdfe2(self):
# res_items = (
# "mdfe",
# "samples",
# "v3_0",
# "50170876063965000276580010000011311421039568-mdfe.xml",
# )
# resource_path = "/".join(res_items)
# nfe_stream = pkg_resources.resource_stream(nfelib.__name__, resource_path)
# parser = XmlParser()
# binding = parser.from_string(nfe_stream.read().decode(), Mdfe)
# inf_mdfe = self.env["mdfe.30.tmdfe_infmdfe"].build_fake(
# binding.infMDFe, create=False
# )
# self.assertEqual(inf_mdfe.mdfe30_emit.mdfe30_xNome, "TESTE")
# Copyright 2020 Akretion - Raphael Valyi <[email protected]>
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl-3.0.en.html).

import re
from datetime import datetime

import nfelib
import pkg_resources
from nfelib.mdfe.bindings.v3_0.mdfe_v3_00 import Mdfe
from xsdata.formats.dataclass.parsers import XmlParser

from odoo import api
from odoo.tests import SavepointCase

from ..models import spec_mixin

tz_datetime = re.compile(r".*[-+]0[0-9]:00$")


@api.model
def build_fake(self, node, create=False):
attrs = self.build_attrs_fake(node, create_m2o=True)
return self.new(attrs)


# flake8: noqa: C901
@api.model
def build_attrs_fake(self, node, create_m2o=False):
"""
Similar to build_attrs from spec_driven_model but simpler: assuming
generated abstract mixins are not injected into concrete Odoo models.
"""
fields = self.fields_get()
vals = self.default_get(fields.keys())
for fname, fspec in node.__dataclass_fields__.items():
value = getattr(node, fname)
if value is None:
continue
key = f"mdfe30_{fspec.metadata.get('name', fname)}"
if (
fspec.type == str or not any(["." in str(i) for i in fspec.type.__args__])
) and not str(fspec.type).startswith("typing.List"):
# SimpleType
if fields[key]["type"] == "datetime":
if "T" in value:
if tz_datetime.match(value):
old_value = value
value = old_value[:19]
# TODO see python3/pysped/xml_sped/base.py#L692
value = datetime.strptime(value, "%Y-%m-%dT%H:%M:%S")
vals[key] = value

else:
if hasattr(fspec.type.__args__[0], "__name__"):
binding_type = fspec.type.__args__[0].__name__
else:
binding_type = fspec.type.__args__[0].__forward_arg__

# ComplexType
if fields.get(key) and fields[key].get("related"):
key = fields[key]["related"][0]
comodel_name = fields[key]["relation"]
else:
clean_type = binding_type.lower()
comodel_name = f"mdfe.30.{clean_type.split('.')[-1]}"
comodel = self.env.get(comodel_name)
if comodel is None: # example skip ICMS100 class
continue

if not str(fspec.type).startswith("typing.List"):
# m2o
new_value = comodel.build_attrs_fake(
value,
create_m2o=create_m2o,
)
if new_value is None:
continue
if comodel._name == self._name: # stacked m2o
vals.update(new_value)
else:
vals[key] = self.match_or_create_m2o_fake(
comodel, new_value, create_m2o
)
else: # if attr.get_container() == 1:
# o2m
lines = []
for line in [li for li in value if li]:
line_vals = comodel.build_attrs_fake(line, create_m2o=create_m2o)
lines.append((0, 0, line_vals))
vals[key] = lines

for k, v in fields.items():
if (
v.get("related") is not None
and len(v["related"]) == 1
and vals.get(k) is not None
):
vals[v["related"][0]] = vals.get(k)

return vals


@api.model
def match_or_create_m2o_fake(self, comodel, new_value, create_m2o=False):
return comodel.new(new_value).id


# spec_models.CteSpecMixin._update_cache = _update_cache
spec_mixin.MdfeSpecMixin.build_fake = build_fake
spec_mixin.MdfeSpecMixin.build_attrs_fake = build_attrs_fake
spec_mixin.MdfeSpecMixin.match_or_create_m2o_fake = match_or_create_m2o_fake


class MdfeImportTest(SavepointCase):
def test_import_mdfe(self):
res_items = (
"mdfe",
"samples",
"v3_0",
"41190876676436000167580010000500001000437558-mdfe.xml",
)
resource_path = "/".join(res_items)
nfe_stream = pkg_resources.resource_stream(nfelib.__name__, resource_path)
parser = XmlParser()
binding = parser.from_string(nfe_stream.read().decode(), Mdfe)
inf_mdfe = self.env["mdfe.30.tmdfe_infmdfe"].build_fake(
binding.infMDFe, create=False
)
self.assertEqual(inf_mdfe.mdfe30_emit.mdfe30_CNPJ, "76676436000167")

def test_import_mdfe2(self):
res_items = (
"mdfe",
"samples",
"v3_0",
"50170876063965000276580010000011311421039568-mdfe.xml",
)
resource_path = "/".join(res_items)
nfe_stream = pkg_resources.resource_stream(nfelib.__name__, resource_path)
parser = XmlParser()
binding = parser.from_string(nfe_stream.read().decode(), Mdfe)
inf_mdfe = self.env["mdfe.30.tmdfe_infmdfe"].build_fake(
binding.infMDFe, create=False
)
self.assertEqual(inf_mdfe.mdfe30_emit.mdfe30_xNome, "TESTE")

0 comments on commit 86c8bbe

Please sign in to comment.