diff --git a/pytrustnfe/test/test_assinatura.py b/pytrustnfe/test/test_assinatura.py index d9aa829..4d2fad9 100644 --- a/pytrustnfe/test/test_assinatura.py +++ b/pytrustnfe/test/test_assinatura.py @@ -18,6 +18,12 @@ XML_ASSINAR = '' \ ' ' \ '' +XML_ERRADO = '' \ + '' \ + ' ' \ + ' Hello, World!' \ + ' ' \ + '' class test_assinatura(unittest.TestCase): @@ -32,6 +38,10 @@ class test_assinatura(unittest.TestCase): assinatura = Assinatura(os.path.join(self.caminho,'teste.pfx'), '123') self.assertRaises(Exception, assinatura.assina_xml, XML_ASSINAR) + def test_assinar_xml_invalido(self): + assinatura = Assinatura(os.path.join(self.caminho,'teste.pfx'), '123456') + self.assertRaises(RuntimeError, assinatura.assina_xml, XML_ERRADO) + def test_assinar_xml_valido(self): assinatura = Assinatura(os.path.join(self.caminho,'teste.pfx'), '123456') xml = assinatura.assina_xml(XML_ASSINAR) @@ -39,7 +49,4 @@ class test_assinatura(unittest.TestCase): self.assertEqual(xml_assinado, xml, 'Xml assinado é inválido') -if __name__ == "__main__": - #import sys;sys.argv = ['', 'Test.testName'] - unittest.main() \ No newline at end of file