def test_execute_task_raises_exception_raised_when_updating_status_in_engine(self): client = ExternalTaskClient(worker_id=1) task = ExternalTask({"id": "1", "topicName": "my_topic"}) executor = ExternalTaskExecutor(worker_id=1, external_task_client=client) TaskResultStatusInput = collections.namedtuple('TaskResultStatusInput', ['task_status', 'task_action', 'task_status_url', 'error_message']) task_result_tests = [ TaskResultStatusInput("complete", self.task_success_action, client.get_task_complete_url(task.get_task_id()), "cannot update task status to complete"), TaskResultStatusInput("failure", self.task_failure_action, client.get_task_failure_url(task.get_task_id()), "cannot update task status to failure"), TaskResultStatusInput("bpmn_error", self.task_bpmn_error_action, client.get_task_bpmn_error_url(task.get_task_id()), "cannot update task status to BPMN err") ] for task_result_test in task_result_tests: with self.subTest(task_result_test.task_status): responses.add(responses.POST, task_result_test.task_status_url, body=Exception(task_result_test.error_message)) with self.assertRaises(Exception) as exception_ctx: executor.execute_task(task, task_result_test.task_action) self.assertEqual(task_result_test.error_message, str(exception_ctx.exception))
def test_execute_task_raises_exception_if_engine_returns_http_status_other_than_no_content_204(self): client = ExternalTaskClient(worker_id=1) task = ExternalTask({"id": "1", "topicName": "my_topic"}) executor = ExternalTaskExecutor(worker_id=1, external_task_client=client) TaskResultStatusInput = collections.namedtuple('TaskResultStatusInput', ['task_status', 'task_action', 'task_status_url', 'http_status_code', 'expected_error_message']) task_result_tests = [ TaskResultStatusInput("complete", self.task_success_action, client.get_task_complete_url(task.get_task_id()), HTTPStatus.OK, 'Not able to mark complete for task_id=1 for topic=my_topic, worker_id=1'), TaskResultStatusInput("failure", self.task_failure_action, client.get_task_failure_url(task.get_task_id()), HTTPStatus.CREATED, 'Not able to mark failure for task_id=1 for topic=my_topic, worker_id=1'), TaskResultStatusInput("bpmn_error", self.task_bpmn_error_action, client.get_task_bpmn_error_url(task.get_task_id()), HTTPStatus.ACCEPTED, 'Not able to mark BPMN Error for task_id=1 for topic=my_topic, worker_id=1') ] for task_result_test in task_result_tests: with self.subTest(task_result_test.task_status): responses.add(responses.POST, task_result_test.task_status_url, status=task_result_test.http_status_code) with self.assertRaises(Exception) as exception_ctx: executor.execute_task(task, task_result_test.task_action) self.assertEqual(task_result_test.expected_error_message, str(exception_ctx.exception))
def test_task_bpmn_error(self): task = ExternalTask({"id": "1", "topicName": "my_topic"}) expected_task_result = TaskResult.bpmn_error(task, error_code="bpmn_err_code_1", error_message="bpmn error") external_task_client = ExternalTaskClient(worker_id=1) responses.add(responses.POST, external_task_client.get_task_bpmn_error_url(task.get_task_id()), status=HTTPStatus.NO_CONTENT) executor = ExternalTaskExecutor(worker_id=1, external_task_client=external_task_client) actual_task_result = executor.execute_task(task, self.task_bpmn_error_action) self.assertEqual(str(expected_task_result), str(actual_task_result))