def test_has_no_menu(self): user = User(username='******', email='*****@*****.**', password='******') user.save() rest = Restaurant(business_name="I love donuts", email='*****@*****.**', address='1122 City Center', city='LV', state='NV', zip_code='89178', phone_number='555-1212', user=user) rest.save() self.assertFalse(rest.has_menu(), "No Menu")
def test_has_menu(self): user = User(username='******', email='*****@*****.**', password='******') user.save() rest = Restaurant(business_name="I love donuts", email='*****@*****.**', address='1122 City Center', city='LV', state='NV', zip_code='89178', phone_number='555-1212', user=user) rest.save() menu = FoodItem(restaurant=rest, name='Sprinkle Donut', price='2.99', description='your favorite sprinkle donut', category='D') menu.save() self.assertTrue(rest.has_menu(), "Yes, there's a Menu")