def test_delete_unavailable_promotion(self): """ Delete all unavailable promotions """ unavailable_promo_count = Promotion.find_by_availability(False).count() # initial count of unavailable promotions, make sure it's not empty self.assertNotEqual(unavailable_promo_count, 0) resp = self.app.delete('/promotions/unavailable') self.assertEqual(resp.status_code, status.HTTP_204_NO_CONTENT) self.assertEqual(len(resp.data), 0) new_unavailable_promo_count = Promotion.find_by_availability( False).count() self.assertEqual(new_unavailable_promo_count, 0)
def test_find_by_availability(self): """ Find a Promotion by Availability """ Promotion("A1234", "BOGO", True, "20").save() Promotion("B4321", "dollar", False, "5").save() promotions = Promotion.find_by_availability(True) self.assertEqual(len(promotions), 1) self.assertEqual(promotions[0].productid, "A1234")
def test_find_by_availability(self): """ Find all Promotion goods by availability """ Promotion(promo_name="random", goods_name="random_good", category="random_category", price=20, discount=20, available=True).save() Promotion(promo_name="random2", goods_name="random2_good", category="random2_category", price=2, discount=2, available=False).save() promotions = Promotion.find_by_availability(False) self.assertEqual(promotions[0].category, "random2_category") self.assertEqual(promotions[0].promo_name, "random2") self.assertEqual(promotions[0].goods_name, "random2_good") self.assertEqual(promotions[0].price, 2) self.assertEqual(promotions[0].discount, 2) self.assertEqual(promotions[0].available, False)