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)