mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-23 09:20:31 -06:00
Add tests for NO_COLOR support
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from __future__ import unicode_literals
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
import mock
|
||||
@@ -50,3 +51,20 @@ def test_use_color_tty_without_color_support():
|
||||
def test_use_color_raises_if_given_shenanigans():
|
||||
with pytest.raises(InvalidColorSetting):
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user