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.
class PollCreateTestCase(unittest.TestCase): 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 tearDown(self): for user in User.objects.all(): user.delete() 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 test_GetPoll(self): testslug = "is-foo-the-new-bar" testguid = create_poll_guid(testslug) poll = Poll.objects.get(guid=testguid) pitch = poll.pitch_set.get(editor=self.user) self.assertEquals(self._question, poll.question) self.assertEquals(self.user, poll.owner) self.assertEquals(self._content, pitch.content) 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() #print getpoll.pitch_set.all() def test_VoteForPitch(self): pitch = Pitch.objects.get(poll__guid=self._guid, choice_id="a") pitch.vote() def test_editPitch(self): pitch = Pitch.objects.get(poll__guid=self._guid, choice_id="a") pitch.content = "Foo is best eva." pitch.save() revs = PitchEditRevision.objects.all() self.assertEquals(len(revs), 2) self.assertEquals(revs[0].content, pitch.content) self.assertEquals(revs[1].content, self._content)
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()