def test_add_notification(self):
data = {'data': 'some data here'}
for i, uid in enumerate(self.uids[:10], start=1):
Notification(uid, self.receivers, data).send()
assert len(Notification.get_data(self.test_user1)) == i
assert Notification.unread_count(self.test_user1) == i
assert len(Notification.get_data(self.test_user2)) == i
assert Notification.unread_count(self.test_user2) == i
def test_mark_as_read(self):
data = {'data': 'some data here'}
for uid in self.uids:
Notification(uid, self.receivers, data).send()
to_mark_as_read = self.uids[:5] + self.uids[20:30] + self.uids[40:50]
for i, uid in enumerate(to_mark_as_read, start=1):
Notification.mark_as_read(self.test_user1, uid)
assert Notification.unread_count(self.test_user1) == 50-i
assert Notification.unread_count(self.test_user2) == 50
Notification.mark_all_as_read(self.test_user2)
assert Notification.unread_count(self.test_user2) == 0
def unread_notification_count(self):
from vilya.models.notification import Notification
return Notification.unread_count(self.name)
def unread_notification_count(self):
from vilya.models.notification import Notification
return Notification.unread_count(self.name)