from datetime import datetime, timedelta
from polls.models import Poll
from polls.tests.utils import BaseTestCase
[docs]class TestPolls(BaseTestCase):
[docs] def test_poll_create(self):
""" Can we create a poll?
* Seems trivial now
* But for complex systems what started out as a simple create can get complex
* Get your test coverage up!
"""
poll_count = Poll.objects.count()
poll = Poll(
question="Why is Python awesome?",
pub_date=datetime.now()
)
poll.save()
self.assertTrue(poll_count < Poll.objects.count())
[docs] def test_was_published_today(self):
poll = Poll(
question="Django is for the internets",
pub_date=datetime.now()
)
poll.save()
self.assertTrue(poll.was_published_today())
poll.pub_date = datetime.now() - timedelta(days=3)
poll.save()
self.assertFalse(poll.was_published_today())