pre-commit/tests/color_test.py at v5 · pre-commit/pre-commit · GitHub

Repository navigation

History

from __future__ import annotations

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.stderr, 'isatty', return_value=False):

assert use_color('auto') is False

def test_use_color_tty_with_color_support():

with mock.patch.object(sys.stderr, '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.stderr, '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.stderr, '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')

pre-commit/tests/color_test.py at v5 · pre-commit/pre-commit · GitHub

Repository navigation

Expand file tree

Breadcrumbs

color_test.py

Latest commit

History

Breadcrumbs

color_test.py

File metadata and controls