def test_find_by_availability(self):
""" Find a Pet by Availability """
Pet(0, "fido", "dog", False).save()
Pet(0, "kitty", "cat", True).save()
pets = Pet.find_by_availability(True)
self.assertEqual(len(pets), 1)
self.assertEqual(pets[0].name, "kitty")
def list_pets():
""" Returns all of the Pets """
pets = []
category = request.args.get('category')
name = request.args.get('name')
available = request.args.get('available')
if category:
pets = Pet.find_by_category(category)
elif name:
pets = Pet.find_by_name(name)
elif available:
pets = Pet.find_by_availability(available)
else:
pets = Pet.all()
results = [pet.serialize() for pet in pets]
return make_response(jsonify(results), HTTP_200_OK)
