def test_add_sidedrink(): order = Order(1) ing = WeightIngredient(2, 'fries', 's', 2000, 60, 100, 140, 2, 2.8, 3.4) order.add_sideDrink(ing, 2, 'l') assert order.sideDrinks[0][0] == ing assert order.sideDrinks[0][1] == 2 assert order.sideDrinks[0][2] == 'l'
def test_del_sideDrinks_incorrect_input(): order = Order(1) ing = WeightIngredient(2, 'fries', 's', 2000, 60, 100, 140, 2, 2.8, 3.4) order.add_sideDrink(ing, 2, 'l') ing2 = QuantityIngredient(1, 'can of coke', 'd', 40, 3) with pytest.raises(UserError) as e: order.del_sideDrink(ing2, 1, 's') assert "You don't have any can of coke in your order to remove." in str(e.value)
def test_calc_price(): order = Order(1) main = order.add_main('burger') ing1 = QuantityIngredient(1, 'sesame bun', 'm', 20, 1.00) main.addIngredient(ing1, 2, None) ing2 = WeightIngredient(2, 'fries', 's', 2000, 60, 100, 140, 2, 2.8, 3.4) order.add_sideDrink(ing2, 1, None) assert order.calc_price() == 5.0
def test_del_sideDrinks_correct_input(): order = Order(1) ing = WeightIngredient(2, 'fries', 's', 2000, 60, 100, 140, 2, 2.8, 3.4) order.add_sideDrink(ing, 2, 'l') order.del_sideDrink(ing, 2, 'l') assert not order.sideDrinks