mirror of
https://github.com/Sosokker/ku-polls.git
synced 2025-12-18 13:04:05 +01:00
106 lines
4.0 KiB
Python
106 lines
4.0 KiB
Python
import datetime
|
|
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from django.urls import reverse
|
|
|
|
from ..models import Question
|
|
|
|
|
|
class QuestionModelTests(TestCase):
|
|
def test_was_published_recently_with_future_question(self):
|
|
"""
|
|
was_published_recently() returns False for questions whose pub_date
|
|
is in the future.
|
|
"""
|
|
time = timezone.now() + datetime.timedelta(days=30)
|
|
future_question = Question(pub_date=time)
|
|
self.assertIs(future_question.was_published_recently(), False)
|
|
|
|
def test_was_published_recently_with_old_question(self):
|
|
"""
|
|
was_published_recently() returns False for questions whose pub_date
|
|
is older than 1 day.
|
|
"""
|
|
time = timezone.now() - datetime.timedelta(days=1, seconds=1)
|
|
old_question = Question(pub_date=time)
|
|
self.assertIs(old_question.was_published_recently(), False)
|
|
|
|
def test_was_published_recently_with_recent_question(self):
|
|
"""
|
|
was_published_recently() returns True for questions whose pub_date
|
|
is within the last day.
|
|
"""
|
|
time = timezone.now() - datetime.timedelta(hours=23, minutes=59, seconds=59)
|
|
recent_question = Question(pub_date=time)
|
|
self.assertIs(recent_question.was_published_recently(), True)
|
|
|
|
def test_is_published_with_future_question(self):
|
|
"""
|
|
is_published() should return False for questions whos pub_date is in the
|
|
future.
|
|
"""
|
|
future_date = timezone.now() + datetime.timedelta(days=30)
|
|
future_question = Question(pub_date=future_date)
|
|
self.assertIs(future_question.is_published(), False)
|
|
|
|
def test_default_pub_date(self):
|
|
"""
|
|
Questions with the default pub_date (now) are displayed on the index page.
|
|
"""
|
|
question = Question.objects.create(question_text="Default pub date question.")
|
|
|
|
response = self.client.get(reverse("polls:index"))
|
|
self.assertQuerySetEqual(
|
|
response.context["latest_question_list"]["all_poll"],
|
|
[question],
|
|
)
|
|
|
|
def test_is_published_with_past_question(self):
|
|
"""
|
|
is_published() should return True for questions whose pub_date is in the
|
|
past.
|
|
"""
|
|
past_date = timezone.now() - datetime.timedelta(days=1)
|
|
past_question = Question(pub_date=past_date)
|
|
self.assertIs(past_question.is_published(), True)
|
|
|
|
def test_can_vote_with_question_not_ended(self):
|
|
"""
|
|
can_vote() should return True for questions that are published and have not
|
|
ended.
|
|
"""
|
|
pub_date = timezone.now() - datetime.timedelta(hours=1)
|
|
end_date = timezone.now() + datetime.timedelta(hours=1)
|
|
question = Question(pub_date=pub_date, end_date=end_date)
|
|
self.assertIs(question.can_vote(), True)
|
|
|
|
def test_can_vote_with_question_ended(self):
|
|
"""
|
|
can_vote() should return False for questions that are published but have
|
|
ended.
|
|
"""
|
|
pub_date = timezone.now() - datetime.timedelta(hours=2)
|
|
end_date = timezone.now() - datetime.timedelta(hours=1)
|
|
question = Question(pub_date=pub_date, end_date=end_date)
|
|
self.assertIs(question.can_vote(), False)
|
|
|
|
def test_can_vote_with_question_no_end_date(self):
|
|
"""
|
|
can_vote() should return True for questions that are published and have no
|
|
specified end date.
|
|
"""
|
|
pub_date = timezone.now() - datetime.timedelta(hours=1)
|
|
question = Question(pub_date=pub_date, end_date=None)
|
|
self.assertIs(question.can_vote(), True)
|
|
|
|
def test_can_vote_with_question_ending_in_future(self):
|
|
"""
|
|
can_vote() should return True for questions that are published and
|
|
the current time is within the allowed voting period.
|
|
"""
|
|
pub_date = timezone.now() - datetime.timedelta(hours=1)
|
|
end_date = timezone.now() + datetime.timedelta(hours=2)
|
|
question = Question(pub_date=pub_date, end_date=end_date)
|
|
self.assertIs(question.can_vote(), True)
|