Implementando testes da NFe

pytrustnfe/nfe/__init__.py

@@ -72,6 +72,8 @@ def _send(certificado, method, sign, **kwargs):
         signer = Assinatura(certificado.pfx, certificado.password)
         xml_send = signer.assina_xml(
             xmlElem_send, kwargs['NFes'][0]['infNFe']['Id'])
+    else:
+        xml_send = etree.tostring(xmlElem_send)
 
     url = localizar_url(method,  kwargs['estado'], kwargs['ambiente'])
     cabecalho = _build_header(method, **kwargs)

pytrustnfe/nfe/comunicacao.py

@@ -8,11 +8,6 @@ from pytrustnfe.certificado import save_cert_key, extract_cert_and_key_from_pfx
 
 from ..xml import sanitize_response
 
-common_namespaces = {'soap': 'http://www.w3.org/2003/05/soap-envelope'}
-
-soap_body_path = './soap:Envelope/soap:Body'
-soap_fault_path = './soap:Envelope/soap:Body/soap:Fault'
-
 
 def _soap_xml(body, cabecalho):
     xml = '<?xml version="1.0" encoding="utf-8"?>'

pytrustnfe/test/test_consulta_cadastro.py

@@ -0,0 +1,19 @@
+# coding=utf-8
+
+import mock
+import os.path
+import unittest
+from pytrustnfe.certificado import Certificado
+from pytrustnfe.nfe import consulta_cadastro
+
+
+class test_consulta_cadastro(unittest.TestCase):
+
+    caminho = os.path.dirname(__file__)
+
+    def test_conta_de_cadastro(self):
+        pfx_source = open(os.path.join(self.caminho, 'teste.pfx'), 'r').read()
+        pfx = Certificado(pfx_source, '123456')
+
+        obj = {'cnpj': '12345678901234', 'estado': '42'}
+        consulta_cadastro(pfx, obj=obj, ambiente=1, estado='42')