def test_warning_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.WARNING)
# When a warning record is emitted
handler.emit(record=record)
# Then info warning is called
collectd.warning.assert_called_once_with("message")
def test_fatal_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.FATAL)
# When a fatal record is emitted
handler.emit(record=record)
# Then error hook is called
collectd.error.assert_called_once_with("message")
def test_info_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.INFO)
# When an info record is emitted
handler.emit(record=record)
# Then info hook is called
collectd.info.assert_called_once_with("message")
def test_fatal_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.FATAL)
# When a fatal record is emitted
handler.emit(record=record)
# Then error hook is called
collectd.error.assert_called_once_with("message")
def test_warning_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.WARNING)
# When a warning record is emitted
handler.emit(record=record)
# Then info warning is called
collectd.warning.assert_called_once_with("message")
def test_info_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.INFO)
# When an info record is emitted
handler.emit(record=record)
# Then info hook is called
collectd.info.assert_called_once_with("message")
def test_verbose_debug_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
handler.verbose = True
record = make_record(msg="message", level=logging.DEBUG)
# When an info record is emitted
handler.emit(record=record)
# Then info hook is called
collectd.info.assert_called_once_with("message")
def test_debug_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.DEBUG)
# When a debug record is emitted
handler.emit(record=record)
# Then debug hook is called
collectd.debug.assert_called_once_with("message")
def test_verbose_debug_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
handler.verbose = True
record = make_record(msg="message", level=logging.DEBUG)
# When an info record is emitted
handler.emit(record=record)
# Then info hook is called
collectd.info.assert_called_once_with("message")
def test_debug_when_emit(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
record = make_record(msg="message", level=logging.DEBUG)
# When a debug record is emitted
handler.emit(record=record)
# Then debug hook is called
collectd.debug.assert_called_once_with("message")
def test_non_verbose_when_logger_info(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
handler.verbose = False
logger = logging.Logger('some_logger')
logger.addHandler(handler)
# When debug is called
logger.debug('Say cheese: %s %d', 'string', 10)
# Then debug hook is called
collectd.debug.assert_called_once_with('Say cheese: string 10')
collectd.info.assert_not_called()
def test_non_verbose_when_logger_info(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
handler.verbose = False
logger = logging.Logger('some_logger')
logger.addHandler(handler)
# When debug is called
logger.debug('Say cheese: %s %d', 'string', 10)
# Then debug hook is called
collectd.debug.assert_called_once_with('Say cheese: string 10')
collectd.info.assert_not_called()
def test_long_message_when_emit(self, collectd):
# Given
long_message = "LONG " * 20 + "MESSAGE."
handler = CollectdLogHandler(collectd=collectd)
handler.max_message_length = 10
record = make_record(msg=long_message)
# When a long message is emitted
handler.emit(record=record)
# Then info hook is called n times with split message
collectd.info.assert_has_calls([
mock.call(long_message[i:i + 10])
for i in range(0, len(long_message), 10)])
def test_long_message_when_emit(self, collectd):
# Given
long_message = "LONG " * 20 + "MESSAGE."
handler = CollectdLogHandler(collectd=collectd)
handler.max_message_length = 10
record = make_record(msg=long_message)
# When a long message is emitted
handler.emit(record=record)
# Then info hook is called n times with split message
collectd.info.assert_has_calls([
mock.call(long_message[i:i + 10])
for i in range(0, len(long_message), 10)
])
def register_plugin(collectd):
"""Bind plugin hooks to collectd and viceversa."""
config = Config.instance()
# Setup loggging
log_handler = CollectdLogHandler(collectd=collectd)
log_handler.cfg = config
ROOT_LOGGER.addHandler(log_handler)
ROOT_LOGGER.setLevel(logging.NOTSET)
# Creates collectd plugin instance
instance = Plugin(collectd=collectd, config=config)
# Register plugin callbacks
collectd.register_config(instance.config)
collectd.register_shutdown(instance.shutdown)
collectd.register_notification(instance.notify)
def register_plugin(collectd):
"Bind plugin hooks to collectd and viceversa"
config = Config.instance()
# Setup loggging
log_handler = CollectdLogHandler(collectd=collectd)
log_handler.cfg = config
ROOT_LOGGER.addHandler(log_handler)
ROOT_LOGGER.setLevel(logging.NOTSET)
# Creates collectd plugin instance
instance = Plugin(collectd=collectd, config=config)
# Register plugin callbacks
collectd.register_config(instance.config)
collectd.register_write(instance.write)
collectd.register_shutdown(instance.shutdown)
def test_warning_from_logger(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
logger = logging.Logger('some_logger')
logger.addHandler(handler)
# When warning is called
logger.warning('Say cheese: %s %d', 'string', 10)
# Then warning hook is called
collectd.warning.assert_called_once_with('Say cheese: string 10')
def test_fatal_from_logger(self, collectd):
# Given
handler = CollectdLogHandler(collectd=collectd)
logger = logging.Logger('some_logger')
logger.addHandler(handler)
# When fatal is called
logger.fatal('Say cheese: %s %d', 'string', 10)
# Then error hook is called
collectd.error.assert_called_once_with('Say cheese: string 10')
def test_registered_hooks_when_init(self, collectd):
# When CollectdLogHandler is created
handler = CollectdLogHandler(collectd=collectd)
# Then collectd logging hooks are registered
# pylint: disable=protected-access
self.assertEqual(
{
logging.DEBUG: collectd.debug,
logging.INFO: collectd.info,
logging.WARNING: collectd.warning,
logging.ERROR: collectd.error,
logging.FATAL: collectd.error
}, handler.priority_map)
