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)