Refactor get raffle constraints depending on reversed constraints

UnitapApp · Nov 30, 2023 · 40649ac · 40649ac
1 parent ccc3397
commit 40649ac
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/prizetap/tests.py b/prizetap/tests.py
@@ -435,6 +435,27 @@ def test_get_raffle_constraints_when_is_verified(self):
         self.assertEqual(data["pk"], self.meet_constraint.pk)
         self.assertEqual(data["is_verified"], True)
 
+    @patch(
+        "authentication.helpers.BrightIDSoulboundAPIInterface.get_verification_status",
+        lambda a, b, c: (True, None),
+    )
+    def test_get_raffle_constraints_when_constraint_is_reversed(self):
+        self.client.force_authenticate(user=self.user_profile.user)
+        self.raffle.reversed_constraints = str(self.meet_constraint.pk)
+        self.raffle.save()
+        response = self.client.get(reverse("get-raffle-constraints", kwargs={"raffle_pk": self.raffle.pk}))
+        data = response.data["constraints"][0]
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(data["pk"], self.meet_constraint.pk)
+        self.assertEqual(data["is_verified"], False)
+
+        BrightIDSoulboundAPIInterface.get_verification_status = MagicMock(return_value=(False, None))
+        response = self.client.get(reverse("get-raffle-constraints", kwargs={"raffle_pk": self.raffle.pk}))
+        data = response.data["constraints"][0]
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(data["pk"], self.meet_constraint.pk)
+        self.assertEqual(data["is_verified"], True)
+
 
 class RaffleEntryTestCase(RaffleTestCase):
     def setUp(self):

diff --git a/prizetap/views.py b/prizetap/views.py
@@ -145,6 +145,7 @@ def get(self, request, raffle_pk):
         except Exception:
             param_values = {}
 
+        reversed_constraints = raffle.reversed_constraints.split(",") if raffle.reversed_constraints else []
         response_constraints = []
 
         for c in raffle.constraints.all():
@@ -155,8 +156,12 @@ def get(self, request, raffle_pk):
             except KeyError:
                 pass
             is_verified = False
-            if constraint.is_observed():
-                is_verified = True
+            if str(c.pk) in reversed_constraints:
+                if not constraint.is_observed():
+                    is_verified = True
+            else:
+                if constraint.is_observed():
+                    is_verified = True
             response_constraints.append({**ConstraintSerializer(c).data, "is_verified": is_verified})
 
         return Response({"success": True, "constraints": response_constraints}, status=200)