mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-02-17 08:14:42 +04:00
Add tests for NO_COLOR support
This commit is contained in:
parent
01d3a72a0e
commit
8520455042
1 changed files with 18 additions and 0 deletions
|
|
@ -1,5 +1,6 @@
|
||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
import mock
|
import mock
|
||||||
|
|
@ -50,3 +51,20 @@ def test_use_color_tty_without_color_support():
|
||||||
def test_use_color_raises_if_given_shenanigans():
|
def test_use_color_raises_if_given_shenanigans():
|
||||||
with pytest.raises(InvalidColorSetting):
|
with pytest.raises(InvalidColorSetting):
|
||||||
use_color('herpaderp')
|
use_color('herpaderp')
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_color_env_unset():
|
||||||
|
with mock.patch.dict(os.environ):
|
||||||
|
if 'NO_COLOR' in os.environ:
|
||||||
|
del os.environ['NO_COLOR']
|
||||||
|
assert use_color('always') is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_color_env_empty():
|
||||||
|
with mock.patch.dict(os.environ, NO_COLOR=''):
|
||||||
|
assert use_color('always') is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_color_env_non_empty():
|
||||||
|
with mock.patch.dict(os.environ, NO_COLOR=' '):
|
||||||
|
assert use_color('always') is False
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue