def setUp(self):
self.cdh = CategoryDispatchHandler()
class CategoryDispatchHandlerTests(TestCase):
def setUp(self):
self.cdh = CategoryDispatchHandler()
def test_log(self):
handler = mock.Mock()
self.cdh.addHandler('category', handler)
self.cdh.log('category', 'message')
handler.log.assert_called_with('category', 'message')
def test_log_multi_categories(self):
handler1 = mock.Mock()
handler2 = mock.Mock()
self.cdh.addHandler('category1', handler1)
self.cdh.addHandler('category2', handler2)
self.cdh.log('category1', 'message')
handler1.log.assert_called_with('category1', 'message')
handler2.log.assert_not_called()
self.cdh.log('category2', 'message')
handler1.log.assert_not_called()
handler2.log.assert_called_with('category2', 'message')
def test_log_multi_handlers(self):
handler1 = mock.Mock()
handler2 = mock.Mock()
self.cdh.addHandler('category', handler1)
self.cdh.addHandler('category', handler2)
self.cdh.log('category', 'message')
handler1.log.assert_called_with('category', 'message')
handler2.log.assert_called_with('category', 'message')
