def test_create_meal_ok_response(
self,
mock_get_unpaginated,
mock_request_params,
mock_get_location,
mock_has_value,
mock_new_meal_item
):
'''Test create_meal OK response.
'''
# Arrange
with self.app.app_context():
mock_request_params.return_value = (
'Mock', 'Mock', 'Mock'
)
mock_get_unpaginated.return_value = None
mock_get_location.return_value = 1
mock_has_value.return_value = True
mock_new_meal_item.return_value = self.mock_meal_item
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.create_meal()
# Assert
assert result.status_code == 201
assert result.get_json()['msg'] == 'OK'
def test_list_meals_page_ok_response(
self,
mock_pagination_meta,
mock_filter_by,
mock_get_location
):
'''Test list_meals_page OK response.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_filter_by.return_value.items = [self.mock_meal_item, ]
mock_pagination_meta.return_value = {
'total_rows': 1,
'total_pages': 1,
'current_page': 1,
'next_page': 1,
'prev_page': 1
}
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.list_meals_page(1, 10)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_create_meal_when_meal_type_doesnot_exist(
self,
mock_get_unpaginated,
mock_request_params,
mock_get_location,
mock_has_value
):
'''Test create_meal when the meal type doesnot exists.
'''
# Arrange
with self.app.app_context():
mock_request_params.return_value = (
'Mock', 'Mock', 'Mock'
)
mock_get_unpaginated.return_value = None
mock_get_location.return_value = 1
mock_has_value.return_value = False
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.create_meal()
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Invalid meal type. Must be' \
' main, protein or side'
def test_delete_meal_when_meal_doesnot_exist(
self,
mock_get
):
'''Test delete_meal when meal item doesnot exist.
'''
with self.app.app_context():
mock_get.return_value = None
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.delete_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Invalid or incorrect meal_id' \
' provided'
def test_get_meal_ok_response(
self,
mock_get
):
'''Test get_meal OK response.
'''
# Arrange
with self.app.app_context():
mock_get.return_value = self.mock_meal_item
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.get_meal(1)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_get_meal_when_meal_item_doesnot_exist(
self,
mock_get
):
'''Test get_meal when meal item doesn't exist.
'''
# Arrange
with self.app.app_context():
mock_get.return_value = None
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.get_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Bad Request. This meal id ' \
'does not exist'
def test_delete_meal_when_meal_is_already_deleted(
self,
mock_get
):
'''Test delete_meal when the meal is already deleted.
'''
# Arrange
with self.app.app_context():
mock_get.return_value = self.mock_deleted_meal_item
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.delete_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Bad Request. This meal item' \
' is deleted'
def test_update_meal_when_meal_is_deleted(
self,
mock_get,
mock_request_params
):
'''Test update_meal when the meal is deleted.
'''
# Arrange
with self.app.app_context():
mock_get.return_value = self.mock_deleted_meal_item
mock_request_params.return_value = ('Mock', 'Mock', 'Mock')
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.update_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Bad Request. This meal item ' \
'is deleted'
def test_update_meal_when_meal_doesnot_exist(
self,
mock_get,
mock_request_params
):
'''Test update_meal when the meal doesnot exist.
'''
# Arrange
with self.app.app_context():
mock_request_params.return_value = ('Mock', 'Mock', 'Mock')
mock_get.return_value = None
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.update_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Invalid or incorrect meal_id' \
' provided'
def test_list_meals_ok_response(
self,
mock_get_location,
mock_get_unpaginated,
mock_get_params_dict
):
'''Test list_meals OK response.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_get_unpaginated.return_value = [self.mock_meal_item, ]
mock_get_params_dict.return_value = {}
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.list_meals()
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_update_meal_ok_response(
self,
mock_get_unpaginated,
mock_get,
mock_request_params
):
'''Test update_meal OK response.
'''
# Arrange
with self.app.app_context():
mock_get.return_value = MealItemFactory(id=1)
mock_request_params.return_value = ('mock', 'mock', 'mock')
mock_get_unpaginated.return_value = None
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.update_meal(1)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_update_meal_when_name_already_exists(
self,
mock_get_unpaginated,
mock_get,
mock_request_params
):
'''Test update_meal when meal name already exists.
'''
# Arrange
with self.app.app_context():
mock_request_params.return_value = ('mock', 'mock', 'mock')
mock_get.return_value = self.mock_meal_item
mock_get_unpaginated.return_value = self.mock_meal_item
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.update_meal(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Meal item with this name ' \
'already exists'
def test_create_meal_when_meal_already_exists(
self,
mock_get_unpaginated,
mock_request_params,
mock_get_location
):
'''Test create_meal when the meal already exists.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_request_params.return_value = (
'Mock', 'Mock', 'Mock'
)
mock_get_unpaginated.return_value = self.mock_meal_item
meal_item_controller = MealItemController(self.request_context)
# Act
result = meal_item_controller.create_meal()
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Meal item with this name ' \
'already exists'
from app.blueprints.base_blueprint import Blueprint, BaseBlueprint, Security, request, Auth
from app.controllers.meal_item_controller import MealItemController
from app.models import MealItem
from flasgger import swag_from
url_prefix = '{}/meal-items'.format(BaseBlueprint.base_url_prefix)
meal_item_blueprint = Blueprint('meal_item', __name__, url_prefix=url_prefix)
meal_item_controller = MealItemController(request)
@meal_item_blueprint.route('/', methods=['GET'])
@Auth.has_permission('view_meal_item')
@Security.validate_query_params(MealItem)
@swag_from('documentation/get_all_meal_items.yml')
def list_meals():
return meal_item_controller.list_meals()
# @meal_item_blueprint.route('/page/<int:page_id>', methods=['GET'])
# @swag_from('documentation/get_all_meal_items_page.yml')
# def list_meals_page(page_id):
# meals_per_page = int(request.args.get('per_page')) if request.args.get('per_page') != None else 10
# return meal_item_controller.list_meals_page(page_id, meals_per_page)
@meal_item_blueprint.route('/<int:meal_item_id>', methods=['GET'])
@Auth.has_permission('view_meal_item')
@swag_from('documentation/get_single_meal_item.yml')
def get_meal(meal_item_id):
return meal_item_controller.get_meal(meal_item_id)