Revisión con tests unitarios cortesía de @javi20gu

Author: hcosta

.gitignore

@@ -4,6 +4,7 @@ __pycache__/
 *$py.class
 Pipfile
 Pipfile.lock
+.vscode/
 
 # C extensions
 *.so

README.md

@@ -4,7 +4,7 @@ La idea es hacer un pequeño gestor de clientes paso a paso añadiendo mejoras p
 
 1. ~~[Desarrollo del prototipo funcional](https://github.com/hcosta/curso-gestor-clientes-python/tree/1.0)~~
 2. ~~[Revisión utilizando clases estáticas](https://github.com/hcosta/curso-gestor-clientes-python/tree/2.0)~~
-3. Revisión añadiendo pruebas unitarias
+3. ~~[Revisión añadiendo pruebas unitarias](https://github.com/hcosta/curso-gestor-clientes-python/tree/3.0)~~
 4. Revisión guardando datos en ficheros
 5. Revisión guardando datos con SQLite
 6. Revisión añadiendo una interfaz gráfica

gestor/manager.py

@@ -55,16 +55,8 @@ def add():
 
     @staticmethod
     def is_valid(dni: str):
-        """
-        Hace diferentes validaciones en el campo dni
-        >>> Manager.is_valid('48H')  # No válido, en uso
-        False
-        >>> Manager.is_valid('X82')  # No válido, incorrecto
-        False
-        >>> Manager.is_valid('21A')  # Válido
-        True
-        """
-        if not re.match('[0-9]{2}[A-Z]', dni):
+        """ Hace diferentes validaciones en el campo dni """
+        if not re.match('[0-9]{2}[A-Z]$', dni):
             return False
         for client in Manager.clients:
             if client.dni == dni:
@@ -102,8 +94,3 @@ def edit():
             Manager.clients[i].apellido = helpers.input_text(2, 30)
             return True
         return False
-
-
-if __name__ == "__main__":
-    import doctest
-    doctest.testmod(extraglobs={'Manager': Manager})

tests/test_manager.py

@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import patch
 
-from gestor.manager import Manager, Client
+from gestor.manager import Client, Manager
 
 
 class TestManager(unittest.TestCase):
@@ -15,11 +15,40 @@ def setUp(self):
 
     def testAddClient(self):
         add_client_inputs = ['Héctor', 'Costa', '39J']
-        with patch('builtins.input',  side_effect=add_client_inputs):
+        with patch('builtins.input', side_effect=add_client_inputs):
             Manager.add()
         self.assertEqual(len(Manager.clients), 4)
         self.assertEqual(Manager.clients[-1].nombre, 'Héctor')
 
+    def testDeleteClient(self):
+        cliente_a_eliminar = Manager.clients[0]
+        delete_client_inputs = [cliente_a_eliminar.dni]
+        self.assertEqual(len(Manager.clients), 3)
+        with patch('builtins.input', side_effect=delete_client_inputs):
+            Manager.delete()
+        self.assertEqual(len(Manager.clients), 2)
+        self.assertTrue(Manager.clients[0] != cliente_a_eliminar)
+        self.assertEqual(Manager.clients[0].nombre, 'Manolo')
+        self.assertEqual(Manager.clients[-1].nombre, 'Ana')
+
+    def testEditClient(self):
+        edit_client_input = ['15J', 'Javier', 'Rosales']
+        with patch('builtins.input', side_effect=edit_client_input):
+            self.assertTrue(Manager.edit())
+        self.assertEqual(Manager.clients[0].nombre, 'Javier')
+
+    def testFindClient(self):
+        find_client_edit = ['15J']
+        with patch('builtins.input', side_effect=find_client_edit):
+            self.assertEqual(Manager.find(), (0, Manager.clients[0]))
+
+    def testDniIsValid(self):
+        self.assertFalse(Manager.is_valid('48H'))
+        self.assertFalse(Manager.is_valid('X82'))
+        self.assertFalse(Manager.is_valid('23223S'))
+        self.assertFalse(Manager.is_valid('13FF'))
+        self.assertTrue(Manager.is_valid('21A'))
+
 
 if __name__ == '__main__':
     unittest.main()