示例#1
0
    def test_get_votes_for_idea(self):
        # Create two ideas, give one of them one vote, and the other two votes,
        # make sure the expected counts are correct.
        user_1 = User.objects.create(username='******')
        user_2 = User.objects.create(username='******')

        # Two ideas
        idea_1 = create_idea(
            user=user_1,
            title='test',
            description='foobar',
            tag=CREATIVITY,
        )
        idea_2 = create_idea(
            user=user_1,
            title='test',
            description='foobar',
            tag=CREATIVITY,
        )

        # Three votes, one 1 and two for 2
        vote_1, _ = create_vote(
            user=user_1,
            idea=idea_1,
        )
        vote_2, _ = create_vote(
            user=user_1,
            idea=idea_2,
        )
        vote_3, _ = create_vote(
            user=user_2,
            idea=idea_2,
        )

        # Make sure the vote counts, vote pks are correct
        votes = get_votes_for_idea(idea_1)
        self.assertEqual(votes.count(), 1)
        self.assertEqual(
            set(votes.values_list('pk', flat=True)),
            set([vote_1.pk]),
        )

        votes = get_votes_for_idea(idea_2)
        self.assertEqual(votes.count(), 2)
        self.assertEqual(
            set(votes.values_list('pk', flat=True)),
            set([vote_2.pk, vote_3.pk]),
        )
示例#2
0
    def test_unique_together(self):
        user = User.objects.create()
        idea = create_idea(
            user=user,
            title='test',
            description='foobar',
            tag=CREATIVITY,
        )

        vote = Vote.objects.create(
            voter=user,
            idea=idea,
        )

        # Show that there is a unique together restraint here.
        with self.assertRaises(IntegrityError):
            Vote.objects.create(
                voter=user,
                idea=idea,
            )

        # Show that the helper method for creating votes deals with the unique
        # together restraint.
        second_vote, _ = create_vote(
            user=user,
            idea=idea,
        )
        self.assertEqual(vote.pk, second_vote.pk)
示例#3
0
 def test_no_ip_with_anon_user(self):
     user = AnonymousUser()
     idea = create_idea(
         user=User.objects.create(),
         title='test',
         description='foobar',
         tag=CREATIVITY,
     )
     with self.assertRaises(IntegrityError) as e:
         create_vote(
             user=user,
             idea=idea,
         )
     self.assertEqual(
         str(e.exception),
         'Must pass in ip address for anon users.')
示例#4
0
 def test_ip_address_for_anon_users(self):
     user = AnonymousUser()
     idea = create_idea(
         user=User.objects.create(),
         title='test',
         description='foobar',
         tag=CREATIVITY,
     )
     vote, _ = create_vote(
         user=user,
         idea=idea,
         ip='127.0.0.1',
     )
     self.assertEqual(vote.voter, None)
     self.assertEqual(vote.ip, '127.0.0.1')
     self.assertEqual(vote.idea, idea)
示例#5
0
def vote(request, idea_id):
    idea = get_object_or_404(
        Idea.objects.select_related(),
        pk=idea_id,
    )
    ip = request.META['REMOTE_ADDR']
    vote, created = create_vote(
        request.user,
        idea,
        ip='%s' % ip,
    )
    return HttpResponse(
        dumps(
            {'created': created},
        ),
        mimetype="application/json",
    )
示例#6
0
 def test_create_vote(self):
     # We will be using the create_idea function to create idea objects, so
     # I only ever have to change it in one place.
     user = User.objects.create()
     idea = create_idea(
         user=user,
         title='test',
         description='foobar',
         tag=CREATIVITY,
     )
     vote, _ = create_vote(
         user=user,
         idea=idea,
     )
     self.assertNotEqual(idea, None)
     self.assertEqual(vote.voter, user)
     self.assertEqual(vote.idea, idea)
     self.assertEqual(vote.ip, None)