@@ -4314,24 +4314,25 @@ def foo(): |
4314 | 4314 | self.assertEqual(actual, expected) |
4315 | 4315 |
|
4316 | 4316 | def test_colorized_detection_checks_for_environment_variables(self): |
4317 | | - with unittest.mock.patch("os.isatty") as isatty_mock: |
4318 | | - isatty_mock.return_value = True |
4319 | | - with unittest.mock.patch("os.environ", {'TERM': 'dumb'}): |
| 4317 | + with unittest.mock.patch("nt._supports_virtual_terminal", return_value=True): |
| 4318 | + with unittest.mock.patch("os.isatty") as isatty_mock: |
| 4319 | + isatty_mock.return_value = True |
| 4320 | + with unittest.mock.patch("os.environ", {'TERM': 'dumb'}): |
| 4321 | + self.assertEqual(traceback._can_colorize(), False) |
| 4322 | + with unittest.mock.patch("os.environ", {'PY_COLORS': '1'}): |
| 4323 | + self.assertEqual(traceback._can_colorize(), True) |
| 4324 | + with unittest.mock.patch("os.environ", {'PY_COLORS': '0'}): |
| 4325 | + self.assertEqual(traceback._can_colorize(), False) |
| 4326 | + with unittest.mock.patch("os.environ", {'NO_COLOR': '1'}): |
| 4327 | + self.assertEqual(traceback._can_colorize(), False) |
| 4328 | + with unittest.mock.patch("os.environ", {'NO_COLOR': '1', "PY_COLORS": '1'}): |
| 4329 | + self.assertEqual(traceback._can_colorize(), False) |
| 4330 | + with unittest.mock.patch("os.environ", {'FORCE_COLOR': '1'}): |
| 4331 | + self.assertEqual(traceback._can_colorize(), True) |
| 4332 | + with unittest.mock.patch("os.environ", {'FORCE_COLOR': '1', 'NO_COLOR': '1'}): |
| 4333 | + self.assertEqual(traceback._can_colorize(), False) |
| 4334 | + isatty_mock.return_value = False |
4320 | 4335 | self.assertEqual(traceback._can_colorize(), False) |
4321 | | - with unittest.mock.patch("os.environ", {'PY_COLORS': '1'}): |
4322 | | - self.assertEqual(traceback._can_colorize(), True) |
4323 | | - with unittest.mock.patch("os.environ", {'PY_COLORS': '0'}): |
4324 | | - self.assertEqual(traceback._can_colorize(), False) |
4325 | | - with unittest.mock.patch("os.environ", {'NO_COLOR': '1'}): |
4326 | | - self.assertEqual(traceback._can_colorize(), False) |
4327 | | - with unittest.mock.patch("os.environ", {'NO_COLOR': '1', "PY_COLORS": '1'}): |
4328 | | - self.assertEqual(traceback._can_colorize(), False) |
4329 | | - with unittest.mock.patch("os.environ", {'FORCE_COLOR': '1'}): |
4330 | | - self.assertEqual(traceback._can_colorize(), True) |
4331 | | - with unittest.mock.patch("os.environ", {'FORCE_COLOR': '1', 'NO_COLOR': '1'}): |
4332 | | - self.assertEqual(traceback._can_colorize(), False) |
4333 | | - isatty_mock.return_value = False |
4334 | | - self.assertEqual(traceback._can_colorize(), False) |
4335 | 4336 |
|
4336 | 4337 | if __name__ == "__main__": |
4337 | 4338 | unittest.main() |