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)