def test_add_note_http_error(self, mock_post):
mock_post.side_effect = HTTPError(self.error_msg)
with self.assertRaises(ApplicationError) as context:
verification_api = VerificationAPI()
verification_api.add_note(self.item_id, self.staff_id,
self.note_text)
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_add_note_timeout_error(self, mock_post):
mock_post.side_effect = Timeout(self.error_msg)
with self.assertRaises(ApplicationError) as context:
verification_api = VerificationAPI()
verification_api.add_note(self.item_id, self.staff_id,
self.note_text)
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 add_note():
try:
session['search_items'] = None
staff_id = _get_user_name()
item_id = request.form['item_id']
note_text = request.form['note_text']
current_app.logger.info(
'Adding note "{}" to notepad for worklist item {}...'.format(
note_text, item_id))
verification_api = VerificationAPI()
verification_api.add_note(item_id, staff_id, note_text)
except ApplicationError:
raise ApplicationError(
'Something went wrong when adding a note. '
'Please raise an incident quoting the following id: {}'.format(
g.trace_id))
else:
current_app.logger.info('Note was added to the notepad')
flash('Your note was added to the notepad')
return redirect(url_for('verification.get_item', item_id=item_id))
def test_add_note(self, mock_post):
response_message = 'Note added for user {}'.format(self.item_id)
mock_post.return_value.json.return_value = {
"message": response_message
}
mock_post.return_value.status_code = 201
verification_api = VerificationAPI()
response = verification_api.add_note(self.item_id, self.staff_id,
self.note_text)
self.assertEqual(response, {"message": response_message})
