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_get_data(self):
for i, uid in enumerate(self.uids[:10], start=1):
data = {'data': 'test' + uid}
Notification(uid, self.receivers, data).send()
data = Notification.get_data(self.test_user1)
assert len(data) == 10
for i, d in enumerate(reversed(data)):
assert d.get('uid') == self.uids[i]
assert d.get('data') == 'test'+self.uids[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 noti(self):
return Notification(self._uid, self.noti_receivers, self.noti_data)
