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)