Run local acceptance tests as part of test unit scirpt

Author: Themitchell

lung_cancer_screening/questions/jinja2/date_of_birth.jinja

@@ -36,8 +36,6 @@
           ]
         }) }}
 
-        <input type="hidden" name="participant_id" value="{{ participant_id }}">
-
         {{ button({
           "text": "Continue"
         }) }}

lung_cancer_screening/questions/jinja2/have_you_ever_smoked.jinja

@@ -29,8 +29,6 @@
           ]
         }) }}
 
-        <input type="hidden" name="participant_id" value="{{ participant_id }}">
-
         {{ button({
             "text": "Continue"
         }) }}

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

@@ -0,0 +1,28 @@
+from django.test import TestCase
+from django.urls import reverse
+
+from lung_cancer_screening.questions.models.participant import Participant
+
+class TestPostAgeRangeExit(TestCase):
+    def setUp(self):
+        participant = Participant.objects.create(unique_id="12345")
+
+        session = self.client.session
+        session['participant_id'] = participant.unique_id
+        session.save()
+
+    def test_get_redirects_if_the_particpant_does_not_exist(self):
+        session = self.client.session
+        session['participant_id'] = "somebody none existant participant"
+        session.save()
+
+        response = self.client.get(
+            reverse("questions:age_range_exit")
+        )
+
+        self.assertRedirects(response, reverse("questions:start"))
+
+    def test_get_responds_successfully(self):
+        response = self.client.get(reverse("questions:age_range_exit"))
+
+        self.assertEqual(response.status_code, 200)

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

@@ -43,14 +43,6 @@ def test_get_responds_successfully(self):
 
         self.assertEqual(response.status_code, 200)
 
-    def test_get_contains_the_participant_id_in_the_form(self):
-        response = self.client.get(reverse("questions:date_of_birth"))
-
-        self.assertContains(
-            response,
-            f"<input type=\"hidden\" name=\"participant_id\" value=\"{self.participant.unique_id}\">"
-        )
-
     def test_post_redirects_if_the_particpant_does_not_exist(self):
         session = self.client.session
         session['participant_id'] = "somebody none existant participant"

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

@@ -29,14 +29,6 @@ def test_get_responds_successfully(self):
 
         self.assertEqual(response.status_code, 200)
 
-    def test_get_contains_the_participant_id_in_the_form(self):
-        response = self.client.get(reverse("questions:have_you_ever_smoked"))
-
-        self.assertContains(
-            response,
-            f"<input type=\"hidden\" name=\"participant_id\" value=\"{self.participant.unique_id}\">"
-        )
-
     def test_post_redirects_if_the_particpant_does_not_exist(self):
         session = self.client.session
         session['participant_id'] = "somebody none existant participant"

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

@@ -0,0 +1,28 @@
+from django.test import TestCase
+from django.urls import reverse
+
+from lung_cancer_screening.questions.models.participant import Participant
+
+class TestPostNonSmokerExit(TestCase):
+    def setUp(self):
+        participant = Participant.objects.create(unique_id="12345")
+
+        session = self.client.session
+        session['participant_id'] = participant.unique_id
+        session.save()
+
+    def test_get_redirects_if_the_particpant_does_not_exist(self):
+        session = self.client.session
+        session['participant_id'] = "somebody none existant participant"
+        session.save()
+
+        response = self.client.get(
+            reverse("questions:non_smoker_exit")
+        )
+
+        self.assertRedirects(response, reverse("questions:start"))
+
+    def test_get_responds_successfully(self):
+        response = self.client.get(reverse("questions:non_smoker_exit"))
+
+        self.assertEqual(response.status_code, 200)

lung_cancer_screening/questions/views/age_range_exit.py

@@ -1,5 +1,8 @@
 from django.shortcuts import render
 
+from .decorators.participant_decorators import require_participant
+
+@require_participant
 def age_range_exit(request):
     return render(
         request,

lung_cancer_screening/questions/views/date_of_birth.py

@@ -4,17 +4,12 @@
 from dateutil.relativedelta import relativedelta
 from django.core.exceptions import ValidationError
 
-from ..models.participant import Participant
-from ..models.date_response import DateResponse
+from .decorators.participant_decorators import require_participant
 
+from ..models.date_response import DateResponse
 
+@require_participant
 def date_of_birth(request):
-    try:
-        participant = Participant.objects.get(
-            unique_id=request.session['participant_id'])
-    except Participant.DoesNotExist:
-        return redirect(reverse("questions:start"))
-
     if request.method == "POST":
         try:
             value = date(
@@ -28,7 +23,7 @@ def date_of_birth(request):
 
             if (seventy_five_years_ago < value <= fifty_five_years_ago):
                 DateResponse.objects.create(
-                    participant=participant,
+                    participant=request.participant,
                     value=value,
                     question="What is your date of birth?"
                 )
@@ -41,12 +36,10 @@ def date_of_birth(request):
             return render(
                 request,
                 "date_of_birth.jinja",
-                {"participant_id": participant.unique_id},
                 status=422
             )
 
     return render(
         request,
-        "date_of_birth.jinja",
-        { "participant_id": participant.unique_id }
+        "date_of_birth.jinja"
     )

lung_cancer_screening/questions/views/decorators/__init__.py

@@ -0,0 +1,17 @@
+from functools import wraps
+from django.shortcuts import redirect
+from django.urls import reverse
+
+from lung_cancer_screening.questions.models.participant import Participant
+
+def require_participant(function):
+    @wraps(function)
+    def wrap(request, *args, **kwargs):
+        try:
+            request.participant = Participant.objects.get(
+                unique_id=request.session['participant_id'])
+        except Participant.DoesNotExist:
+            return redirect(reverse("questions:start"))
+
+        return function(request, *args, **kwargs)
+    return wrap