Testing the DetailView

ddriddle · ddriddle · commit 1172d90634f0 · 2015-06-03T16:02:53.000-05:00
modified: mysite/polls/tests.py modified: mysite/polls/views.py https://docs.djangoproject.com/en/1.8/intro/tutorial05/#testing-the-detailview
diff --git a/mysite/polls/tests.py b/mysite/polls/tests.py
@@ -105,3 +105,28 @@ def test_index_view_with_two_past_questions(self):
             response.context['latest_question_list'],
             ['<Question: Past question 2.>', '<Question: Past question 1.>']
         )
+
+
+class QuestionIndexDetailTests(TestCase):
+    def test_detail_view_with_a_future_question(self):
+        """
+        The detail view of a question with a pub_date in the future should
+        return a 404 not found.
+        """
+        future_question = create_question(question_text='Future question.',
+                                          days=5)
+        response = self.client.get(reverse('polls:detail',
+                                   args=(future_question.id,)))
+        self.assertEqual(response.status_code, 404)
+
+    def test_detail_view_with_a_past_question(self):
+        """
+        The detail view of a question with a pub_date in the past should
+        display the question's text.
+        """
+        past_question = create_question(question_text='Past Question.',
+                                        days=-5)
+        response = self.client.get(reverse('polls:detail',
+                                   args=(past_question.id,)))
+        self.assertContains(response, past_question.question_text,
+                            status_code=200)
diff --git a/mysite/polls/views.py b/mysite/polls/views.py
@@ -25,6 +25,12 @@ class DetailView(generic.DetailView):
     model = Question
     template_name = 'polls/detail.html'
 
+    def get_queryset(self):
+        """
+        Excludes any questions that aren't published yet.
+        """
+        return Question.objects.filter(pub_date__lte=timezone.now())
+
 
 class ResultsView(generic.DetailView):
     model = Question
