def test_add_and_get_dishes(self):
restaurant = restaurant_logic.add('FulinMen', 'restaurant', self.image_key)
category = category_logic.add(restaurant.key, "category")
dish_logic.add_by_category_key(category.key, 'Dish1')
dish_logic.add_by_category_key(category.key, 'Dish2')
dishes = dish_logic.get_all_by_category_key(category.key)
self.assertEqual(2, len(dishes))
"Test adding by restaurant_uid and category name"
dish_logic.add('FulinMen', 'category', 'Dish3')
dish_logic.add('FulinMen', 'category', 'Dish4')
dishes = dish_logic.get_all_by_category_key(category.key)
self.assertEqual(4, len(dishes))
dishes = dish_logic.get_all_by_restaurant_uid('FulinMen')
self.assertEquals(4, len(dishes))
dish_logic.delete_by_id(dishes[0].key.id())
dishes = dish_logic.get_all_by_restaurant_uid('FulinMen')
self.assertEquals(3, len(dishes))
def test_category_not_exist(self):
non_exist_key = ndb.Key('Category', 'id')
try:
dishes = dish_logic.get_all_by_category_key(non_exist_key)
self.fail('Shoud throw CategoryNotExistError')
except CategoryNotExistError:
pass
try:
dish = dish_logic.add_by_category_key(non_exist_key, 'dish name')
self.fail('Shoud throw CategoryNotExistError')
except CategoryNotExistError:
pass
