1 files changed, 21 insertions, 0 deletions
diff --git a/tests/logging_handler_test.py b/tests/logging_handler_test.py
new file mode 100644
index 0000000..fe68593
--- /dev/null
+++ b/tests/logging_handler_test.py
@@ -0,0 +1,21 @@
+import logging
+
+from pre_commit import color
+from pre_commit.logging_handler import LoggingHandler
+
+
+def _log_record(message, level):
+    return logging.LogRecord('name', level, '', 1, message, {}, None)
+
+
+def test_logging_handler_color(cap_out):
+    handler = LoggingHandler(True)
+    handler.emit(_log_record('hi', logging.WARNING))
+    ret = cap_out.get()
+    assert ret == f'{color.YELLOW}[WARNING]{color.NORMAL} hi\n'
+
+
+def test_logging_handler_no_color(cap_out):
+    handler = LoggingHandler(False)
+    handler.emit(_log_record('hi', logging.WARNING))
+    assert cap_out.get() == '[WARNING] hi\n'