Show More
@@ -61,14 +61,16 class TestNotifications(unittest.TestCas | |||
|
61 | 61 | notifications = Notification.query().all() |
|
62 | 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 | 68 | unotification = UserNotification.query()\ |
|
65 | 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 | 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 | 75 | def test_user_notifications(self): |
|
74 | 76 | self.assertEqual([], Notification.query().all()) |
General Comments 0
You need to be logged in to leave comments.
Login now