Add tests for NO_COLOR support

This commit is contained in:
Edgar Geier 2019-07-22 20:07:16 +02:00
parent 01d3a72a0e
commit 8520455042

View file

@ -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