def test_get_worklist_timeout_error(self, mock_get):
mock_get.side_effect = Timeout(self.error_msg)
with self.assertRaises(ApplicationError) as context:
verification_api = VerificationAPI()
verification_api.get_worklist()
self.assertEqual(
context.exception.message,
'Connection to verification_api timed out: {}'.format(
self.error_msg))
self.assertEqual(context.exception.code, 'E403')
self.assertEqual(context.exception.http_code, 500)
def test_get_worklist_connection_error(self, mock_get):
mock_get.side_effect = ConnectionError(self.error_msg)
with self.assertRaises(ApplicationError) as context:
verification_api = VerificationAPI()
verification_api.get_worklist()
self.assertEqual(
context.exception.message,
'Encountered an error connecting to verification_api: {}'.format(
self.error_msg))
self.assertEqual(context.exception.code, 'E402')
self.assertEqual(context.exception.http_code, 500)
def test_get_worklist_http_error(self, mock_get):
mock_get.side_effect = HTTPError(self.error_msg)
with self.assertRaises(ApplicationError) as context:
verification_api = VerificationAPI()
verification_api.get_worklist()
self.assertEqual(
context.exception.message,
'Received the following response from verification_api: {}'.format(
self.error_msg))
self.assertEqual(context.exception.code, 'E401')
self.assertEqual(context.exception.http_code, 500)
def test_get_worklist(self, mock_get):
mock_get.return_value.json.return_value = [{'case_id': 1}]
mock_get.return_value.status_code = 200
verification_api = VerificationAPI()
response = verification_api.get_worklist()
self.assertEqual(response, [{'case_id': 1}])
def test_get_verification_api(self):
with app.app_context() as ac:
ac.g.trace_id = None
ac.g.requests = requests.Session()
with app.test_request_context():
verification_api = VerificationAPI()
resp_data = verification_api.get_worklist()
self.assertIsNotNone(resp_data)
def get_worklist():
current_app.logger.info('User requested to view worklist...')
try:
verification_api = VerificationAPI()
worklist = verification_api.get_worklist()
_get_user_name()
except ApplicationError:
raise ApplicationError(
'Something went wrong when retrieving the worklist. '
'Please raise an incident quoting the following id: {}'.format(
g.trace_id))
else:
current_app.logger.info('Putting worklist into viewable format...')
worklist_items = [build_row(item) for item in worklist]
return render_template('app/worklist.html',
worklist_items=worklist_items)
