def test_create_menu_when_menu_exists(self, mock_get_location,
mock_request_params,
mock_get_unpaginated):
'''Test the response returned when menu already exists.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_request_params.return_value = (None, None, None, None, None,
None, None, None)
mock_get_unpaginated.return_value = [
{
'date': datetime.now(),
'meal_period': 'lunch',
'location_id': 1,
'main_meal_id': 1,
'allowed_side': 1,
'allowed_protein': 1,
'side_items': '',
'protein_items': ''
},
]
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.create_menu()
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == "You can't create multiple " \
"menus with same main item on the same day"
def test_list_menus_range_ok_response(
self, mock_menu_controller_pagination_meta,
mock_meal_item_repo_get,
mock_menu_repo_get_range_paginated_options,
mock_meal_periods_has_value, mock_auth_get_location):
'''Test list_menus_range OK response.
'''
# Arrange
with self.app.app_context():
mock_auth_get_location.return_value = 1
mock_meal_periods_has_value.return_value = True
mock_menu = Menu(date=datetime.now(),
meal_period='',
location_id=1,
main_meal_id=1,
allowed_side=1,
allowed_protein=1,
side_items='1,2',
protein_items='1,2',
vendor_engagement_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_meal_item = MealItem(id=1,
meal_type=1,
name='',
image='',
location_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_menu_repo_get_range_paginated_options.return_value.items = [
mock_menu,
]
mock_meal_item_repo_get.return_value = mock_meal_item
mock_period_start_date = '2019-01-01'
mock_period_end_date = '2019-02-28'
mock_menu_controller_pagination_meta.return_value = {
'total_rows': 1,
'total_pages': 1,
'current_page': 1,
'next_page': False,
'prev_page': False
}
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus_range('lunch',
mock_period_start_date,
mock_period_end_date)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_delete_menu_not_found(self, mock_menu_repo_get):
'''Test response on delete_menu when menu does not exist.
'''
# Arrange
with self.app.app_context():
mock_menu_repo_get.return_value = None
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.delete_menu(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Invalid or incorrect ' \
'menu_id provided'
def test_delete_menu_already_deleted(self, mock_menu_repo_get):
'''Test response on delete_menu when menu is already
deleted.
'''
# Arrange
with self.app.app_context():
mock_menu_repo_get.return_value = Menu(is_deleted=True)
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.delete_menu(1)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Menu has already been deleted'
def test_delete_menu_successful(self, mock_menu_repo_get,
mock_meal_repo_update):
'''Test delete_menu success.
'''
with self.app.app_context():
mock_menu_repo_get.return_value = Menu(is_deleted=False)
mock_meal_repo_update.return_value = Mock()
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.delete_menu(1)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'Menu deleted'
assert result.get_json()['payload']['status'] == 'success'
def test_list_menus_invalid_meal_period_date(self, mock_get_location,
mock_meal_periods_has_value):
'''Test list_menus response when the meal period or date is invalid.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_meal_periods_has_value.return_value = False
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus('', '')
# Assert
assert result.status_code == 404
assert result.get_json()['msg'] == 'Provide valid meal period ' \
'and date'
def test_list_menus_range_admin_period_doesnot_exist(
self, mock_meal_periods_has_value, mock_auth_get_location):
'''Test list_menus_range_admin when period doesn't exist.
'''
# Arrange
with self.app.app_context():
mock_auth_get_location.return_value = 1
mock_meal_periods_has_value.return_value = False
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus_range_admin(
'lunch', datetime.now(), (datetime.now() + timedelta(days=10)))
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Provide valid meal period ' \
'and date range'
def test_list_menus_valid_meal_period_date(self,
mock_menu_repo_get_meal_items,
mock_meal_repo_get,
mock_menu_repo_get_unpaginated,
mock_auth_get_location,
mock_meal_periods_has_value):
'''Test list_menus response when the meal period or date is valid.
'''
# Arrange
with self.app.app_context():
mock_menu = Menu(date=datetime.now(),
meal_period='',
location_id=1,
main_meal_id=1,
allowed_side=1,
allowed_protein=1,
side_items='',
protein_items='',
vendor_engagement_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_meal_item = MealItem(id=1,
meal_type=1,
name='',
image='',
location_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_meal_periods_has_value.return_value = True
mock_auth_get_location.return_value = 1
mock_menu_repo_get_unpaginated.return_value = [
mock_menu,
]
mock_meal_repo_get.return_value = mock_meal_item
mock_menu_repo_get_meal_items.return_value = []
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus('lunch', '2019-02-01')
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_update_menu_ok_response(self, mock_meal_item_repo_get,
mock_menu_repo_update, mock_menu_repo_get,
mock_menu_controller_request_params):
'''Test update_menu when the response is OK.
'''
# Arrange
with self.app.app_context():
mock_menu_controller_request_params.return_value = ('2019-01-01',
'lunch', 1, 1,
1, '1,1',
'1,2', 1)
mock_menu = Menu(date=datetime.now(),
meal_period='',
location_id=1,
main_meal_id=1,
allowed_side=1,
allowed_protein=1,
side_items='1,2',
protein_items='1,2',
vendor_engagement_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_meal_item = MealItem(id=1,
meal_type=1,
name='',
image='',
location_id=1,
created_at=datetime.now(),
updated_at=datetime.now())
mock_menu_repo_get.return_value = mock_menu
mock_menu_repo_update.return_value = mock_menu
mock_meal_item_repo_get.return_value = mock_meal_item
mock_menu_id = 1
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.update_menu(mock_menu_id)
# Assert
assert result.status_code == 200
assert result.get_json()['msg'] == 'OK'
def test_update_menu_when_menu_doesnot_exist(
self, mock_menu_repo_get, mock_menu_controller_request_params):
'''Test update_menu when the menu does not exist.
'''
# Arrange
with self.app.app_context():
mock_menu_controller_request_params.return_value = (None, None,
None, None,
None, None,
None, None)
mock_menu_repo_get.return_value = None
mock_menu_id = 1
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.update_menu(mock_menu_id)
# Assert
assert result.status_code == 404
assert result.get_json()['msg'] == 'This menu_id ' \
'does not exist'
def test_list_menus_range_admin_start_date_gte_end_date(
self, mock_meal_periods_has_value, mock_auth_get_location):
'''Test list_menus_range_admin period start date is gte
period end date.
'''
# Arrange
with self.app.app_context():
mock_auth_get_location.return_value = 1
mock_meal_periods_has_value.return_value = True
mock_period_start_date = '2019-02-04'
mock_period_end_date = '2019-02-01'
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus_range_admin(
'lunch', mock_period_start_date, mock_period_end_date)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Provide valid date range.' \
' start_date cannot be greater than end_date'
def test_update_menu_when_menu_is_deleted(
self, mock_menu_repo_get, mock_menu_controller_request_params):
'''Test update_menu when the menu is deleted.
'''
# Arrange
with self.app.app_context():
mock_deleted_menu = Menu(is_deleted=True)
mock_menu_repo_get.return_value = mock_deleted_menu
mock_menu_controller_request_params.return_value = (None, None,
None, None,
None, None,
None, None)
mock_menu_id = 1
menu_controller = MenuController(mock_menu_id)
# Act
result = menu_controller.update_menu(mock_menu_id)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'This menu is ' \
'already deleted'
def test_create_menu_successful(self, mock_get_location,
mock_request_params, mock_get_unpaginated,
mock_new_menu, mock_get,
mock_get_meal_items):
'''Test response returned when the menu is created successfully.
'''
# Arrange
with self.app.app_context():
mock_get_location.return_value = 1
mock_request_params.return_value = (None, None, None, None, None,
None, None, None)
mock_get_unpaginated.return_value = None
mock_new_menu.return_value = MealItem(meal_type='',
name='',
image='',
location_id=0,
created_at=datetime.now(),
updated_at=datetime.now(),
is_deleted=False,
id=1,
location={})
mock_get.return_value = MealItem(meal_type='',
name='',
image='',
location_id=0,
created_at=datetime.now(),
updated_at=datetime.now(),
is_deleted=False,
id=1)
mock_get_meal_items.return_value = []
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.create_menu()
# Assert
assert result.status_code == 201
assert result.get_json()['msg'] == "OK"
def test_list_menus_range_start_date_gte_end_date(
self, mock_meal_periods_has_value, mock_auth_get_location):
'''Test list_menu_ranges response when the start date
is gte end date.
'''
# Arrange
with self.app.app_context():
mock_meal_periods_has_value.return_value = True
mock_auth_get_location.return_value = 1
mock_meal_period = 'lunch'
mock_period_start = '2019-01-28'
mock_period_end = '2019-01-01'
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus_range(mock_meal_period,
mock_period_start,
mock_period_end)
# Assert
assert result.status_code == 400
assert result.get_json()['msg'] == 'Provide valid date ' \
'range. start_date cannot be greater than end_date'
def test_list_menus_ranges_invalid_meal_period(self,
mock_meal_periods_has_value,
mock_auth_get_location):
'''Test list_menu_ranges response when the meal period is invalid.
'''
# Arrange
with self.app.app_context():
mock_auth_get_location.return_value = 1
mock_meal_periods_has_value.return_value = False
mock_meal_period = 'lunch'
mock_period_start = '2019-01-01'
mock_period_end = '2019-02-28'
menu_controller = MenuController(self.request_context)
# Act
result = menu_controller.list_menus_range(mock_meal_period,
mock_period_start,
mock_period_end)
# Arrange
assert result.status_code == 400
assert result.get_json()['msg'] == 'Provide valid meal period ' \
'and date range'
'''A module of menu blueprint'''
from app.blueprints.base_blueprint import Blueprint, BaseBlueprint, request, Security, Auth
from app.controllers.menu_controller import MenuController
from flasgger import swag_from
menu_blueprint = Blueprint('menu', __name__, url_prefix='{}/admin/menus'.format(BaseBlueprint.base_url_prefix))
user_menu_blueprint = Blueprint('user_menu', __name__, url_prefix='{}/menus'.format(BaseBlueprint.base_url_prefix))
menu_controller = MenuController(request)
"""Menu"""
@menu_blueprint.route('/', methods=['POST'])
@Security.validator([
'date|required:date', 'mealPeriod|required',
'mainMealId|required:int', 'allowedSide|required:int',
'allowedProtein|required:int', 'sideItems|required:list_int',
'proteinItems|required:list_int', 'vendorEngagementId|required:int',
'sideItems|exists|meal_item|id', 'proteinItems|exists|meal_item|id',
'mainMealId|exists|meal_item|id'
])
@Auth.has_permission('create_menu')
@swag_from('documentation/create_menu.yml')
def create_menu():
'''Blueprint function for creating a menu'''
return menu_controller.create_menu()
@menu_blueprint.route('/<int:menu_id>', methods=['DELETE'])
@Auth.has_permission('delete_menu')
@swag_from('documentation/delete_menu.yml')