@@ -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 ()
0 commit comments