Update tests for BDS

bds/tests/test_views.py

@@ -1,3 +1,4 @@
+from datetime import date
 from unittest import mock
 
 from django.conf import settings
@@ -6,6 +7,8 @@ from django.contrib.auth.models import Permission
 from django.test import Client, TestCase
 from django.urls import reverse, reverse_lazy
 
+from bds.models import BDSProfile
+
 User = get_user_model()
 
 
@@ -24,7 +27,7 @@ def login_url(next=None):
 
 class TestHomeView(TestCase):
     @mock.patch("gestioncof.signals.messages")
-    def test_get(self, mock_messages):
+    def test_get(self, _):
         user = User.objects.create_user(username="random_user")
         give_bds_buro_permissions(user)
         self.client.force_login(
@@ -36,7 +39,7 @@ class TestHomeView(TestCase):
 
 class TestRegistrationView(TestCase):
     @mock.patch("gestioncof.signals.messages")
-    def test_get_autocomplete(self, mock_messages):
+    def test_get_autocomplete(self, _):
         user = User.objects.create_user(username="toto")
         url = reverse("bds:autocomplete") + "?q=foo"
         client = Client()
@@ -56,7 +59,7 @@ class TestRegistrationView(TestCase):
         self.assertEqual(resp.status_code, 200)
 
     @mock.patch("gestioncof.signals.messages")
-    def test_get(self, mock_messages):
+    def test_get(self, _):
         user = User.objects.create_user(username="toto")
         url = reverse("bds:user.update", args=(user.id,))
         client = Client()
@@ -74,3 +77,174 @@ class TestRegistrationView(TestCase):
         give_bds_buro_permissions(user)
         resp = client.get(url)
         self.assertEqual(resp.status_code, 200)
+
+
+class TestResetMemberships(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.admin = User.objects.create_user(username="bds_admin")
+
+        # Create users and profiles
+        cls.u_1 = User.objects.create_user(username="bds-01")
+        cls.p_1 = BDSProfile.objects.create(user=cls.u_1)
+
+        cls.u_2 = User.objects.create_user(username="bds-02")
+        cls.p_2 = BDSProfile.objects.create(user=cls.u_2)
+
+        cls.u_3 = User.objects.create_user(username="bds-03")
+        cls.p_3 = BDSProfile.objects.create(user=cls.u_3)
+
+        cls.u_4 = User.objects.create_user(username="bds-04")
+        cls.p_4 = BDSProfile.objects.create(user=cls.u_4)
+
+    @mock.patch("gestioncof.signals.messages")
+    def setUp(self, _):
+        give_bds_buro_permissions(self.admin)
+        # bds-01 est membre à l'année
+        self.p_1.is_member = True
+        self.p_1.mails_bds = True
+        self.p_1.cotisation_period = "ANN"
+        self.p_1.save()
+
+        # bds-02 est membre au S1
+        self.p_2.is_member = True
+        self.p_2.mails_bds = True
+        self.p_2.cotisation_period = "SE1"
+        self.p_2.save()
+
+        # bds-03 est membre au S2
+        self.p_3.is_member = True
+        self.p_3.mails_bds = True
+        self.p_3.cotisation_period = "SE2"
+        self.p_3.save()
+
+    @mock.patch("gestioncof.signals.messages")
+    def test_get_expired(self, _):
+        user = User.objects.create_user(username="toto")
+        url = reverse("bds:members.expired")
+        client = Client()
+
+        # Anonymous GET
+        resp = client.get(url)
+        self.assertRedirects(resp, login_url(next=url))
+
+        # Logged-in but unprivileged GET
+        client.force_login(user, backend="django.contrib.auth.backends.ModelBackend")
+        resp = client.get(url)
+        self.assertEquals(resp.status_code, 403)
+
+        # Burô user GET
+        give_bds_buro_permissions(user)
+        resp = client.get(url)
+        self.assertEquals(resp.status_code, 200)
+
+        self.assertQuerysetEqual(
+            resp.context_data["object_list"], BDSProfile.expired_members()
+        )
+
+    @mock.patch("gestioncof.signals.messages")
+    def test_get_reset(self, _):
+        user = User.objects.create_user(username="tata")
+        url = reverse("bds:members.reset")
+        client = Client()
+
+        # Anonymous GET
+        resp = client.get(url)
+        self.assertRedirects(resp, login_url(next=url))
+
+        # Logged-in but unprivileged GET
+        client.force_login(user, backend="django.contrib.auth.backends.ModelBackend")
+        resp = client.get(url)
+        self.assertEquals(resp.status_code, 403)
+
+        # Burô user GET
+        give_bds_buro_permissions(user)
+        resp = client.get(url)
+        self.assertRedirects(resp, reverse("bds:members.expired"))
+
+    def test_expired_memberships(self):
+        # In september, there is no expired members
+        with mock.patch("bds.models.timezone.now", return_value=date(2022, 9, 1)):
+            self.assertQuerysetEqual(BDSProfile.expired_members(), [])
+
+        # In march, only bds-02's membership is expired
+        with mock.patch("django.utils.timezone.now", return_value=date(2023, 3, 1)):
+            self.assertQuerysetEqual(BDSProfile.expired_members(), [self.p_2])
+
+        # During summer, all memberships are expired
+        with mock.patch("django.utils.timezone.now", return_value=date(2023, 7, 1)):
+            self.assertQuerysetEqual(
+                BDSProfile.expired_members(),
+                [self.p_1, self.p_2, self.p_3],
+                ordered=False,
+            )
+
+    @mock.patch("gestioncof.signals.messages")
+    def test_reset_memberships(self, _):
+        url = reverse("bds:members.reset")
+
+        # Reset during S1 does nothig
+        with mock.patch("bds.models.timezone.now", return_value=date(2022, 9, 1)):
+            self.client.force_login(
+                self.admin, backend="django.contrib.auth.backends.ModelBackend"
+            )
+            resp = self.client.get(url)
+
+            self.assertRedirects(resp, reverse("bds:members.expired"))
+
+            self.assertQuerysetEqual(BDSProfile.expired_members(), [])
+
+            self.assertQuerysetEqual(
+                BDSProfile.objects.filter(is_member=True),
+                [self.p_1, self.p_2, self.p_3],
+                ordered=False,
+            )
+
+        # Reset in march
+        with mock.patch("django.utils.timezone.now", return_value=date(2023, 3, 1)):
+            self.client.force_login(
+                self.admin, backend="django.contrib.auth.backends.ModelBackend"
+            )
+            resp = self.client.get(url)
+
+            self.assertRedirects(resp, reverse("bds:members.expired"))
+
+            # After a reset we have no expired memberships
+            self.assertQuerysetEqual(BDSProfile.expired_members(), [])
+
+            # Test reset attributes
+            self.p_2.refresh_from_db()
+            self.assertEqual(
+                [self.p_2.is_member, self.p_2.mails_bds, self.p_2.cotisation_period],
+                [False, False, "NO"],
+            )
+
+        # Reset during summer
+        with mock.patch("django.utils.timezone.now", return_value=date(2023, 7, 1)):
+            # bds-02's membership wa already reset
+            self.assertQuerysetEqual(
+                BDSProfile.expired_members(), [self.p_1, self.p_3], ordered=False
+            )
+
+            self.client.force_login(
+                self.admin, backend="django.contrib.auth.backends.ModelBackend"
+            )
+            resp = self.client.get(url)
+
+            self.assertRedirects(resp, reverse("bds:members.expired"))
+
+            # After a reset we have no expired memberships
+            self.assertQuerysetEqual(BDSProfile.expired_members(), [])
+
+            # Test reset attributes
+            self.p_1.refresh_from_db()
+            self.assertEqual(
+                [self.p_1.is_member, self.p_1.mails_bds, self.p_1.cotisation_period],
+                [False, False, "NO"],
+            )
+
+            self.p_3.refresh_from_db()
+            self.assertEqual(
+                [self.p_3.is_member, self.p_3.mails_bds, self.p_3.cotisation_period],
+                [False, False, "NO"],
+            )