1111
1212
1313class LogTest (unittest .TestCase ):
14+ old_level = None
15+ old_handlers = None
16+
17+ @classmethod
18+ def setUpClass (cls ):
19+ cls .old_level = logger .level
20+ cls .old_handlers = list (logger .handlers )
21+ for h in cls .old_handlers :
22+ logger .removeHandler (h )
23+
24+ def tearDown (self ):
25+ logger .setLevel (self .old_level )
26+ for h in self .old_handlers :
27+ logger .addHandler (h )
28+
1429 def test_logger (self ):
1530 self .assertIsInstance (logger , logging .Logger )
1631 self .assertEqual (logging .NOTSET , logger .level )
1732 self .assertEqual (1 , len (logger .handlers ))
1833
19- def test_configure_logging (self ):
34+ def test_configure_logging_dict (self ):
2035 logging_config = {
2136 "version" : 1 ,
2237 "formatters" : {
@@ -31,19 +46,31 @@ def test_configure_logging(self):
3146 "loggers" : {"zappend" : {"level" : "INFO" , "handlers" : ["console" ]}},
3247 }
3348
34- old_level = logger . level
35- old_handlers = list ( logger .handlers )
36- for h in old_handlers :
37- logger . removeHandler ( h )
49+ configure_logging ( logging_config )
50+ self . assertEqual ( logging . INFO , logger .level )
51+ self . assertEqual ( 1 , len ( logger . handlers ))
52+ self . assertIsInstance ( logger . handlers [ 0 ], logging . StreamHandler )
3853
39- try :
40- configure_logging (logging_config )
41- self .assertEqual (logging .INFO , logger .level )
42- self .assertEqual (1 , len (logger .handlers ))
43- finally :
44- logger .setLevel (old_level )
45- for h in old_handlers :
46- logger .addHandler (h )
54+ def test_configure_logging_false (self ):
55+ logging_config = False
56+ configure_logging (logging_config )
57+ self .assertEqual (logging .NOTSET , logger .level )
58+ self .assertEqual (1 , len (logger .handlers ))
59+ self .assertIsInstance (logger .handlers [0 ], logging .StreamHandler )
60+
61+ def test_configure_logging_true (self ):
62+ logging_config = True
63+ configure_logging (logging_config )
64+ self .assertEqual (logging .INFO , logger .level )
65+ self .assertEqual (1 , len (logger .handlers ))
66+ self .assertIsInstance (logger .handlers [0 ], logging .StreamHandler )
67+
68+ def test_configure_logging_level (self ):
69+ logging_config = "WARNING"
70+ configure_logging (logging_config )
71+ self .assertEqual (logging .WARNING , logger .level )
72+ self .assertEqual (1 , len (logger .handlers ))
73+ self .assertIsInstance (logger .handlers [0 ], logging .StreamHandler )
4774
4875 def test_get_log_level (self ):
4976 self .assertEqual (logging .DEBUG , get_log_level ("DEBUG" ))
0 commit comments