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'))