PPHA-525: Ensure user has to change periods stopped smoking if changeing age started smoking

Author: Themitchell

features/age_when_started_smoking.feature

@@ -40,5 +40,4 @@ Feature: Age when started smoking
     Then I am on "/age-when-started-smoking?change=True"
     And I see "18" filled in for "How old were you when you started smoking?"
     When I fill in "How old were you when you started smoking?" as "22" and submit
-    Then I am on "/check-your-answers"
-    And I see "22" as a response to "Age you started smoking" under "Smoking history"
+    Then I am on "/periods-when-you-stopped-smoking?change=True"

features/steps/preflight_steps.py

@@ -1,6 +1,12 @@
 from behave import given
-from datetime import datetime
-from dateutil.relativedelta import relativedelta
+
+from features.steps.form_steps import (
+    when_i_fill_in_and_submit_my_date_of_birth_as_x_years_ago,
+    when_i_submit_the_form,
+    when_i_check_label,
+    when_i_fill_in_label_with_value
+)
+
 from lung_cancer_screening.questions.tests.factories.have_you_ever_smoked_response_factory import (
     ResponseSetFactory,
 )
@@ -23,6 +29,7 @@ def get_or_create_response_set(context):
         )
     )
 
+
 @given('I have answered have you ever smoked with an eligible response')
 def given_i_have_answered_have_your_ever_smoked_with_an_eligible_response(context):
     response_set = get_or_create_response_set(context)
@@ -57,3 +64,16 @@ def given_i_have_answered_questions_showing_i_have_smoked_for_years_years(contex
         response_set=response_set,
         value=response_set.date_of_birth_response.age_in_years() - int(years),
     )
+
+@given('I have answered questions showing I am aged "{years}" years old')
+def given_i_have_answered_questions_showing_i_am_aged_60_years_old(context, years):
+    context.page.goto(f"{context.live_server_url}/date-of-birth")
+    when_i_fill_in_and_submit_my_date_of_birth_as_x_years_ago(context, years)
+
+
+@given('I have answered questions showing I stopped smoking for "{years}" years')
+def given_i_have_answered_questions_showing_i_stopped_smoking_for_years_years(context, years):
+    context.page.goto(f"{context.live_server_url}/periods-when-you-stopped-smoking")
+    when_i_check_label(context, "Yes")
+    when_i_fill_in_label_with_value(context, "Enter the total number of years you stopped smoking for", years)
+    when_i_submit_the_form(context)

lung_cancer_screening/questions/forms/age_when_started_smoking_form.py

@@ -28,3 +28,18 @@ def __init__(self, *args, **kwargs):
     class Meta:
         model = AgeWhenStartedSmokingResponse
         fields = ["value"]
+
+
+    def save(self, commit=True):
+        instance = super(AgeWhenStartedSmokingForm, self).save(commit=False)
+
+        self._cleanup_periods_stopped_smoking_if_value_changed()
+
+        if commit:
+            instance.save()
+
+        return instance
+
+    def _cleanup_periods_stopped_smoking_if_value_changed(self):
+        if 'value' in self.changed_data and hasattr(self.instance.response_set, 'periods_when_you_stopped_smoking_response'):
+            self.instance.response_set.periods_when_you_stopped_smoking_response.delete()

lung_cancer_screening/questions/tests/factories/periods_when_you_stopped_smoking_response_factory.py

@@ -10,3 +10,4 @@ class Meta:
 
     response_set = factory.SubFactory(ResponseSetFactory)
     value = False
+    duration_years = None

lung_cancer_screening/questions/tests/unit/forms/test_age_when_started_smoking_form.py

@@ -8,6 +8,8 @@
 from ...factories.response_set_factory import ResponseSetFactory
 from ...factories.date_of_birth_response_factory import DateOfBirthResponseFactory
 from ....forms.age_when_started_smoking_form import AgeWhenStartedSmokingForm
+from ...factories.periods_when_you_stopped_smoking_response_factory import PeriodsWhenYouStoppedSmokingResponseFactory
+from ....models.periods_when_you_stopped_smoking_response import PeriodsWhenYouStoppedSmokingResponse
 
 
 @tag("AgeWhenStartedSmoking")
@@ -106,3 +108,43 @@ def test_is_invalid_when_age_entered_greater_than_current_age(self):
             form.errors["value"],
             ["The age you started smoking must be the same as, or less than your current age"]
         )
+
+
+    def test_deletes_periods_when_stopped_smoking_response_if_age_started_smoking_is_changed(self):
+        self.response.value = 17
+        self.response.save()
+
+        PeriodsWhenYouStoppedSmokingResponseFactory.create(
+            response_set=self.response_set,
+            value=True,
+            duration_years=self.response.years_smoked_including_stopped() - 1
+        )
+        form = AgeWhenStartedSmokingForm(
+            instance=self.response,
+            data={
+                "value": 18
+            }
+        )
+        form.save()
+
+        self.assertFalse(PeriodsWhenYouStoppedSmokingResponse.objects.filter(response_set=self.response_set).exists())
+
+
+    def test_does_not_delete_periods_when_stopped_smoking_response_if_age_started_smoking_is_not_changed(self):
+        self.response.value = 18
+        self.response.save()
+
+        PeriodsWhenYouStoppedSmokingResponseFactory.create(
+            response_set=self.response_set,
+            value=True,
+            duration_years=self.response.years_smoked_including_stopped() - 1
+        )
+        form = AgeWhenStartedSmokingForm(
+            instance=self.response,
+            data={
+                "value": 18
+            }
+        )
+        form.save()
+
+        self.assertTrue(PeriodsWhenYouStoppedSmokingResponse.objects.filter(response_set=self.response_set).exists())

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

@@ -18,7 +18,7 @@ def setUp(self):
 
         # Following responses required by validator
         self.date_of_birth_response = DateOfBirthResponseFactory.create(response_set=self.response_set)
-        self.age_when_started_smoking_response = q.create(response_set=self.response_set)
+        self.age_when_started_smoking_response = AgeWhenStartedSmokingResponseFactory.create(response_set=self.response_set)
 
 
     def test_has_a_valid_factory(self):

lung_cancer_screening/questions/tests/unit/views/test_age_when_started_smoking.py

@@ -115,7 +115,7 @@ def test_redirects_to_periods_when_you_stopped_smoking(self):
 
         self.assertRedirects(response, reverse("questions:periods_when_you_stopped_smoking"))
 
-    def test_redirects_to_responses_if_change_query_param_is_true(self):
+    def test_redirects_to_periods_when_you_stopped_smoking_if_change_query_param_is_true(self):
         ResponseSetFactory.create(user=self.user, eligible=True)
 
         response = self.client.post(
@@ -126,7 +126,11 @@ def test_redirects_to_responses_if_change_query_param_is_true(self):
             }
         )
 
-        self.assertRedirects(response, reverse("questions:responses"))
+        self.assertRedirects(
+            response,
+            reverse("questions:periods_when_you_stopped_smoking", query={"change": "True"})
+        )
+
 
     def test_responds_with_422_if_the_response_fails_to_create(self):
         ResponseSetFactory.create(user=self.user, eligible=True)

lung_cancer_screening/questions/views/age_when_started_smoking.py

@@ -1,4 +1,4 @@
-from django.urls import reverse_lazy
+from django.urls import reverse, reverse_lazy
 from django.contrib.auth.mixins import LoginRequiredMixin
 
 from .mixins.ensure_response_set import EnsureResponseSet
@@ -13,3 +13,12 @@ class AgeWhenStartedSmokingView(LoginRequiredMixin, EnsureResponseSet, EnsureEli
     model = AgeWhenStartedSmokingResponse
     success_url = reverse_lazy("questions:periods_when_you_stopped_smoking")
     back_link_url = reverse_lazy("questions:relatives_age_when_diagnosed")
+
+    def get_success_url(self):
+        if self.should_redirect_to_responses(self.request):
+            return reverse(
+                "questions:periods_when_you_stopped_smoking",
+                query={"change": "True"}
+            )
+        else:
+            return super().get_success_url()