feature: S3ToRedshiftOperator templating aws_conn_id (apache#37195)

* feature: S3ToRedshiftOperator templating aws_conn_id --------- Co-authored-by: raphaelauv <[email protected]>
lwyszomi · Feb 6, 2024 · af30e1b · af30e1b
1 parent 3e27b4f
commit af30e1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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):