def test_find_by_name(self):
""" Find Inventory items by name """
Inventory(name="Rolex Watch",
sku="R1232020",
quantity=10,
restockLevel=12).create()
Inventory(name="Cartier Watch",
sku="C1232020",
quantity=12,
restockLevel=6).create()
inv_items = Inventory.find_by_name("Cartier Watch")
self.assertEqual(inv_items[0].sku, "C1232020")
self.assertEqual(inv_items[0].quantity, 12)
self.assertEqual(inv_items[0].restockLevel, 6)
def list_inventory():
""" Returns entire Inventory """
app.logger.info("Request for entire inventory")
inventory = []
sku = request.args.get("sku")
name = request.args.get("name")
if sku:
inventory = Inventory.find_by_sku(sku)
elif name:
inventory = Inventory.find_by_name(name)
else:
inventory = Inventory.all()
results = [inv.serialize() for inv in inventory]
return make_response(jsonify(results), status.HTTP_200_OK)
