henry3260 commented on code in PR #71495:
URL: https://github.com/apache/airflow/pull/71495#discussion_r3777814050


##########
providers/amazon/tests/unit/amazon/aws/triggers/test_glue.py:
##########
@@ -171,13 +171,14 @@ async def test_verbose_run_success(self, mock_glue_conn, 
mock_logs_conn):
         assert logs_client.get_log_events.call_count >= 2
 
     @pytest.mark.asyncio
+    @pytest.mark.parametrize("job_run_state", ["FAILED", "TIMEOUT", "STOPPED"])
     @mock.patch.object(AwsLogsHook, "get_async_conn")
     @mock.patch.object(GlueJobHook, "get_async_conn")
-    async def test_verbose_run_job_failed(self, mock_glue_conn, 
mock_logs_conn):
-        """When verbose=True and the job fails, the trigger yields an error 
event."""
+    async def test_verbose_run_job_failed(self, mock_glue_conn, 
mock_logs_conn, job_run_state):

Review Comment:
   ```suggestion
       async def test_verbose_run_failure_states(self, mock_glue_conn, 
mock_logs_conn, job_run_state):
   ```



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to