Write tests for new announcement filters

webkom · Mar 7, 2024 · e414577 · e414577
1 parent fa474ce
commit e414577
Show file tree

Hide file tree

Showing 6 changed files with 133 additions and 11 deletions.
diff --git a/lego/apps/events/fixtures/test_events.yaml b/lego/apps/events/fixtures/test_events.yaml
@@ -198,7 +198,7 @@
     created_by: 1
     require_auth: False
     responsible_group: 25
-  
+
 - model: events.Pool
   pk: 1
   fields:
@@ -287,9 +287,9 @@
   pk: 9
   fields:
     name: Abakusmember
-    capacity: 1 
+    capacity: 1
     event: 13
-    counter: 1 
+    counter: 1
     activation_date: "2019-09-01T13:20:30+03:00"
     permission_groups:
       - - Abakus
@@ -361,3 +361,12 @@
     pool: 9
     registration_date: "2011-09-01T13:20:30+03:00"
     status: "SUCCESS_REGISTER"
+
+- model: events.Registration
+  pk: 8
+  fields:
+    user: 10
+    event: 1
+    pool: null
+    registration_date: "2011-09-01T13:20:30+03:00"
+    status: "SUCCESS_REGISTER"
diff --git a/lego/apps/events/tests/test_registrations.py b/lego/apps/events/tests/test_registrations.py
@@ -413,13 +413,13 @@ def test_register_to_waiting_list_after_unregister(self):
 
         registration = Registration.objects.get_or_create(event=event, user=user)[0]
         event.register(registration)
-        self.assertEqual(event.waiting_registrations.count(), 1)
+        self.assertEqual(event.waiting_registrations.count(), 2)
 
         event.unregister(registration)
-        self.assertEqual(event.waiting_registrations.count(), 0)
+        self.assertEqual(event.waiting_registrations.count(), 1)
 
         event.register(registration)
-        self.assertEqual(event.waiting_registrations.count(), 1)
+        self.assertEqual(event.waiting_registrations.count(), 2)
 
     def test_unregistering_non_existing_user(self):
         """Test that non existing user trying to unregister raises error"""

diff --git a/lego/apps/meetings/fixtures/test_meetings.yaml b/lego/apps/meetings/fixtures/test_meetings.yaml
@@ -27,3 +27,26 @@
     end_time: "2011-02-01T23:59:00+00:00"
     report: "tomp"
     report_author: 3
+
+- model: meetings.Meeting
+  pk: 4
+  fields:
+    title: Enda et tomt møte
+    location: "null"
+    start_time: "2024-02-01T17:15:00+00:00"
+    end_time: "2024-02-01T23:59:00+00:00"
+    report: "hei"
+    report_author: 3
+
+- model: meetings.MeetingInvitation
+  pk: 1
+  fields:
+    user: 8
+    meeting: 4
+    status: "ATTENDING"
+
+- model: meetings.MeetingInvitation
+  pk: 2
+  fields:
+    user: 9
+    meeting: 4
diff --git a/lego/apps/notifications/fixtures/test_announcements.yaml b/lego/apps/notifications/fixtures/test_announcements.yaml
@@ -2,7 +2,7 @@
   pk: 1
   fields:
     message: abakom
-    sent: '2019-02-10T14:38:49.164657Z'
+    sent: "2019-02-10T14:38:49.164657Z"
     groups: [3]
     created_by: 4
 
@@ -11,15 +11,15 @@
   fields:
     message: event_from_arrkom
     from_group: 4
-    sent: '2019-02-10T14:38:49.164657Z'
+    sent: "2019-02-10T14:38:49.164657Z"
     events: [1]
     created_by: 4
 
 - model: notifications.Announcement
   pk: 3
   fields:
     message: event_abakom
-    sent: '2019-02-10T14:38:49.164657Z'
+    sent: "2019-02-10T14:38:49.164657Z"
     groups: [3]
     events: [1]
     created_by: 4
@@ -29,7 +29,7 @@
   fields:
     message: user_from_webkom
     from_group: 12
-    users: [1, 2,3]
+    users: [1, 2, 3]
     created_by: 4
 
 - model: notifications.Announcement
@@ -46,3 +46,34 @@
     message: test_from_user_webkom
     groups: [1]
     created_by: 3
+
+- model: notifications.Announcement
+  pk: 7
+  fields:
+    message: another_event_from_arrkom
+    events: [1]
+    exclude_waiting_list: false
+    created_by: 9
+
+- model: notifications.Announcement
+  pk: 8
+  fields:
+    message: yet_another_event_from_arrkom
+    events: [1]
+    exclude_waiting_list: true
+    created_by: 9
+
+- model: notifications.Announcement
+  pk: 9
+  fields:
+    message: meeting
+    meetings: [4]
+    created_by: 1
+
+- model: notifications.Announcement
+  pk: 10
+  fields:
+    message: meeting
+    meetings: [4]
+    meeting_invitation_status: "ATTENDING"
+    created_by: 1
diff --git a/lego/apps/notifications/tests/test_views.py b/lego/apps/notifications/tests/test_views.py
@@ -81,6 +81,7 @@ class AnnouncementViewSetTestCase(BaseAPITestCase):
         "test_events.yaml",
         "test_companies.yaml",
         "test_announcements.yaml",
+        "test_meetings.yaml",
     ]
 
     def setUp(self):
@@ -238,3 +239,59 @@ def test_send_announcement_unauthorized(self):
         self.client.force_authenticate(self.unauthorized_user)
         response = self.client.post(f"{self.url}{self.unsent_announcement.id}/send/")
         self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
+
+    def test_send_announcement_twice(self):
+        """
+        An announcement can not be sent twice
+        """
+
+        self.client.force_authenticate(self.authorized_user)
+        response = self.client.post(f"{self.url}{self.unsent_announcement.id}/send/")
+        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
+        response = self.client.post(f"{self.url}{self.unsent_announcement.id}/send/")
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+
+    def test_recipients_lookup_to_users(self):
+        """
+        An announcement should be sent to the specified users
+        """
+
+        announcement = Announcement.objects.get(pk=4)
+        recipients = announcement.lookup_recipients()
+        self.assertEqual(len(recipients), 3)
+
+    def test_recipients_lookup_for_event(self):
+        """
+        Everyone registered to an event should receive the announcement
+        """
+        announcement = Announcement.objects.get(pk=7)
+        recipients = announcement.lookup_recipients()
+        self.assertEqual(len(recipients), 3)
+
+    def test_recipients_lookup_for_event_exclude_waiting_list(self):
+        """
+        Everyone registered to an event, except those on waiting list,
+        should receive the announcement
+        """
+
+        announcement = Announcement.objects.get(pk=8)
+        recipients = announcement.lookup_recipients()
+        self.assertEqual(len(recipients), 2)
+
+    def test_recipients_lookup_for_meeting(self):
+        """
+        Everyone invited to a meeting should receive the announcement
+        """
+
+        announcement = Announcement.objects.get(pk=9)
+        recipients = announcement.lookup_recipients()
+        self.assertEqual(len(recipients), 2)
+
+    def test_recipients_lookup_for_only_meeting_attendees(self):
+        """
+        Only those invited to a meeting and accepted should receive the announcement
+        """
+
+        announcement = Announcement.objects.get(pk=10)
+        recipients = announcement.lookup_recipients()
+        self.assertEqual(len(recipients), 1)
diff --git a/lego/apps/restricted/tests/test_models.py b/lego/apps/restricted/tests/test_models.py
@@ -22,7 +22,9 @@ def test_lookup_recipients(self):
         restricted_mail = RestrictedMail.objects.get(id=1)
         recipients = restricted_mail.lookup_recipients()
 
-        self.assertCountEqual(recipients, ["[email protected]", "[email protected]"])
+        self.assertCountEqual(
+            recipients, ["[email protected]", "[email protected]", "[email protected]"]
+        )
 
     def test_mark_used(self):
         """The used field is not None when the item is marked as used"""