Pass custom headers through in SES email backend (#22667)

GitOrigin-RevId: de84eaf1b042304dd966219da22c7c529afbb662

airflow/providers/amazon/aws/utils/emailer.py

@@ -16,7 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Airflow module for email backend using AWS SES"""
-from typing import List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 from airflow.providers.amazon.aws.hooks.ses import SesHook
 
@@ -32,6 +32,7 @@ def send_email(
     mime_charset: str = 'utf-8',
     conn_id: str = 'aws_default',
     from_email: Optional[str] = None,
+    custom_headers: Optional[Dict[str, Any]] = None,
     **kwargs,
 ) -> None:
     """Email backend for SES."""
@@ -48,4 +49,5 @@ def send_email(
         bcc=bcc,
         mime_subtype=mime_subtype,
         mime_charset=mime_charset,
+        custom_headers=custom_headers,
     )

tests/providers/amazon/aws/utils/test_emailer.py

@@ -31,6 +31,7 @@ def test_send_ses_email(self, mock_hook):
             subject="subject",
             html_content="content",
             from_email="From Test <from@test.com>",
+            custom_headers={"X-Test-Header": "test-val"},
         )
 
         mock_hook.return_value.send_email.assert_called_once_with(
@@ -43,6 +44,7 @@ def test_send_ses_email(self, mock_hook):
             files=None,
             mime_charset="utf-8",
             mime_subtype="mixed",
+            custom_headers={"X-Test-Header": "test-val"},
         )
 
     @mock.patch("airflow.providers.amazon.aws.utils.emailer.SesHook")