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_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'