mirror of
https://github.com/python/cpython.git
synced 2025-10-20 16:33:53 +00:00
60 lines
2.6 KiB
Python
60 lines
2.6 KiB
Python
|
import contextlib
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
import unittest.mock
|
||
|
|
import _colorize
|
||
|
|
from test.support import force_not_colorized
|
||
|
|
|
||
|
|
ORIGINAL_CAN_COLORIZE = _colorize.can_colorize
|
||
|
|
|
||
|
|
|
||
|
|
def setUpModule():
|
||
|
|
_colorize.can_colorize = lambda: False
|
||
|
|
|
||
|
|
|
||
|
|
def tearDownModule():
|
||
|
|
_colorize.can_colorize = ORIGINAL_CAN_COLORIZE
|
||
|
|
|
||
|
|
|
||
|
|
class TestColorizeFunction(unittest.TestCase):
|
||
|
|
@force_not_colorized
|
||
|
|
def test_colorized_detection_checks_for_environment_variables(self):
|
||
|
|
if sys.platform == "win32":
|
||
|
|
virtual_patching = unittest.mock.patch("nt._supports_virtual_terminal",
|
||
|
|
return_value=True)
|
||
|
|
else:
|
||
|
|
virtual_patching = contextlib.nullcontext()
|
||
|
|
with virtual_patching:
|
||
|
|
|
||
|
|
flags = unittest.mock.MagicMock(ignore_environment=False)
|
||
|
|
with (unittest.mock.patch("os.isatty") as isatty_mock,
|
||
|
|
unittest.mock.patch("sys.flags", flags),
|
||
|
|
unittest.mock.patch("_colorize.can_colorize", ORIGINAL_CAN_COLORIZE)):
|
||
|
|
isatty_mock.return_value = True
|
||
|
|
with unittest.mock.patch("os.environ", {'TERM': 'dumb'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
with unittest.mock.patch("os.environ", {'PYTHON_COLORS': '1'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), True)
|
||
|
|
with unittest.mock.patch("os.environ", {'PYTHON_COLORS': '0'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
with unittest.mock.patch("os.environ", {'NO_COLOR': '1'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
with unittest.mock.patch("os.environ",
|
||
|
|
{'NO_COLOR': '1', "PYTHON_COLORS": '1'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), True)
|
||
|
|
with unittest.mock.patch("os.environ", {'FORCE_COLOR': '1'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), True)
|
||
|
|
with unittest.mock.patch("os.environ",
|
||
|
|
{'FORCE_COLOR': '1', 'NO_COLOR': '1'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
with unittest.mock.patch("os.environ",
|
||
|
|
{'FORCE_COLOR': '1', "PYTHON_COLORS": '0'}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
isatty_mock.return_value = False
|
||
|
|
with unittest.mock.patch("os.environ", {}):
|
||
|
|
self.assertEqual(_colorize.can_colorize(), False)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|