class TestPendingNotifications(unittest.TestCase): """Test PendingNotifications.""" def setUp(self): super(TestPendingNotifications, self).setUp() self.notifier = FakeNotifier() self.pending_notifications = PendingNotifications(self.notifier) def test_send_with_no_events(self): """If no events are pending, no notifications are sent.""" self.pending_notifications.send() self.assertEqual(0, len(self.notifier.notifications)) def test_send_does_send_single_event(self): """If a single event is pending, it should still be sent.""" share = FakeShare() events = [ShareCreated(*share.event_args)] self.pending_notifications.add_events(events) self.pending_notifications.send() self.assertEqual(1, len(self.notifier.notifications)) self.assertEqual(events, self.notifier.notifications)
def setUp(self): super(TestPendingNotifications, self).setUp() self.notifier = FakeNotifier() self.pending_notifications = PendingNotifications(self.notifier)