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