Run local acceptance tests as part of test unit scirpt

Author: Themitchell

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_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 value in (fifty_five_years_ago, seventy_five_years_ago):
                 DateResponse.objects.create(
-                    participant=participant,
+                    participant=request.participant,
                     value=value,
                     question="What is your date of birth?"
                 )
@@ -41,12 +36,12 @@ def date_of_birth(request):
             return render(
                 request,
                 "date_of_birth.jinja",
-                {"participant_id": participant.unique_id},
+                {"participant_id": request.participant.unique_id},
                 status=422
             )
 
     return render(
         request,
         "date_of_birth.jinja",
-        { "participant_id": participant.unique_id }
+        { "participant_id": request.participant.unique_id }
     )

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

lung_cancer_screening/questions/views/decorators/participant_decorators.py

@@ -2,24 +2,19 @@
 from django.urls import reverse
 from django.core.exceptions import ValidationError
 
-from ..models.participant import Participant
-from ..models.boolean_response import BooleanResponse
+from .decorators.participant_decorators import require_participant
 
+from ..models.boolean_response import BooleanResponse
 
+@require_participant
 def have_you_ever_smoked(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 = int(request.POST['value'])
 
             if value:
                 BooleanResponse.objects.create(
-                    participant=participant,
+                    participant=request.participant,
                     value=value,
                     question="Have you ever smoked?"
                 )
@@ -32,12 +27,12 @@ def have_you_ever_smoked(request):
             return render(
                 request,
                 "have_you_ever_smoked.jinja",
-                {"participant_id": participant.unique_id},
+                {"participant_id": request.participant.unique_id},
                 status=422
             )
 
     return render(
         request,
         "have_you_ever_smoked.jinja",
-        {"participant_id": participant.unique_id}
+        {"participant_id": request.participant.unique_id}
     )

lung_cancer_screening/questions/views/have_you_ever_smoked.py

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

lung_cancer_screening/questions/views/non_smoker_exit.py

@@ -1,16 +1,11 @@
-from django.shortcuts import render, redirect
-from django.urls import reverse
+from django.shortcuts import render
 
-from ..models.participant import Participant
+from .decorators.participant_decorators import require_participant
 
+@require_participant
 def responses(request):
-    try:
-        participant = Participant.objects.get(unique_id=request.session['participant_id'])
-    except Participant.DoesNotExist:
-        return redirect(reverse("questions:start"))
-
     return render(
         request,
         "responses.jinja",
-        {"responses": participant.responses()}
+        {"responses": request.participant.responses()}
     )