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)
|
|
print(type(xml))
|
|
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')
|