Add optional support for coloredlogs

1 files changed, 41 insertions, 3 deletions

diff --git a/tests/test_command.py b/tests/test_command.py
index 1d1402c..667fef6 100644
--- a/tests/test_command.py
+++ b/tests/test_command.py
@@ -155,8 +155,9 @@ class testCommand(hazwaz.unittest.HazwazTestCase):
         stream = self.run_with_argv(cmd, ["mycommand"])
         self.assertEqual(stream["stdout"].getvalue(), cmd_help)
 
-    def test_logging_regular(self):
+    def test_logging_regular_coloredlogs(self):
         cmd = MyCommand()
+        cmd.coloredlogs = True
         with self.assertLogs():
             stream = self.run_with_argv(cmd, [
                 "mycommand",
@@ -165,8 +166,9 @@ class testCommand(hazwaz.unittest.HazwazTestCase):
         log_lines = stream["stderr"].getvalue().strip().split("\n")
         self.assertEqual(len(log_lines), 1)
 
-    def test_logging_verbose(self):
+    def test_logging_verbose_coloredlogs(self):
         cmd = MyCommand()
+        cmd.coloredlogs = True
         with self.assertLogs():
             stream = self.run_with_argv(cmd, [
                 "mycommand",
@@ -176,8 +178,44 @@ class testCommand(hazwaz.unittest.HazwazTestCase):
         log_lines = stream["stderr"].getvalue().strip().split("\n")
         self.assertEqual(len(log_lines), 2)
 
-    def test_logging_debug(self):
+    def test_logging_debug_coloredlogs(self):
         cmd = MyCommand()
+        cmd.coloredlogs = True
+        with self.assertLogs():
+            stream = self.run_with_argv(cmd, [
+                "mycommand",
+                "--debug",
+                "loggingsubcommand",
+            ])
+        log_lines = stream["stderr"].getvalue().strip().split("\n")
+        self.assertEqual(len(log_lines), 3)
+
+    def test_logging_regular_no_coloredlogs(self):
+        cmd = MyCommand()
+        cmd.coloredlogs = False
+        with self.assertLogs():
+            stream = self.run_with_argv(cmd, [
+                "mycommand",
+                "loggingsubcommand",
+            ])
+        log_lines = stream["stderr"].getvalue().strip().split("\n")
+        self.assertEqual(len(log_lines), 1)
+
+    def test_logging_verbose_no_coloredlogs(self):
+        cmd = MyCommand()
+        cmd.coloredlogs = False
+        with self.assertLogs():
+            stream = self.run_with_argv(cmd, [
+                "mycommand",
+                "--verbose",
+                "loggingsubcommand",
+            ])
+        log_lines = stream["stderr"].getvalue().strip().split("\n")
+        self.assertEqual(len(log_lines), 2)
+
+    def test_logging_debug_no_coloredlogs(self):
+        cmd = MyCommand()
+        cmd.coloredlogs = False
         with self.assertLogs():
             stream = self.run_with_argv(cmd, [
                 "mycommand",