def test_get_discount_when_discount_applies(self): order = Order() products = ['VOUCHER', 'TSHIRT', 'VOUCHER'] for product in products: order.add_product(product) discount = TwoForOneDiscount('VOUCHER') assert discount.get_discount(order) == 5.0
def test_get_discount_when_discount_applies(self): order = Order() products = ['TSHIRT', 'TSHIRT', 'TSHIRT', 'VOUCHER', 'TSHIRT'] for product in products: order.add_product(product) discount = BulkPurchaseDiscount('TSHIRT', 3, 1.00) assert discount.get_discount(order) == 4.0
def test_add_product(self): order = Order() order.add_product('MUG') assert 'MUG' in order.product_orders assert order.product_orders.get('MUG').quantity == 1