def test_delete_a_pet(self):
""" Delete a Pet """
pet = Pet("fido", "dog")
pet.save()
self.assertEqual(len(Pet.all()), 1)
# delete the pet and make sure it isn't in the database
pet.delete()
self.assertEqual(len(Pet.all()), 0)
def test_delete_a_pet(self):
""" Delete a Pet """
pet = Pet(name="fido", category="dog", available=True)
pet.create()
self.assertEqual(len(Pet.all()), 1)
# delete the pet and make sure it isn't in the database
pet.delete()
self.assertEqual(len(Pet.all()), 0)
def test_key_error_on_delete(self, bad_mock):
""" Test KeyError on delete """
bad_mock.side_effect = KeyError()
pet = Pet("fido", "dog", False)
pet.create()
pet.delete()
