This is an automated email from the ASF dual-hosted git repository.
ferruzzi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 86193f5608 Increase the number of attempts in AWS system test
`example_rds_export` (#32976)
86193f5608 is described below
commit 86193f560815507b9abf1008c19b133d95c4da9f
Author: Vincent <[email protected]>
AuthorDate: Mon Jul 31 15:18:57 2023 -0400
Increase the number of attempts in AWS system test `example_rds_export`
(#32976)
---
airflow/providers/amazon/aws/operators/rds.py | 9 ++++++++-
tests/system/providers/amazon/aws/example_rds_export.py | 1 +
2 files changed, 9 insertions(+), 1 deletion(-)
diff --git a/airflow/providers/amazon/aws/operators/rds.py
b/airflow/providers/amazon/aws/operators/rds.py
index 7a544a8c00..a4a2ac5c33 100644
--- a/airflow/providers/amazon/aws/operators/rds.py
+++ b/airflow/providers/amazon/aws/operators/rds.py
@@ -392,6 +392,8 @@ class RdsCancelExportTaskOperator(RdsBaseOperator):
:param export_task_identifier: The identifier of the snapshot export task
to cancel
:param wait_for_completion: If True, waits for DB snapshot export to
cancel. (default: True)
+ :param check_interval: The amount of time in seconds to wait between
attempts
+ :param max_attempts: The maximum number of attempts to be made
"""
template_fields = ("export_task_identifier",)
@@ -402,6 +404,7 @@ class RdsCancelExportTaskOperator(RdsBaseOperator):
export_task_identifier: str,
wait_for_completion: bool = True,
check_interval: int = 30,
+ max_attempts: int = 40,
**kwargs,
):
super().__init__(**kwargs)
@@ -409,6 +412,7 @@ class RdsCancelExportTaskOperator(RdsBaseOperator):
self.export_task_identifier = export_task_identifier
self.wait_for_completion = wait_for_completion
self.check_interval = check_interval
+ self.max_attempts = max_attempts
def execute(self, context: Context) -> str:
self.log.info("Canceling export task %s", self.export_task_identifier)
@@ -419,7 +423,10 @@ class RdsCancelExportTaskOperator(RdsBaseOperator):
if self.wait_for_completion:
self.hook.wait_for_export_task_state(
- self.export_task_identifier, target_state="canceled",
check_interval=self.check_interval
+ self.export_task_identifier,
+ target_state="canceled",
+ check_interval=self.check_interval,
+ max_attempts=self.max_attempts,
)
return json.dumps(cancel_export, default=str)
diff --git a/tests/system/providers/amazon/aws/example_rds_export.py
b/tests/system/providers/amazon/aws/example_rds_export.py
index d103763952..a87bce10de 100644
--- a/tests/system/providers/amazon/aws/example_rds_export.py
+++ b/tests/system/providers/amazon/aws/example_rds_export.py
@@ -128,6 +128,7 @@ with DAG(
)
# [END howto_operator_rds_cancel_export]
cancel_export.check_interval = 10
+ cancel_export.max_attempts = 120
# [START howto_sensor_rds_export_task_existence]
export_sensor = RdsExportTaskExistenceSensor(
