diff --git a/validations_libs/tests/cli/fakes.py b/validations_libs/tests/cli/fakes.py index 73d5a493..02df08f7 100644 --- a/validations_libs/tests/cli/fakes.py +++ b/validations_libs/tests/cli/fakes.py @@ -13,10 +13,6 @@ # under the License. # -try: - from unittest import mock -except ImportError: - import mock from unittest import TestCase from validations_libs.cli import app diff --git a/validations_libs/tests/cli/test_colors.py b/validations_libs/tests/cli/test_colors.py index 0176fc32..a1bce47f 100644 --- a/validations_libs/tests/cli/test_colors.py +++ b/validations_libs/tests/cli/test_colors.py @@ -26,8 +26,8 @@ class TestColors(TestCase): RED = "\033[1;31m" GREEN = "\033[0;32m" CYAN = "\033[36m" - RESET = "\033[0;0m" YELLOW = "\033[0;33m" + self.RESET = "\033[0;0m" self.status_color = { 'starting': CYAN, @@ -50,7 +50,7 @@ class TestColors(TestCase): color = self.status_color[status] colored_output = colors.color_output("fizz", status=status) #Checking reset color - self.assertEqual(colored_output[-6:], '\033[0;0m') + self.assertEqual(colored_output[-6:], self.RESET) #Checking output color self.assertEqual(colored_output[:len(color)], color) #Checking output string @@ -61,7 +61,7 @@ class TestColors(TestCase): color = self.status_color['UNKNOWN'] colored_output = colors.color_output("buzz") #Checking reset color - self.assertEqual(colored_output[-6:], '\033[0;0m') + self.assertEqual(colored_output[-6:], self.RESET) #Checking output color self.assertEqual(colored_output[:len(color)], color) #Checking output string diff --git a/validations_libs/tests/cli/test_history.py b/validations_libs/tests/cli/test_history.py index 64cd9d8a..c5b98639 100644 --- a/validations_libs/tests/cli/test_history.py +++ b/validations_libs/tests/cli/test_history.py @@ -16,7 +16,6 @@ try: from unittest import mock except ImportError: import mock -from unittest import TestCase from validations_libs.cli import history from validations_libs.tests import fakes diff --git a/validations_libs/tests/cli/test_list.py b/validations_libs/tests/cli/test_list.py index 1a5ad41c..b0912209 100644 --- a/validations_libs/tests/cli/test_list.py +++ b/validations_libs/tests/cli/test_list.py @@ -16,7 +16,6 @@ try: from unittest import mock except ImportError: import mock -from unittest import TestCase from validations_libs.cli import lister from validations_libs.tests import fakes diff --git a/validations_libs/tests/cli/test_run.py b/validations_libs/tests/cli/test_run.py index c4bca8c3..afce1b8a 100644 --- a/validations_libs/tests/cli/test_run.py +++ b/validations_libs/tests/cli/test_run.py @@ -18,7 +18,6 @@ try: from unittest import mock except ImportError: import mock -from unittest import TestCase from validations_libs.cli import run from validations_libs.tests import fakes diff --git a/validations_libs/tests/cli/test_show.py b/validations_libs/tests/cli/test_show.py index b9915fa6..31d7ae64 100644 --- a/validations_libs/tests/cli/test_show.py +++ b/validations_libs/tests/cli/test_show.py @@ -16,7 +16,6 @@ try: from unittest import mock except ImportError: import mock -from unittest import TestCase from validations_libs.cli import show from validations_libs.tests import fakes