Add eligibility methods to models

Can be used to enforce other questions from being answered if users are
not eligible

Author: Themitchell

lung_cancer_screening/questions/models/check_need_appointment_response.py

@@ -7,3 +7,6 @@
 class CheckNeedAppointmentResponse(BaseModel):
     response_set = models.OneToOneField(ResponseSet, on_delete=models.CASCADE, related_name='check_need_appointment_response')
     value = models.BooleanField()
+
+    def is_eligible(self):
+        return not self.value

lung_cancer_screening/questions/models/date_of_birth_response.py

@@ -10,7 +10,7 @@ class DateOfBirthResponse(BaseModel):
     response_set = models.OneToOneField(ResponseSet, on_delete=models.CASCADE, related_name='date_of_birth_response')
     value = models.DateField()
 
-    def is_currently_in_age_range(self):
+    def is_eligible(self):
         fifty_five_years_ago = date.today() - relativedelta(years=55)
         seventy_five_years_ago = date.today() - relativedelta(years=75)
 

lung_cancer_screening/questions/models/have_you_ever_smoked_response.py

@@ -12,6 +12,15 @@ class HaveYouEverSmokedValues(models.IntegerChoices):
 
 
 class HaveYouEverSmokedResponse(BaseModel):
+    ELIGIBLE_VALUES = [
+        HaveYouEverSmokedValues.YES_I_USED_TO_SMOKE_REGULARLY.value,
+        HaveYouEverSmokedValues.YES_I_CURRENTLY_SMOKE.value
+    ]
+    INELIGIBLE_VALUES = [
+        HaveYouEverSmokedValues.YES_BUT_ONLY_A_FEW_TIMES.value,
+        HaveYouEverSmokedValues.NO_I_HAVE_NEVER_SMOKED.value
+    ]
+
     response_set = models.OneToOneField(ResponseSet, on_delete=models.CASCADE, related_name='have_you_ever_smoked_response')
     value = models.IntegerField(choices=HaveYouEverSmokedValues.choices)
 
@@ -22,7 +31,4 @@ def has_smoked_regularly(self):
         ]
 
     def is_eligible(self):
-        return self.value in [
-            HaveYouEverSmokedValues.YES_I_USED_TO_SMOKE_REGULARLY.value,
-            HaveYouEverSmokedValues.YES_I_CURRENTLY_SMOKE.value
-        ]
+        return self.value in self.ELIGIBLE_VALUES

lung_cancer_screening/questions/models/response_set.py

@@ -88,3 +88,20 @@ def _response_attrs(self):
 
     def is_complete(self):
         return all(hasattr(self, attr) for attr in self._response_attrs())
+
+
+    def is_eligible(self):
+        if not all(
+            hasattr(self, attr) for attr in [
+                'have_you_ever_smoked_response',
+                'date_of_birth_response',
+                'check_need_appointment_response'
+            ]
+        ):
+            return False
+
+        return all([
+            self.have_you_ever_smoked_response.is_eligible(),
+            self.date_of_birth_response.is_eligible(),
+            self.check_need_appointment_response.is_eligible()
+        ])

lung_cancer_screening/questions/tests/factories/check_need_appointment_response_factory.py

@@ -10,3 +10,12 @@ class Meta:
 
     response_set = factory.SubFactory(ResponseSetFactory)
     value = factory.Faker('boolean')
+
+    class Params:
+        eligible = factory.Trait(
+            value=False
+        )
+
+        ineligible = factory.Trait(
+            value=True
+        )

lung_cancer_screening/questions/tests/factories/date_of_birth_response_factory.py

@@ -10,3 +10,12 @@ class Meta:
 
     response_set = factory.SubFactory(ResponseSetFactory)
     value = factory.Faker('date_of_birth')
+
+    class Params:
+        eligible = factory.Trait(
+            value=factory.Faker('date_of_birth', minimum_age=55, maximum_age=74)
+        )
+
+        ineligible = factory.Trait(
+            value=factory.Faker('date_of_birth', maximum_age=54)
+        )

lung_cancer_screening/questions/tests/factories/have_you_ever_smoked_response_factory.py

@@ -10,3 +10,12 @@ class Meta:
 
     response_set = factory.SubFactory(ResponseSetFactory)
     value = factory.Iterator(HaveYouEverSmokedValues)
+
+    class Params:
+        eligible = factory.Trait(
+            value=factory.Iterator(HaveYouEverSmokedResponse.ELIGIBLE_VALUES)
+        )
+
+        ineligible = factory.Trait(
+            value=factory.Iterator(HaveYouEverSmokedResponse.INELIGIBLE_VALUES)
+        )

lung_cancer_screening/questions/tests/factories/response_set_factory.py

@@ -25,6 +25,24 @@ class Meta:
     user = factory.SubFactory(UserFactory)
 
     class Params:
+        eligible = factory.Trait(
+            have_you_ever_smoked_response=factory.RelatedFactory(
+                "lung_cancer_screening.questions.tests.factories.have_you_ever_smoked_response_factory.HaveYouEverSmokedResponseFactory",
+                factory_related_name="response_set",
+                eligible=True
+            ),
+            date_of_birth_response=factory.RelatedFactory(
+                "lung_cancer_screening.questions.tests.factories.date_of_birth_response_factory.DateOfBirthResponseFactory",
+                factory_related_name="response_set",
+                eligible=True
+            ),
+            check_need_appointment_response=factory.RelatedFactory(
+                "lung_cancer_screening.questions.tests.factories.check_need_appointment_response_factory.CheckNeedAppointmentResponseFactory",
+                factory_related_name="response_set",
+                eligible=True
+            ),
+        )
+
         complete = factory.Trait(
             asbestos_exposure_response=factory.RelatedFactory(
                 "lung_cancer_screening.questions.tests.factories.asbestos_exposure_response_factory.AsbestosExposureResponseFactory",

lung_cancer_screening/questions/tests/unit/models/test_check_need_appointment_response.py

@@ -32,3 +32,21 @@ def test_has_value_as_bool(self):
         )
 
         self.assertIsInstance(response.value, bool)
+
+
+    def test_is_eligible_returns_true_when_value_is_false(self):
+        response = CheckNeedAppointmentResponse.objects.create(
+            response_set=self.response_set,
+            value=False
+        )
+
+        self.assertTrue(response.is_eligible())
+
+
+    def test_is_eligible_returns_false_when_value_is_true(self):
+        response = CheckNeedAppointmentResponse.objects.create(
+            response_set=self.response_set,
+            value=True
+        )
+
+        self.assertFalse(response.is_eligible())

lung_cancer_screening/questions/tests/unit/models/test_date_of_birth_response.py

@@ -34,18 +34,18 @@ def test_has_value_as_date(self):
 
         self.assertIsInstance(response.value, date)
 
-    def test_is_currently_in_age_range_returns_true_when_in_range(self):
+    def test_is_eligible_returns_true_when_in_range(self):
         fifty_five_years_ago = date.today() - relativedelta(years=55)
         response = DateOfBirthResponseFactory.build(
             value=fifty_five_years_ago
         )
 
-        self.assertTrue(response.is_currently_in_age_range())
+        self.assertTrue(response.is_eligible())
 
-    def test_is_currently_in_age_range_returns_false_when_too_old(self):
+    def test_is_eligible_returns_false_when_too_old(self):
         fifty_five_years_ago = date.today() - relativedelta(years=55)
         response = DateOfBirthResponseFactory.build(
             value=fifty_five_years_ago
         )
 
-        self.assertTrue(response.is_currently_in_age_range())
+        self.assertTrue(response.is_eligible())