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