You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1.4 KiB
37 lines
1.4 KiB
# coding=utf-8
|
|
|
|
import os.path
|
|
import unittest
|
|
from lxml import etree
|
|
from pytrustnfe.xml import render_xml
|
|
from pytrustnfe.xml import sanitize_response
|
|
|
|
|
|
class test_xml_serializacao(unittest.TestCase):
|
|
def test_serializacao_default(self):
|
|
path = os.path.join(os.path.dirname(__file__), "XMLs")
|
|
xml = render_xml(
|
|
path, "jinja_template.xml", False, tag1="oi", tag2="ola", tag3="comovai"
|
|
)
|
|
|
|
result = open(os.path.join(path, "jinja_result.xml"), "r").read()
|
|
self.assertEqual(xml + "\n", result)
|
|
|
|
def test_serializacao_remove_empty(self):
|
|
path = os.path.join(os.path.dirname(__file__), "XMLs")
|
|
xmlElem = render_xml(
|
|
path, "jinja_template.xml", True, tag1="oi", tag2="ola", tag3="comovai"
|
|
)
|
|
xml = etree.tostring(xmlElem, encoding=str)
|
|
result = open(os.path.join(path, "jinja_remove_empty.xml"), "r").read()
|
|
self.assertEqual(xml + "\n", result)
|
|
|
|
def test_sanitize_response(self):
|
|
path = os.path.join(os.path.dirname(__file__), "XMLs")
|
|
xml_to_clear = open(os.path.join(path, "jinja_result.xml"), "r").read()
|
|
xml, obj = sanitize_response(xml_to_clear)
|
|
self.assertEqual(xml, xml_to_clear)
|
|
self.assertEqual(obj.tpAmb, "oi")
|
|
self.assertEqual(obj.CNPJ, "ola")
|
|
self.assertEqual(obj.indNFe, "")
|
|
self.assertEqual(obj.indEmi, "comovai")
|