def test_find_by_name(self):
""" Find an inventory by Name """
Inventory(name="tools", category="widget1", available=True).save()
Inventory(name="materials", category="widget2", available=False).save()
inventory = Inventory.find_by_name("tools")
self.assertEqual(inventory[0].category, "widget1")
self.assertEqual(inventory[0].name, "tools")
self.assertEqual(inventory[0].available, True)
def list_inventory():
""" Returns all of the Inventory """
app.logger.info('Request for inventory list')
inventory = []
category = request.args.get('category')
name = request.args.get('name')
condition = request.args.get('condition')
count = request.args.get('count')
available = request.args.get('available')
if category:
inventory = Inventory.find_by_category(category)
elif name:
inventory = Inventory.find_by_name(name)
else:
inventory = Inventory.all()
results = [inventory.serialize() for inventory in inventory]
return make_response(jsonify(results), status.HTTP_200_OK)
