def test_newDuplicatePitch(self):
"""Creating a new Pitch for an existing Poll
with the same 'choice_id' should fail.
"""
self.pitch = Pitch(poll=self.poll, content=self._content, choice_id=self._choice, editor=self.user)
try:
self.pitch.save()
except IntegrityError:
return True
def setUp(self):
self.user = User.objects.create_user("user1", "*****@*****.**", password="******")
self.user.save()
self.second_user = User.objects.create_user("user2", "*****@*****.**", password="******")
self.second_user.save()
self._question = "Is Foo the new Bar?"
self._guid = create_poll_guid(self._question)
self._choice = "a"
self._content = "No, Foo will always be tops!"
self.poll = Poll(guid=self._guid, owner=self.user, question=self._question)
self.poll.save()
self.pitch = Pitch(poll=self.poll, content=self._content, choice_id=self._choice, editor=self.user)
self.pitch.save()
self.pitch.vote() #User who creates Pitch automatically Votes for it.
def test_AddNewPitchbyNewUser(self):
getpoll = Poll.objects.get(guid=self._guid)
new_pitch = Pitch(poll=getpoll, editor=self.second_user, content="Bar will rise and defeat Foo!")
new_pitch.save()
