60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
|
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')
|