1 files changed, 59 insertions, 0 deletions
diff --git a/tests/color_test.py b/tests/color_test.py
new file mode 100644
index 0000000..98b39c1
--- /dev/null
+++ b/tests/color_test.py
@@ -0,0 +1,59 @@
+import sys
+from unittest import mock
+
+import pytest
+
+from pre_commit import envcontext
+from pre_commit.color import format_color
+from pre_commit.color import GREEN
+from pre_commit.color import use_color
+
+
+@pytest.mark.parametrize(
+    ('in_text', 'in_color', 'in_use_color', 'expected'), (
+        ('foo', GREEN, True, f'{GREEN}foo\033[m'),
+        ('foo', GREEN, False, 'foo'),
+    ),
+)
+def test_format_color(in_text, in_color, in_use_color, expected):
+    ret = format_color(in_text, in_color, in_use_color)
+    assert ret == expected
+
+
+def test_use_color_never():
+    assert use_color('never') is False
+
+
+def test_use_color_always():
+    assert use_color('always') is True
+
+
+def test_use_color_no_tty():
+    with mock.patch.object(sys.stdout, 'isatty', return_value=False):
+        assert use_color('auto') is False
+
+
+def test_use_color_tty_with_color_support():
+    with mock.patch.object(sys.stdout, 'isatty', return_value=True):
+        with mock.patch('pre_commit.color.terminal_supports_color', True):
+            with envcontext.envcontext((('TERM', envcontext.UNSET),)):
+                assert use_color('auto') is True
+
+
+def test_use_color_tty_without_color_support():
+    with mock.patch.object(sys.stdout, 'isatty', return_value=True):
+        with mock.patch('pre_commit.color.terminal_supports_color', False):
+            with envcontext.envcontext((('TERM', envcontext.UNSET),)):
+                assert use_color('auto') is False
+
+
+def test_use_color_dumb_term():
+    with mock.patch.object(sys.stdout, 'isatty', return_value=True):
+        with mock.patch('pre_commit.color.terminal_supports_color', True):
+            with envcontext.envcontext((('TERM', 'dumb'),)):
+                assert use_color('auto') is False
+
+
+def test_use_color_raises_if_given_shenanigans():
+    with pytest.raises(ValueError):
+        use_color('herpaderp')