PollTest

from datetime import datetime
from django.test import TestCase
from mysite.polls.models import Poll,Choice
class PollTest(TestCase):

    def setUp(self):
        question="So What?"
        now = datetime.now()
        self.poll = Poll.objects.create(question=question, pub_date=now)
        self.poll.choice_set.create(choice="Red", votes=0)
        self.poll.choice_set.create(choice="Blue", votes=0)
        self.poll.choice_set.create(choice="Green", votes=0)
        self.choice_list = self.poll.choice_set.all()
        self.choice = Choice.objects.create(choice="Orange", votes=1,poll=self.poll)

    def test_models(self):
        now = datetime.now()
        #test poll question,choices related to vote
        self.assertEqual(self.poll.choice_set.all().count(), 4)
        self.assertEqual(self.poll.question,"So What?")
        self.assertEqual(self.poll.pub_date.day,now.day)
        self.assertEqual(self.poll.was_published_today(),True)


        self.assertEqual(self.choice_list[0].choice,"Red")
        self.assertEqual(self.choice_list[0].votes,0)
        self.assertEqual(self.choice_list[0].poll.question,"So What?")
        self.assertEqual(self.choice_list[0].poll.was_published_today(),True)

        self.assertEqual(self.choice_list[1].choice,"Blue")
        self.choice_list[0].delete()
        self.assertEqual(self.poll.choice_set.all().count(), 3)