class TestNotifierProxy(TestCase): def setUp(self): self.proxy = NotifierProxy() def test_adding_notifier(self): args = range(4) notifier = MagicMock(PagerdutyNotifier) self.proxy.add_notifier(notifier) self.proxy.notify(*args) notifier.notify.assert_called_once_with(*args) def test_adding_multiple_notifiers(self): args = range(4) notifier_1 = MagicMock(PagerdutyNotifier) notifier_2 = MagicMock(PagerdutyNotifier) self.proxy.add_notifier(notifier_1) self.proxy.add_notifier(notifier_2) self.proxy.notify(*args) notifier_1.notify.assert_called_once_with(*args) notifier_2.notify.assert_called_once_with(*args)
def setUp(self): self.proxy = NotifierProxy()