From af30e1b619701cc093ce6170246eddb53bc6dd51 Mon Sep 17 00:00:00 2001 From: raphaelauv Date: Tue, 6 Feb 2024 13:25:12 +0100 Subject: [PATCH] feature: S3ToRedshiftOperator templating aws_conn_id (#37195) * feature: S3ToRedshiftOperator templating aws_conn_id --------- Co-authored-by: raphaelauv --- airflow/providers/amazon/aws/transfers/s3_to_redshift.py | 1 + tests/providers/amazon/aws/transfers/test_s3_to_redshift.py | 1 + 2 files changed, 2 insertions(+) diff --git a/airflow/providers/amazon/aws/transfers/s3_to_redshift.py b/airflow/providers/amazon/aws/transfers/s3_to_redshift.py index ada12c214b796..1141d6e69daad 100644 --- a/airflow/providers/amazon/aws/transfers/s3_to_redshift.py +++ b/airflow/providers/amazon/aws/transfers/s3_to_redshift.py @@ -77,6 +77,7 @@ class S3ToRedshiftOperator(BaseOperator): "copy_options", "redshift_conn_id", "method", + "aws_conn_id", ) template_ext: Sequence[str] = () ui_color = "#99e699" diff --git a/tests/providers/amazon/aws/transfers/test_s3_to_redshift.py b/tests/providers/amazon/aws/transfers/test_s3_to_redshift.py index dff441f0a6e16..6e3cbb2a1ca56 100644 --- a/tests/providers/amazon/aws/transfers/test_s3_to_redshift.py +++ b/tests/providers/amazon/aws/transfers/test_s3_to_redshift.py @@ -391,6 +391,7 @@ def test_template_fields_overrides(self): "copy_options", "redshift_conn_id", "method", + "aws_conn_id", ) def test_execute_unavailable_method(self):