@@ -4314,24 +4314,25 @@ def foo():
43144314 self .assertEqual (actual , expected )
43154315
43164316 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
43204335 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 )
43354336
43364337if __name__ == "__main__" :
43374338 unittest .main ()
0 commit comments