def test_check_achievements_wallposts(self): create_achievements() now = datetime.utcnow().replace(tzinfo=utc) user = User.objects.create_user(username='******') other = User.objects.create_user(username='******') user.friends.add(other) self.assertFalse(user.has_achievement(get_achievement("Outspoken"))) self.assertFalse(user.has_achievement(get_achievement("Social climber"))) for i in range(1, 51): WallPost.objects.create(message="bla", pub_date=now, poster=user, receiver=other) check_achievements(user.id) self.assertTrue(user.has_achievement(get_achievement("Outspoken"))) self.assertFalse(user.has_achievement(get_achievement("Social climber"))) for i in range(1, 151): WallPost.objects.create(message="bla", pub_date=now, poster=user, receiver=other) check_achievements(user.id) self.assertTrue(user.has_achievement(get_achievement("Outspoken"))) self.assertTrue(user.has_achievement(get_achievement("Social climber")))
def test_check_achievements_friends(self): create_achievements() user = User.objects.create_user(username='******') check_achievements(user.id) self.assertFalse(user.has_achievement(get_achievement("Friendly"))) self.assertFalse(user.has_achievement(get_achievement("Super Friendly"))) self.assertFalse(user.has_achievement(get_achievement("Politician"))) for i in range(1, 51): temp = User.objects.create_user(username=str(i)) user.friends.add(temp) check_achievements(user.id) self.assertTrue(user.has_achievement(get_achievement("Friendly"))) self.assertFalse(user.has_achievement(get_achievement("Super Friendly"))) self.assertFalse(user.has_achievement(get_achievement("Politician"))) for i in range(1, 51): temp = User.objects.create_user(username=str(i*100)) user.friends.add(temp) check_achievements(user.id) self.assertTrue(user.has_achievement(get_achievement("Friendly"))) self.assertTrue(user.has_achievement(get_achievement("Super Friendly"))) self.assertFalse(user.has_achievement(get_achievement("Politician"))) for i in range(1, 500): temp = User.objects.create_user(username=str(i*10000)) user.friends.add(temp) check_achievements(user.id) self.assertTrue(user.has_achievement(get_achievement("Politician"))) self.assertTrue(user.has_achievement(get_achievement("Super Friendly"))) self.assertTrue(user.has_achievement(get_achievement("Politician")))