fix static checks

apache · Jul 24, 2024 · 1740692 · 1740692
1 parent d450367
commit 1740692
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/airflow/providers/amazon/aws/hooks/step_function.py b/airflow/providers/amazon/aws/hooks/step_function.py
@@ -18,7 +18,7 @@
 
 import json
 
-from airflow.exceptions import AirflowException
+from airflow.exceptions import AirflowFailException
 from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
 
 
@@ -61,7 +61,7 @@ def start_execution(
         """
         if is_redrive_execution:
             if not name:
-                raise AirflowException(
+                raise AirflowFailException(
                     "Execution name is required to start RedriveExecution for %s.", state_machine_arn
                 )
             elements = state_machine_arn.split(":stateMachine:")

diff --git a/tests/providers/amazon/aws/hooks/test_step_function.py b/tests/providers/amazon/aws/hooks/test_step_function.py
@@ -23,7 +23,7 @@
 import pytest
 from moto import mock_aws
 
-from airflow.exceptions import AirflowException
+from airflow.exceptions import AirflowFailException
 from airflow.providers.amazon.aws.hooks.step_function import StepFunctionHook
 
 
@@ -64,7 +64,9 @@ def test_redrive_execution(self, mock_conn):
     def test_redrive_execution_without_name_should_fail(self, mock_conn):
         mock_conn.redrive_execution.return_value = {"redriveDate": datetime(2024, 1, 1)}
 
-        with pytest.raises(AirflowException, match="Execution name is required to start RedriveExecution"):
+        with pytest.raises(
+            AirflowFailException, match="Execution name is required to start RedriveExecution"
+        ):
             StepFunctionHook().start_execution(
                 state_machine_arn="arn:aws:states:us-east-1:123456789012:stateMachine:test-state-machine",
                 is_redrive_execution=True,