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_create_idea(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)
self.assertNotEqual(idea, None)
self.assertEqual(idea.author, user)
self.assertEqual(idea.title, "test")
self.assertEqual(idea.description, "foobar")
def test_simple(self):
user = User.objects.create()
idea = create_idea(
user=user,
title='test',
description='foobar',
tag=CREATIVITY,
)
vote = Vote.objects.create(
voter=user,
idea=idea,
)
self.assertNotEqual(vote, None)
def test_view_idea(self):
c = self.client
TITLE = "Title of the idea"
DESCRIPTION = "Description of the idea"
idea = create_idea(user=self.user, title=TITLE, description=DESCRIPTION, tag=CREATIVITY)
view_idea = reverse("idea_view", kwargs={"idea_id": idea.pk})
with self.assertNumQueries(3):
r = c.get(view_idea)
self.assertContains(r, TITLE)
self.assertContains(r, DESCRIPTION)
self.assertContains(r, idea.get_tag_display().lower())
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 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_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)
