def add_favourite_test(self):
self.assertEqual(
0,
len(self.retrieve_user().favourites)
)
self.req.GET = {"draw_id": self.test_draw._id}
add_favorite(self.req)
self.assertEqual(
1,
len(self.retrieve_user().favourites)
)
def add_wrong_drawn_test(self):
self.assertEqual(
0,
len(self.retrieve_user().favourites)
)
self.req.GET = {"draw_id": "FAKE"}
self.assertRaises(
Exception,
lambda: add_favorite(self.req)
)
def add_not_owner_test(self):
self.assertEqual(
0,
len(self.retrieve_user().favourites)
)
self.test_draw.owner = "FAKE"
self._driver.save_draw(self.test_draw)
self.req.GET = {"draw_id": self.test_draw._id}
self.assertRaises(
PermissionDenied,
lambda: add_favorite(self.req)
)
