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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures",
"test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test users.
self.users = [
User.objects.create_user("test%d" % i, "*****@*****.**")
for i in range(0, 3)
]
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures",
"test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_floor",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [
Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)
]
map(lambda f: f.save(), self.floors)
self.users = [
User.objects.create_user("test%d" % i, "*****@*****.**")
for i in range(0, 4)
]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test users.
self.users = [User.objects.create_user("test%d" % i, "*****@*****.**") for i in range(0, 3)]
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="floor_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)]
map(lambda f: f.save(), self.floors)
self.users = [User.objects.create_user("test%d" % i, "*****@*****.**") for i in range(0, 4)]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
class OverallFloorPrizeTest(TestCase):
"""
Tests awarding a prize to a dorm floor points winner.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="floor_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)]
map(lambda f: f.save(), self.floors)
self.users = [User.objects.create_user("test%d" % i, "*****@*****.**") for i in range(0, 4)]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
def testNumAwarded(self):
"""
Simple test to check that the number of prizes to be awarded is one.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(self.prize.num_awarded(), 1, "This prize should not be awarded to more than one user.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user will go ahead in points.
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(profile.floor), profile.floor,
"The user's floor is not leading in the prize.")
# Test that a user in a different floor changes the leader for the original user.
profile2 = self.users[2].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile2.save()
self.assertEqual(self.prize.leader(profile.floor), profile2.floor,
"The leader for this prize did not change.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round = "Overall"
self.prize.save()
# Test one user will go ahead in points.
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(profile.floor), profile.floor,
"The user's floor is not leading in the prize.")
# Test that a user in a different floor but same dorm changes the leader for the original user.
profile2 = self.users[2].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile2.save()
self.assertEqual(self.prize.leader(profile.floor), profile2.floor,
"The leader for this prize did not change.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
class FloorPrizeTest(TestCase):
"""
Tests awarding a prize to the individual on each floor with the most points.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures",
"test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_floor",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [
Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)
]
map(lambda f: f.save(), self.floors)
self.users = [
User.objects.create_user("test%d" % i, "*****@*****.**")
for i in range(0, 4)
]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
def testNumAwarded(self):
"""
Tests that the number of prizes awarded corresponds to the number of floors.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(self.prize.num_awarded(), len(self.floors),
"This should correspond to the number of floors.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile,
"Current prize leader is not the leading user.")
# Have a user on the same floor move ahead in points.
profile3 = self.users[2].get_profile()
profile3.add_points(11, datetime.datetime.today(), "test")
profile3.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the the leader.")
# Have the first user earn more points outside of the round.
profile.add_points(
2,
datetime.datetime.today() - datetime.timedelta(days=2), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3,
"User 3 should still be the leading profile.")
# Try a user on a different floor.
profile2 = self.users[1].get_profile()
profile2.add_points(20, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(
self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the leading profile on user 1's floor.")
self.assertEqual(
self.prize.leader(floor=profile2.floor), profile2,
"User 2 should be the leading profile on user 2's floor.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Overall"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile,
"Current prize leader is not the leading user.")
# Have a user on the same floor move ahead in points.
profile3 = self.users[2].get_profile()
profile3.add_points(11, datetime.datetime.today(), "test")
profile3.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the the leader.")
# Try a user on a different floor.
profile2 = self.users[1].get_profile()
profile2.add_points(20, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(
self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the leading profile on user 1's floor.")
self.assertEqual(
self.prize.leader(floor=profile2.floor), profile2,
"User 2 should be the leading profile on user 2's floor.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
class OverallPrizeTest(TestCase):
"""
Tests awarding a prize to the individual overall points winner.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures",
"test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test users.
self.users = [
User.objects.create_user("test%d" % i, "*****@*****.**")
for i in range(0, 3)
]
def testNumAwarded(self):
"""
Simple test to check that the number of prizes to be awarded is one.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(
self.prize.num_awarded(), 1,
"This prize should not be awarded to more than one user.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
top_points = Profile.objects.all().order_by("-points")[0].points
profile.add_points(
top_points + 1,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile,
"Current prize leader is not the leading user.")
# Have another user move ahead in points
profile2 = self.users[1].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today(),
"test")
profile2.save()
self.assertEqual(self.prize.leader(), profile2,
"User 2 should be the leading profile.")
# Have this user get the same amount of points, but an earlier award date.
profile3 = self.users[2].get_profile()
profile3.add_points(
profile2.points,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile3.save()
self.assertEqual(self.prize.leader(), profile2,
"User 2 should still be the leading profile.")
# Have the first user earn more points outside of the round.
profile.add_points(
2,
datetime.datetime.today() - datetime.timedelta(days=2), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile2,
"User 2 should still be the leading profile.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round = "Overall"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
top_points = Profile.objects.all().order_by("-points")[0].points
profile.add_points(top_points + 1, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile,
"Current prize leader is not the leading user.")
# Have another user move ahead in points
profile2 = self.users[1].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today(),
"test")
profile2.save()
self.assertEqual(self.prize.leader(), profile2,
"User 2 should be the leading profile.")
# Have this user get the same amount of points, but an earlier award date.
profile3 = self.users[2].get_profile()
profile3.add_points(
profile2.points,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile3.save()
self.assertEqual(self.prize.leader(), profile2,
"User 2 should still be the leading profile.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
class FloorPrizeTest(TestCase):
"""
Tests awarding a prize to the individual on each floor with the most points.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_floor",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)]
map(lambda f: f.save(), self.floors)
self.users = [User.objects.create_user("test%d" % i, "*****@*****.**") for i in range(0, 4)]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
def testNumAwarded(self):
"""
Tests that the number of prizes awarded corresponds to the number of floors.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(self.prize.num_awarded(), len(self.floors), "This should correspond to the number of floors.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile, "Current prize leader is not the leading user.")
# Have a user on the same floor move ahead in points.
profile3 = self.users[2].get_profile()
profile3.add_points(11, datetime.datetime.today(), "test")
profile3.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3, "User 3 should be the the leader.")
# Have the first user earn more points outside of the round.
profile.add_points(2, datetime.datetime.today() - datetime.timedelta(days=2), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3, "User 3 should still be the leading profile.")
# Try a user on a different floor.
profile2 = self.users[1].get_profile()
profile2.add_points(20, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the leading profile on user 1's floor.")
self.assertEqual(self.prize.leader(floor=profile2.floor), profile2,
"User 2 should be the leading profile on user 2's floor.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Overall"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
profile.add_points(10, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile, "Current prize leader is not the leading user.")
# Have a user on the same floor move ahead in points.
profile3 = self.users[2].get_profile()
profile3.add_points(11, datetime.datetime.today(), "test")
profile3.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3, "User 3 should be the the leader.")
# Try a user on a different floor.
profile2 = self.users[1].get_profile()
profile2.add_points(20, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(self.prize.leader(floor=profile.floor), profile3,
"User 3 should be the leading profile on user 1's floor.")
self.assertEqual(self.prize.leader(floor=profile2.floor), profile2,
"User 2 should be the leading profile on user 2's floor.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
class OverallPrizeTest(TestCase):
"""
Tests awarding a prize to the individual overall points winner.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test users.
self.users = [User.objects.create_user("test%d" % i, "*****@*****.**") for i in range(0, 3)]
def testNumAwarded(self):
"""
Simple test to check that the number of prizes to be awarded is one.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(self.prize.num_awarded(), 1, "This prize should not be awarded to more than one user.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
top_points = Profile.objects.all().order_by("-points")[0].points
profile.add_points(top_points + 1, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile, "Current prize leader is not the leading user.")
# Have another user move ahead in points
profile2 = self.users[1].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(self.prize.leader(), profile2, "User 2 should be the leading profile.")
# Have this user get the same amount of points, but an earlier award date.
profile3 = self.users[2].get_profile()
profile3.add_points(profile2.points, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile3.save()
self.assertEqual(self.prize.leader(), profile2, "User 2 should still be the leading profile.")
# Have the first user earn more points outside of the round.
profile.add_points(2, datetime.datetime.today() - datetime.timedelta(days=2), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile2, "User 2 should still be the leading profile.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round = "Overall"
self.prize.save()
# Test one user
profile = self.users[0].get_profile()
top_points = Profile.objects.all().order_by("-points")[0].points
profile.add_points(top_points + 1, datetime.datetime.today(), "test")
profile.save()
self.assertEqual(self.prize.leader(), profile, "Current prize leader is not the leading user.")
# Have another user move ahead in points
profile2 = self.users[1].get_profile()
profile2.add_points(profile.points + 1, datetime.datetime.today(), "test")
profile2.save()
self.assertEqual(self.prize.leader(), profile2, "User 2 should be the leading profile.")
# Have this user get the same amount of points, but an earlier award date.
profile3 = self.users[2].get_profile()
profile3.add_points(profile2.points, datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile3.save()
self.assertEqual(self.prize.leader(), profile2, "User 2 should still be the leading profile.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()
def testConstraints(self):
"""
Tests that the uniqueness constraints are enforced.
A prize with the same round_name, award_to, and competition_type as another cannot be created.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures", "test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
round_name="Round 1",
value=5,
)
prize.save()
prize2 = Prize(
title="Dup prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="individual_overall",
competition_type="points",
round_name="Round 1",
value=5,
)
try:
prize2.save()
self.fail("IntegrityError exception not thrown.")
except IntegrityError:
pass
prize2.round_name = "Overall"
try:
prize2.save()
except IntegrityError:
self.fail("IntegrityError exception should not be thrown.")
prize2.round_name = "Round 1"
prize2.competition_type = "energy"
try:
prize2.save()
except IntegrityError:
self.fail("IntegrityError exception should not be thrown.")
prize2.competition_type = "points"
prize2.award_to = "floor_overall"
try:
prize2.save()
except IntegrityError:
self.fail("IntegrityError exception should not be thrown.")
# Make sure to clean up!
prize.image.delete()
prize.delete()
prize2.image.delete()
prize2.delete()
class OverallFloorPrizeTest(TestCase):
"""
Tests awarding a prize to a dorm floor points winner.
"""
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.
"""
image_path = os.path.join(settings.PROJECT_ROOT, "fixtures",
"test_images", "test.jpg")
image = ImageFile(open(image_path, "r"))
self.prize = Prize(
title="Super prize!",
short_description="A test prize",
long_description="A test prize",
image=image,
award_to="floor_overall",
competition_type="points",
value=5,
)
self.saved_rounds = settings.COMPETITION_ROUNDS
self.current_round = "Round 1"
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 test dorms, floors, and users.
self.dorm = Dorm(name="Test Dorm")
self.dorm.save()
self.floors = [
Floor(number=str(i), dorm=self.dorm) for i in range(0, 2)
]
map(lambda f: f.save(), self.floors)
self.users = [
User.objects.create_user("test%d" % i, "*****@*****.**")
for i in range(0, 4)
]
# Assign users to floors.
for index, user in enumerate(self.users):
user.get_profile().floor = self.floors[index % 2]
user.get_profile().save()
def testNumAwarded(self):
"""
Simple test to check that the number of prizes to be awarded is one.
"""
self.prize.round_name = "Round 1"
self.prize.save()
self.assertEqual(
self.prize.num_awarded(), 1,
"This prize should not be awarded to more than one user.")
def testRoundLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round_name = "Round 1"
self.prize.save()
# Test one user will go ahead in points.
profile = self.users[0].get_profile()
profile.add_points(
10,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(profile.floor), profile.floor,
"The user's floor is not leading in the prize.")
# Test that a user in a different floor changes the leader for the original user.
profile2 = self.users[2].get_profile()
profile2.add_points(
profile.points + 1,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile2.save()
self.assertEqual(self.prize.leader(profile.floor), profile2.floor,
"The leader for this prize did not change.")
def testOverallLeader(self):
"""
Tests that we can retrieve the overall individual points leader for a round prize.
"""
self.prize.round = "Overall"
self.prize.save()
# Test one user will go ahead in points.
profile = self.users[0].get_profile()
profile.add_points(
10,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile.save()
self.assertEqual(self.prize.leader(profile.floor), profile.floor,
"The user's floor is not leading in the prize.")
# Test that a user in a different floor but same dorm changes the leader for the original user.
profile2 = self.users[2].get_profile()
profile2.add_points(
profile.points + 1,
datetime.datetime.today() - datetime.timedelta(minutes=1), "test")
profile2.save()
self.assertEqual(self.prize.leader(profile.floor), profile2.floor,
"The leader for this prize did not change.")
def tearDown(self):
"""
Deletes the created image file in prizes.
"""
settings.COMPETITION_ROUNDS = self.saved_rounds
self.prize.image.delete()
self.prize.delete()