def test_key_error_on_update(self, bad_mock):
""" Test KeyError on update """
bad_mock.side_effect = KeyError()
pet = Pet("fido", "dog", False)
pet.save()
pet.name = 'Fifi'
pet.update()
def test_update_a_pet(self):
""" Update a Pet """
pet = Pet(name="fido", category="dog", available=True)
pet.create()
self.assertEqual(pet.id, 1)
# Change it an update it
pet.category = "k9"
pet.update()
self.assertEqual(pet.id, 1)
# Fetch it back and make sure the id hasn't changed
# but the data did change
pets = Pet.all()
self.assertEqual(len(pets), 1)
self.assertEqual(pets[0].category, "k9")
def test_update_a_pet(self):
""" Update a Pet """
pet = Pet("fido", "dog", True, Gender.Male)
pet.create()
self.assertNotEqual(pet.id, None)
# Change it an update it
pet.category = "k9"
pet.update()
# Fetch it back and make sure the id hasn't changed
# but the data did change
pets = Pet.all()
self.assertEqual(len(pets), 1)
self.assertEqual(pets[0].category, "k9")
self.assertEqual(pets[0].name, "fido")
