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) )