From 8b0a2abaf9c942f2fd49827e898700de54fdb8af Mon Sep 17 00:00:00 2001
From: Anthony Sottile <asottile@umich.edu>
Date: Mon, 7 Apr 2014 16:59:20 -0700
Subject: [PATCH] Add failing 2.6 test

---
 tests/logging_handler_test.py | 38 +++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 tests/logging_handler_test.py

diff --git a/tests/logging_handler_test.py b/tests/logging_handler_test.py
new file mode 100644
index 00000000..d2fed418
--- /dev/null
+++ b/tests/logging_handler_test.py
@@ -0,0 +1,38 @@
+import __builtin__
+import mock
+import pytest
+
+from pre_commit import color
+from pre_commit.logging_handler import LoggingHandler
+
+
+@pytest.yield_fixture
+def print_mock():
+    with mock.patch.object(__builtin__, 'print', autospec=True) as print_mock:
+        yield print_mock
+
+
+class FakeLogRecord(object):
+    def __init__(self, message, levelname, levelno):
+        self.message = message
+        self.levelname = levelname
+        self.levelno = levelno
+
+    def getMessage(self):
+        return self.message
+
+
+def test_logging_handler_color(print_mock):
+    handler = LoggingHandler(True)
+    handler.emit(FakeLogRecord('hi', 'WARNING', 30))
+    print_mock.assert_called_once_with(
+        color.YELLOW + '[WARNING]' + color.NORMAL + ' hi',
+    )
+
+
+def test_logging_handler_no_color(print_mock):
+    handler = LoggingHandler(False)
+    handler.emit(FakeLogRecord('hi', 'WARNING', 30))
+    print_mock.assert_called_once_with(
+        '[WARNING] hi',
+    )