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