From 7d7da387664188314c2f75f1899a960c8f691661 Mon Sep 17 00:00:00 2001
From: Danimar Ribeiro <danimaribeiro@gmail.com>
Date: Sat, 17 Sep 2016 13:11:09 -0300
Subject: [PATCH] Implementando testes da NFe

---
 pytrustnfe/nfe/__init__.py                |  2 ++
 pytrustnfe/nfe/comunicacao.py             |  5 -----
 pytrustnfe/test/test_consulta_cadastro.py | 19 +++++++++++++++++++
 3 files changed, 21 insertions(+), 5 deletions(-)
 create mode 100644 pytrustnfe/test/test_consulta_cadastro.py

diff --git a/pytrustnfe/nfe/__init__.py b/pytrustnfe/nfe/__init__.py
index be01ede..bf080c6 100644
--- a/pytrustnfe/nfe/__init__.py
+++ b/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)
diff --git a/pytrustnfe/nfe/comunicacao.py b/pytrustnfe/nfe/comunicacao.py
index 3f26d56..f15fe15 100644
--- a/pytrustnfe/nfe/comunicacao.py
+++ b/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"?>'
diff --git a/pytrustnfe/test/test_consulta_cadastro.py b/pytrustnfe/test/test_consulta_cadastro.py
new file mode 100644
index 0000000..9c63f03
--- /dev/null
+++ b/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')