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