Source code for polls.tests.test_models

from datetime import datetime, timedelta

from polls.models import Poll
from polls.tests.utils import BaseTestCase


class TestPolls(BaseTestCase):

    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())

    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())