def test_for_user(self): """Notifications should be filterable by a single user.""" user_notification = factories.NotificationFactory(user=self.user) user_notifications = Notification.objects.for_user(self.user) ok_(user_notification in user_notifications, 'A notification for the user should be in the set returned.') ok_(self.notification not in user_notifications, 'A notification for another user should not be in the set returned.')
def test_for_object(self): """Notifications should be filterable by a single object.""" foia = factories.FOIARequestFactory() _action = new_action(factories.UserFactory(), 'submitted', target=foia) object_notification = factories.NotificationFactory(user=self.user, action=_action) object_notifications = Notification.objects.for_object(foia) ok_(object_notification in object_notifications, 'A notification for the object should be in the set returned.') ok_(self.notification not in object_notifications, 'A notification not including the object should not be in the set returned.')
def setUp(self): self.user = factories.UserFactory() self.action = new_action(self.user, 'acted') self.notification = factories.NotificationFactory()
def test_when_unread(self, mock_send): """The send method should be called when a user has unread notifications.""" factories.NotificationFactory(user=self.user) tasks.daily_digest() mock_send.assert_called_with(self.user, u'Daily Digest', relativedelta(days=1))