""" tests.test_logger ~~~~~~~~~~~~~~~~~~ Tests logger component. """ from collections import namedtuple import logging import unittest from homeassistant.components import logger RECORD = namedtuple('record', ('name', 'levelno')) class TestUpdater(unittest.TestCase): """ Test logger component. """ def setUp(self): """ Create default config """ self.log_config = {'logger': {'default': 'warning', 'logs': {'test': 'info'}}} def tearDown(self): """ Reset logs """ del logging.root.handlers[-1] def test_logger_setup(self): """ Uses logger to create a logging filter """ logger.setup(None, self.log_config) self.assertTrue(len(logging.root.handlers) > 0) handler = logging.root.handlers[-1] self.assertEqual(len(handler.filters), 1) log_filter = handler.filters[0].logfilter self.assertEqual(log_filter['default'], logging.WARNING) self.assertEqual(log_filter['logs']['test'], logging.INFO) def test_logger_test_filters(self): """ Tests resulting filter operation """ logger.setup(None, self.log_config) log_filter = logging.root.handlers[-1].filters[0] # blocked default record record = RECORD('asdf', logging.DEBUG) self.assertFalse(log_filter.filter(record)) # allowed default record record = RECORD('asdf', logging.WARNING) self.assertTrue(log_filter.filter(record)) # blocked named record record = RECORD('test', logging.DEBUG) self.assertFalse(log_filter.filter(record)) # allowed named record record = RECORD('test', logging.INFO) self.assertTrue(log_filter.filter(record))