Test handlers

mathom · mathom · commit dc0b2890aa75 · 2014-10-15T13:19:51.000-05:00
diff --git a/tests/test_init.py b/tests/test_init.py
@@ -96,7 +96,7 @@ def test_get_logger_reuse(self):
         logger = stackify.getLogger(config=self.config, auto_shutdown=False)
         self.loggers.append(logger)
         logger_two = stackify.getLogger(config=self.config, auto_shutdown=False)
-        self.assertEqual(id(logger_two), id(logger))
+        self.assertIs(logger_two, logger)
 
     def test_logger_atexit(self):
         '''Logger registers an atexit function to clean up'''
@@ -106,6 +106,12 @@ def test_logger_atexit(self):
             self.loggers.append(logger)
         func.assert_called_with(stackify.stopLogging, logger)
 
+    def test_get_handlers(self):
+        '''Registered handlers are provided by getHandlers'''
+        logger = stackify.getLogger(config=self.config, auto_shutdown=False)
+        self.loggers.append(logger)
+        self.assertEqual(logger.handlers, stackify.getHandlers(logger))
+
 
 if __name__=='__main__':
     unittest.main()
