From ebf239cd283553938585b809e1a2e6e7e1aaf502 Mon Sep 17 00:00:00 2001 From: vincbeck Date: Mon, 29 Jul 2024 15:56:21 -0400 Subject: [PATCH] Add test --- .../aws/operators/test_emr_serverless.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/providers/amazon/aws/operators/test_emr_serverless.py b/tests/providers/amazon/aws/operators/test_emr_serverless.py index 4804f2286993c..12c5cc938018e 100644 --- a/tests/providers/amazon/aws/operators/test_emr_serverless.py +++ b/tests/providers/amazon/aws/operators/test_emr_serverless.py @@ -836,6 +836,27 @@ def test_start_job_deferrable(self, mock_conn): with pytest.raises(TaskDeferred): operator.execute(self.mock_context) + @mock.patch.object(EmrServerlessHook, "conn") + def test_start_job_deferrable_without_wait_for_completion(self, mock_conn): + mock_conn.get_application.return_value = {"application": {"state": "STARTED"}} + mock_conn.start_job_run.return_value = { + "jobRunId": job_run_id, + "ResponseMetadata": {"HTTPStatusCode": 200}, + } + operator = EmrServerlessStartJobOperator( + task_id=task_id, + application_id=application_id, + execution_role_arn=execution_role_arn, + job_driver=job_driver, + configuration_overrides=configuration_overrides, + deferrable=True, + wait_for_completion=False, + ) + + result = operator.execute(self.mock_context) + + assert result == job_run_id + @mock.patch.object(EmrServerlessHook, "get_waiter") @mock.patch.object(EmrServerlessHook, "conn") def test_start_job_deferrable_app_not_started(self, mock_conn, mock_get_waiter):