def test_handlers_called():
'''Test that registered handlers are called.'''
mock_handler_a = Mock()
mock_handler_b = Mock()
distribute = Distribute(handlers={
'a': mock_handler_a,
'b': mock_handler_b
})
log = Log()
distribute.handle(log)
mock_handler_a.handle.assert_called_once_with(log)
mock_handler_b.handle.assert_called_once_with(log)
def test_handlers_not_called_due_to_filter():
'''Test that distribute filter prevents handlers being called.'''
mock_handler_a = Mock()
deny_all = Mock()
deny_all.filter = Mock(return_value=[])
distribute = Distribute(
filterer=deny_all,
handlers={
'a': mock_handler_a,
}
)
log = Log()
distribute.handle(log)
mock_handler_a.handle.assert_called_once_with()
assert mock_handler_a.output.called is False
def test_no_handlers():
'''Test calling when no handlers registered.'''
distribute = Distribute()
log = Log()
distribute.handle(log)
