Resolve #91 - PR de correção de Tests - assert_console

.gitignore

@@ -2,6 +2,11 @@ Pipfile
 Pipfile.lock
 .vscode/*
 .idea/*
+.tmp/*
+.tmp
+.pytest_cache/*
+tests/discentes*
+tests/main.py
 venv/*
 dist/*
 build/*

odufrn_downloader/modules/Env.py

@@ -20,7 +20,7 @@ class Env(ABC):
     MSG_ERRORS = {
         'download_error': (
             "Ocorreu algum erro durante o download do pacote."
-            "Verifique sua conexão, o nome do conjunto de dados"
+            "Verifique sua conexão, o nome do conjunto de dados "
             "e tente novamente."
         ),
         'none_package': 'Nenhum pacote foi encontrado',
@@ -34,11 +34,11 @@ def __init__(self):
     def _print_exception(self, ex: Exception,
                          msg: str = MSG_ERRORS['download_error']):
         """Imprime mensagem padrão para exceções."""
-        print('\033[91m{}\033[0m'.format(ex))
+        print('\033[91m{}\033[0m'.format(type(ex).__name__))
         print(msg)
 
     def _print_not_found(self, name: str, type_name: str):
-        """Imprime mensagem padrão para nome de dados não encontrados.
+        """Imprime mensagem padrão para nomes de dados não encontrados.
         """
         print('{} de dados "{}" não foi encontrado.'.format(type_name, name))
 

tests/test_env.py

@@ -6,21 +6,23 @@
 
 class Env(unittest.TestCase):
     def setUp(self):
-        """ Inicia novo objeto em todo os testes """
+        """ Inicia novo objeto em todos os testes """
         self.ufrn_data = ODUFRNDownloader()
         self.test_dir = 'temporary_test_dir'
 
     def test_can_print_exception(self):
         """ Verifica se uma exceção consegue ser printada no console """
-        assert_console(lambda: self.ufrn_data._print_exception(ValueError()))
+        assert_console(lambda: self.ufrn_data._print_exception(ValueError()),
+                       "\033[91mValueError\033[0m\nOcorreu algum erro durante"
+                       " o download do pacote.Verifique sua conexão, o nome do"
+                       " conjunto de dados e tente novamente.\n")
 
-    def test_can_print_list(self):
+    def test_can_print_group_list(self):
         """ Verifica se consegue-se printar informação da lista no console """
         assert_console(
             lambda: self.ufrn_data._print_list(
-                "conjuntos de dados", ['discente']
-            )
-        )
+                "grupos de dados", ['discente']
+            ), "Os grupos de dados disponíveis são:\n['discente']\n")
 
     def test_can_load_list(self):
         """ Verifica se consegue carregar uma lista advinda de requisição """

tests/test_group.py

@@ -8,7 +8,20 @@ def setUp(self):
 
     def test_can_print_groups(self):
         """Verifica se a lista de grupos é impressa na tela """
-        assert_console(self.ufrn_data.print_groups)
+        assert_console(self.ufrn_data.print_groups,
+                       """Os grupos de dados disponíveis são:
+[   'biblioteca',
+    'comunicados-e-documentos',
+    'contratos-e-convenios',
+    'despesas-e-orcamento',
+    'ensino',
+    'extensao',
+    'institucional',
+    'materiais',
+    'patrimonio',
+    'pesquisa',
+    'pessoas',
+    'processos']\n""")
 
     def test_can_load_groups(self):
         """Verifica se a lista de grupos é carregada no objeto """
@@ -60,12 +73,18 @@ def test_can_search_groups(self):
 
     def test_can_print_files_from_group(self):
         """Verifica se os arquivos de um grupo podem ser impresso na tela."""
+        # Uma vez que os arquivos não são estáticos
+        # e que a extensão de alguns pode mudar, não seria
+        # dinâmico fazer esse assert com mensagem especificada (message!=None)
         assert_console(
             lambda: self.ufrn_data.print_files_from_group('processos')
         )
 
     def test_can_print_files_from_group_with_typo(self):
         """Verifica se o tratamento de erro com o Levenshtein funciona."""
         assert_console(
-            lambda: self.ufrn_data.print_files_from_package('process')
+            lambda: self.ufrn_data.print_files_from_package('process'),
+            """TypeError
+
+Você pode estar procurando por processos ou processos-seletivos\n"""
         )

tests/utils.py

@@ -3,9 +3,10 @@
 import sys
 import shutil
 import unittest
+from odufrn_downloader import ODUFRNDownloader
 from os.path import dirname, join, abspath
+
 sys.path.insert(0, abspath(join(dirname(__file__), '..')))
-from odufrn_downloader import ODUFRNDownloader
 
 
 def input_value(fun):
@@ -17,8 +18,11 @@ def input_value(fun):
     return capturedOutput.getvalue()
 
 
-def assert_console(fun):
+def assert_console(fun, message: str = None):
     """Recebe função que printa algo na tela e realiza assert
     que verifica se foi printado."""
     unit = unittest.TestCase()
-    return unit.assertTrue(len(input_value(fun)) > 0)
+    if message is not None:
+        return unit.assertEqual(message, input_value(fun))
+    else:
+        return unit.assertTrue(len(input_value(fun)) > 0)