class RafflePrizeTests(TestCase):
"""
Tests the RafflePrize model.
"""
def setUp(self):
"""
Sets up a test individual prize for the rest of the tests.
This prize is not saved, as the round field is not yet set.
"""
self.saved_rounds = settings.COMPETITION_ROUNDS
start = datetime.date.today()
end = start + datetime.timedelta(days=7)
settings.COMPETITION_ROUNDS = {
"Round 1" : {
"start": start.strftime("%Y-%m-%d"),
"end": end.strftime("%Y-%m-%d"),
},
}
# Create a test user
self.user = User.objects.create_user("user", "*****@*****.**", password="******")
# Set up raffle deadline
self.deadline = RaffleDeadline(
round_name="Round 1",
pub_date=datetime.datetime.today() - datetime.timedelta(hours=1),
end_date=datetime.datetime.today() + datetime.timedelta(days=5),
)
self.deadline.save()
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = RafflePrize(
title="Super prize!",
description="A test prize",
image=image,
value=5,
deadline=self.deadline,
)
def testTicketAllocation(self):
"""
Tests that a user can allocate a ticket.
"""
self.prize.round_name = "Round 1"
self.prize.save()
profile = self.user.get_profile()
profile.add_points(25, datetime.datetime.today(), "test")
profile.save()
# Add a ticket to the prize
self.assertEqual(profile.available_tickets(), 1, "User should have one raffle ticket.")
self.prize.add_ticket(self.user)
self.assertEqual(profile.available_tickets(), 0, "User should not have any raffle tickets.")
self.assertEqual(self.prize.allocated_tickets(), 1, "1 ticket should be allocated to this prize.")
self.assertEqual(self.prize.allocated_tickets(self.user), 1, "1 ticket should be allocated by this user to this prize.")
# Have another user add a ticket to the prize.
user2 = User.objects.create_user("user2", "*****@*****.**", password="******")
profile = user2.get_profile()
profile.add_points(25, datetime.datetime.today(), "test")
profile.save()
# Add a ticket to the prize
self.prize.add_ticket(user2)
self.assertEqual(self.prize.allocated_tickets(), 2, "2 tickets should be allocated to this prize.")
self.assertEqual(self.prize.allocated_tickets(user2), 1, "1 ticket should be allocated by this user to this prize.")
# Add another ticket to the prize.
profile.add_points(25, datetime.datetime.today(), "test")
profile.save()
self.prize.add_ticket(user2)
self.assertEqual(self.prize.allocated_tickets(), 3, "3 tickets should be allocated to this prize.")
self.assertEqual(self.prize.allocated_tickets(user2), 2, "2 tickets should be allocated by this user to this prize.")
# Remove a ticket from the prize.
self.prize.remove_ticket(self.user)
self.assertEqual(self.prize.allocated_tickets(), 2, "2 tickets should be allocated to this prize.")
self.assertEqual(self.prize.allocated_tickets(self.user), 0, "No tickets should be allocated by this user to this prize.")
self.prize.remove_ticket(user2)
self.assertEqual(self.prize.allocated_tickets(), 1, "1 ticket should be allocated to this prize.")
self.assertEqual(self.prize.allocated_tickets(user2), 1, "1 ticket should be allocated by this user to this prize.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
