def test_delete_a_wishlist(self):
""" Delete a Wishlists """
wishlist = Wishlist("mike's wishlist", "mike")
wishlist.save()
self.assertEqual(len(Wishlist.all()), 1)
# delete the wishlist and make sure it isn't in the database
wishlist.delete_wishlist()
self.assertEqual(len(Wishlist.all()), 0)
def test_delete_error(self):
""" Test failure of the delete try-catch"""
wishlist = Wishlist("mike's wishlist", "mike")
wishlist.save()
self.assertEqual(len(Wishlist.all()), 1)
og_id = wishlist.id
# modify id to trigger try-catch
wishlist.id = "asdf123"
# delete the wishlist and make sure it failed and is still int the database
wishlist.delete_wishlist()
self.assertEqual(len(Wishlist.all()), 1)
# Fetch it back and make sure the id and data hasn't changed
wishlists = Wishlist.all()
self.assertEqual(len(wishlists), 1)
self.assertEqual(wishlists[0].name, "mike's wishlist")
self.assertEqual(wishlists[0].id, og_id)
def test_key_error_on_delete(self, bad_mock):
""" Test KeyError on delete """
bad_mock.side_effect = KeyError()
wishlist = Wishlist("joan's wishlist", "joan")
wishlist.create()
wishlist.delete_wishlist()
