Updated utils tests to check for null and non-null handlers before and after enabling loggers respectively

Author: Dan

tests/test_utils.py

@@ -1,6 +1,7 @@
 from pssh import utils
 import unittest
 import os
+from logging import NullHandler
 try:
     from cStringIO import StringIO as BytesIO
 except ImportError:
@@ -14,15 +15,21 @@
 class ParallelSSHUtilsTest(unittest.TestCase):
 
     def test_enabling_host_logger(self):
+        self.assertTrue(len([h for h in utils.host_logger.handlers
+                             if isinstance(h, NullHandler)]) == 1)
         utils.enable_host_logger()
-        # And again to test only one handler is attached
+        # And again to test only one non-null handler is attached
         utils.enable_host_logger()
-        self.assertTrue(len(utils.host_logger.handlers) == 1)
+        self.assertTrue(len([h for h in utils.host_logger.handlers
+                             if not isinstance(h, NullHandler)]) == 1)
 
     def test_enabling_pssh_logger(self):
+        self.assertTrue(len([h for h in utils.logger.handlers
+                             if isinstance(h, NullHandler)]) == 1)
         utils.enable_logger(utils.logger)
-        self.assertTrue(len(utils.logger.handlers) == 1)
-    
+        self.assertTrue(len([h for h in utils.host_logger.handlers
+                             if not isinstance(h, NullHandler)]) == 1)
+
     def test_loading_key_files(self):
         for key_filename in [PKEY_FILENAME, DSA_KEY_FILENAME, ECDSA_KEY_FILENAME]:
             pkey = utils.load_private_key(key_filename)