Exemplo n.º 1
0
    def test_votes(self):
        profile2 = ProfileFactory()
        user2 = profile2.user

        alist = ListFactory()
        alist.owner = user2
        alist.save()

        self.assertEqual(alist.get_votes(), 0)

        alist.vote(self.user)

        self.assertEqual(alist.get_votes(), 1)

        alist.unvote(self.user)

        self.assertEqual(alist.get_votes(), 0)