diff --git a/polls/tests.py b/polls/tests.py index be761e3..d5da4b0 100644 --- a/polls/tests.py +++ b/polls/tests.py @@ -14,4 +14,22 @@ class QuestionModelTests(TestCase): """ time = timezone.now() + datetime.timedelta(days=30) future_question = Question(pub_date=time) - self.assertIs(future_question.was_published_recently(), False) \ No newline at end of file + 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)