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')