def test_get_menu_photos_not_exist(self): """ test about the services restaurant to test the result of get photos of a menu without photos :return: """ new_restaurant = Utils.create_restaurant() new_menu = Utils.create_menu(new_restaurant.id, "Italian food") photos = RestaurantService.get_menu_photos(new_menu.id) assert len(photos) == 0 Utils.delete_menu(new_menu.id) Utils.delete_restaurant(new_restaurant.id)
def test_get_menu_photos_ok(self): """ test about the services restaurant to test the result of get photos of a menu :return: """ new_restaurant = Utils.create_restaurant() new_menu = Utils.create_menu(new_restaurant.id, "Italian food") new_photo1 = Utils.create_menu_photo(new_menu.id, "http://phototest1.com") new_photo2 = Utils.create_menu_photo(new_menu.id, "http://phototest2.com") photos = RestaurantService.get_menu_photos(new_menu.id) assert len(photos) == 2 Utils.delete_menu_photo(new_photo1.id) Utils.delete_menu_photo(new_photo2.id) Utils.delete_menu(new_menu.id) Utils.delete_restaurant(new_restaurant.id)