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]),
)
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)
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.')
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)
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",
)
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)