def test_claim(self):
dlist = DiningList.get_latest()
dlist.save()
self.client.login(username="******", password="******")
# We test claiming the dining list without participation
r = self.client.get(self.view_url)
dlist = DiningList.get_latest()
self.assertIsNone(dlist.owner)
# We test claiming the dining list with a participation
dpart = DiningParticipation(user=self.testUser, dining_list=dlist)
dpart.save()
self.assertIsNone(dlist.owner)
r = self.client.get(self.view_url)
self.assertEqual(r.status_code, 302)
dlist = DiningList.get_latest()
self.assertIsNotNone(dlist.owner)
self.assertEqual(dlist.owner, self.testUser)
# Test that you cannot claim a dining list over someone else
dpart = DiningParticipation(user=self.testUser2, dining_list=dlist)
dpart.save()
self.client.login(username=self.testUser2.username, password="******")
r = self.client.get(self.view_url)
self.assertEqual(r.status_code, 302)
dlist = DiningList.get_latest()
self.assertNotEqual(dlist.owner, self.testUser2)
self.assertEqual(dlist.owner, self.testUser)
dpart.delete()
dlist.delete()
def test_get_registered_user(self):
dlist = DiningList().get_latest()
dlist.save()
dpart = DiningParticipation(user=self.testUser, dining_list=dlist)
dpart.save()
# Tests
self.assertEqual(len(dlist.get_participants()), 1)
dpart.delete()
dlist.delete()
def test_new_participation(self):
dlist = DiningList.get_latest()
dlist.owner = self.testUser
dlist.save()
dpart = DiningParticipation(user=self.testUser, dining_list=dlist)
dpart.work_cook = True
dpart.save()
dstats = DiningStats.objects.get(user=self.testUser)
self.assertEquals(dstats.total_helped, 1)
self.assertEquals(dstats.total_participated, 1)
# Now we update
dpart.work_cook = False
dpart.save()
dstats = DiningStats.objects.get(user=self.testUser)
self.assertEqual(dstats.total_helped, 0)
self.assertEqual(dstats.total_participated, 1)
# Now we test for cooking and dishes at the same time
dpart.work_cook = True
dpart.work_dishes = True
dpart.save()
dstats = DiningStats.objects.get(user=self.testUser)
self.assertEqual(dstats.total_helped, 2)
self.assertEqual(dstats.total_participated, 1)