sandbox/django/mysite/polls/tests.py

88 lines
3.5 KiB
Python
Raw Normal View History

2018-05-07 08:12:14 +00:00
import datetime
2018-05-06 23:47:59 +00:00
from django.test import TestCase
2018-05-07 08:12:14 +00:00
from django.urls import reverse
from django.utils import timezone
from . import models
def create_question(question_text, days):
"""
Create a question with the given `question_text` and published the
given number of `days` offset to now (negative for questions published
in the past, positive for questions that have yet to be published).
"""
time = timezone.now() + datetime.timedelta(days=days)
return models.Question.objects.create(question_text=question_text, pub_date=time)
2018-05-06 23:47:59 +00:00
# Create your tests here.
2018-05-07 08:12:14 +00:00
class QuestionModelTests(TestCase):
def test_was_published_recently(self):
"""
was_published_recently returns False for questions whose
pub_date is in the future.
"""
time = timezone.now() + datetime.timedelta(days=30)
future_question = models.Question(pub_date=time)
self.assertIs(future_question.was_published_recently(), False)
def test_was_published_recently_with_old_question(self):
"""
was_published_recently should return False for questions whose
pub_date falls outside the 'recent' range.
"""
time = timezone.now() + (models.recent + datetime.timedelta(microseconds=1))
old_question = models.Question(pub_date=time)
self.assertIs(old_question.was_published_recently(), False)
def test_was_published_recently_with_recent_question(self):
# build a timestamp that falls just inside our recent gateway.
time = timezone.now() - (models.recent - datetime.timedelta(seconds=1))
recent_question = models.Question(pub_date=time)
self.assertIs(recent_question.was_published_recently(), True)
def test_future_question_and_past_question(self):
"""
Even if both past and future questions exist, only past questions
are displayed.
"""
create_question(question_text="Past question.", days=-30)
create_question(question_text="Future question.", days=30)
response = self.client.get(reverse('polls:index'))
self.assertQuerysetEqual(
response.context['latest_questions'],
['<Question: Past question.>']
)
def test_two_past_questions(self):
"""
The questions index page may display multiple questions.
"""
create_question(question_text="Past question 1.", days=-30)
create_question(question_text="Past question 2.", days=-5)
response = self.client.get(reverse('polls:index'))
self.assertQuerysetEqual(
response.context['latest_questions'],
['<Question: Past question 2.>', '<Question: Past question 1.>']
)
class QuestionDetailViewTests(TestCase):
def test_future_question(self):
"""
The detail view of a question with a pub_date in the future
returns a 404 not found.
"""
future_question = create_question(question_text='Future question.', days=5)
url = reverse('polls:detail', args=(future_question.id,))
response = self.client.get(url)
self.assertEqual(response.status_code, 404)
def test_past_question(self):
"""
The detail view of a question with a pub_date in the past
displays the question's text.
"""
past_question = create_question(question_text='Past Question.', days=-5)
url = reverse('polls:detail', args=(past_question.id,))
response = self.client.get(url)
self.assertContains(response, past_question.question_text)