Show More
@@ -61,14 +61,16 class TestNotifications(unittest.TestCas | |||||
61 | notifications = Notification.query().all() |
|
61 | notifications = Notification.query().all() | |
62 | self.assertEqual(len(notifications), 1) |
|
62 | self.assertEqual(len(notifications), 1) | |
63 |
|
63 | |||
|
64 | self.assertEqual(notifications[0].recipients, [u1, u2]) | |||
|
65 | self.assertEqual(notification.notification_id, | |||
|
66 | notifications[0].notification_id) | |||
|
67 | ||||
64 | unotification = UserNotification.query()\ |
|
68 | unotification = UserNotification.query()\ | |
65 | .filter(UserNotification.notification == notification).all() |
|
69 | .filter(UserNotification.notification == notification).all() | |
66 |
|
70 | |||
67 | self.assertEqual(notifications[0].recipients, [u1, u2]) |
|
|||
68 | self.assertEqual(notification.notification_id, |
|
|||
69 | notifications[0].notification_id) |
|
|||
70 | self.assertEqual(len(unotification), len(usrs)) |
|
71 | self.assertEqual(len(unotification), len(usrs)) | |
71 |
self.assertEqual([x.user.user_id for x in unotification], |
|
72 | self.assertEqual(set([x.user.user_id for x in unotification]), | |
|
73 | set(usrs)) | |||
72 |
|
74 | |||
73 | def test_user_notifications(self): |
|
75 | def test_user_notifications(self): | |
74 | self.assertEqual([], Notification.query().all()) |
|
76 | self.assertEqual([], Notification.query().all()) |
General Comments 0
You need to be logged in to leave comments.
Login now