def test_delete_a_product(self): """ Delete a Product """ product = Product( name="Wagyu Tenderloin Steak", category="food", stock=11, price=20.56, description="The most decadent, succulent cut of beef, ever.") product.save() self.assertEqual(len(Product.all()), 1) # delete the product and make sure it isn't in the database product.delete() self.assertEqual(len(Product.all()), 0)
def test_add_a_product(self): """ Create a product and add it to the database """ products = Product.all() self.assertEqual(products, []) product = Product( name="Wagyu Tenderloin Steak", category="food", stock=11, price=20.56, description="The most decadent, succulent cut of beef, ever.") self.assertTrue(product != None) self.assertEqual(product.id, None) self.assertEqual(product.name, "Wagyu Tenderloin Steak") self.assertEqual(product.category, "food") self.assertAlmostEqual(product.price, Decimal(20.56)) self.assertEqual(product.stock, 11) product.save() # Asert that it was assigned an id and shows up in the database self.assertEqual(product.id, 1) products = Product.all() self.assertEqual(len(products), 1)
def test_update_a_product(self): """ Update a Product """ product = Product( name="Wagyu Tenderloin Steak", category="food", stock=11, price=20.56, description="The most decadent, succulent cut of beef, ever.") product.save() self.assertEqual(product.id, 1) # Change it an save it product.category = "beverage" product.save() self.assertEqual(product.id, 1) # Fetch it back and make sure the id hasn't changed # but the data did change products = Product.all() self.assertEqual(len(products), 1) self.assertEqual(products[0].category, "beverage")
def get(self): """Returns all of the Products""" app.logger.info('Request for product list') products = [] category = request.args.get('category') name = request.args.get('name') price = request.args.get('price') if category: products = Product.find_by_category(category) elif name: products = Product.find_by_name(name) elif price and int(price) > 0 and int( price) < 4: # query price by range if int(price) == 1: products = Product.find_by_price(0, 25) elif int(price) == 2: products = Product.find_by_price(25, 50) else: products = Product.find_by_price(50, 75) else: products = Product.all() results = [product.serialize() for product in products] return results, status.HTTP_200_OK