def test_find_or_404_404(self):
""" Find promotion or 404 expecting 404 """
Promotion(name="20%OFF", product_id=9527, discount_ratio=80).save()
try:
promotion = Promotion.find_or_404(2)
# Should not reach beyond this line.
self.assertIsNone(promotion)
except:
pass
def test_find_or_404(self):
""" Find promotion or 404 """
Promotion(name="20%OFF", product_id=9527, discount_ratio=80).save()
promotion = Promotion.find_or_404(1)
self.assertIsNotNone(promotion)
self.assertEqual(promotion.promotion_id, 1)
self.assertEqual(promotion.name, "20%OFF")
self.assertEqual(promotion.product_id, 9527)
self.assertEqual(promotion.discount_ratio, 80)
self.assertEqual(promotion.counter, 0)