@@ -48,17 +48,34 @@ def test_colorized_detection_checks_for_environment_variables(self):
48
48
({"PYTHON_COLORS" : "1" }, True ),
49
49
({"PYTHON_COLORS" : "0" }, False ),
50
50
({"NO_COLOR" : "1" }, False ),
51
+ ({"NO_COLOR" : "0" }, False ),
51
52
({"NO_COLOR" : "1" , "PYTHON_COLORS" : "1" }, True ),
52
53
({"FORCE_COLOR" : "1" }, True ),
53
54
({"FORCE_COLOR" : "1" , "NO_COLOR" : "1" }, False ),
55
+ ({"FORCE_COLOR" : "1" , "NO_COLOR" : "0" }, False ),
56
+ ({"FORCE_COLOR" : "1" , "NO_COLOR" : "" }, True ),
57
+ ({"FORCE_COLOR" : "0" , "NO_COLOR" : "1" }, False ),
58
+ ({"FORCE_COLOR" : "" , "NO_COLOR" : "1" }, False ),
54
59
({"FORCE_COLOR" : "1" , "PYTHON_COLORS" : "0" }, False ),
60
+ ({"FORCE_COLOR" : "0" , "PYTHON_COLORS" : "0" }, False ),
55
61
]:
56
62
with self .subTest (env_vars = env_vars , expected = expected ):
57
63
with unittest .mock .patch ("os.environ" , env_vars ):
58
64
self .assertEqual (_colorize .can_colorize (), expected )
59
65
66
+ with unittest .mock .patch ("os.environ" , {"NO_COLOR" : "" }):
67
+ if sys .platform == "win32" :
68
+ vt_mock .return_value = False
69
+ self .assertEqual (_colorize .can_colorize (), False )
70
+
71
+ vt_mock .return_value = True
72
+ self .assertEqual (_colorize .can_colorize (), True )
73
+ else :
74
+ self .assertEqual (_colorize .can_colorize (), True )
75
+
60
76
with unittest .mock .patch ("os.environ" , {}):
61
77
if sys .platform == "win32" :
78
+ vt_mock .return_value = False
62
79
self .assertEqual (_colorize .can_colorize (), False )
63
80
64
81
vt_mock .return_value = True
0 commit comments