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