def add(valid_user, restaurant):
product = input("Which product do you want to add? ")
product_price = database.valid_product(restaurant, product)
if product_price:
valid_user.get_basket().append(product_price)
cart(valid_user)
else:
print("This product is not on the menu")
def test_not_valid_product(self):
database.create_menu_table('speedy')
database.add('speedy', 'pizza', 3.5)
self.assertFalse(database.valid_product('speedy', 'spaghetti'))
def test_valid_product(self):
database.create_menu_table('speedy')
database.add('speedy', 'pizza', 3.5)
self.assertTrue(database.valid_product('speedy', 'pizza'))
