Agregar pruebas unitarias

.gitignore

@@ -109,6 +109,7 @@ celerybeat.pid
 .env
 .venv
 env/
+ev/
 venv/
 ENV/
 env.bak/

requirements.txt

@@ -1,2 +1,2 @@
 xmlsec
-cryptography==3.4.8
+cryptography

source/cfdi_cert.py

@@ -66,7 +66,9 @@ class SATCertificate(object):
         return digest.finalize()
 
     def _get_data_cer(self, cer):
+
         obj = x509.load_der_x509_certificate(cer, default_backend())
+
         self._rfc = obj.subject.get_attributes_for_oid(
             NameOID.X500_UNIQUE_IDENTIFIER)[0].value.split(' ')[0]
         self._serial_number2 = obj.serial_number

source/tests/tests_cert.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+
+import sys
+import unittest
+from pathlib import Path
+
+sys.path.append('..')
+from cfdi_cert import SATCertificate
+
+
+PASSWORD = '12345678a'
+PATH_CERT = '/home/mau/Projects/cfdi/cfdi-cert/cert'
+
+
+class TestCert(unittest.TestCase):
+
+    def setUp(self):
+        print(f'In method: {self._testMethodName}')
+
+    def test_validate_cert_vigente(self):
+        with open(f'{PATH_CERT}/vigente.cer', 'rb') as f:
+            cer = f.read()
+        with open(f'{PATH_CERT}/vigente.key', 'rb') as f:
+            key = f.read()
+
+        cert = SATCertificate(cer, key, password=PASSWORD)
+
+        self.assertTrue(cert.is_valid)
+        self.assertTrue(cert.is_valid_time)
+        self.assertTrue(cert.are_couple)
+        self.assertFalse(cert.is_fiel)
+
+    def test_validate_cert_vencido(self):
+        with open(f'{PATH_CERT}/vencido.cer', 'rb') as f:
+            cer = f.read()
+        with open(f'{PATH_CERT}/vencido.key', 'rb') as f:
+            key = f.read()
+
+        cert = SATCertificate(cer, key, password=PASSWORD)
+
+        self.assertFalse(cert.is_valid)
+        self.assertFalse(cert.is_valid_time)
+
+
+if __name__ == '__main__':
+    unittest.main()