Ticket #29830: patch.diff

django/core/mail/message.py

@@ -170 +170 @@
         MIMEText.__setitem__(self, name, val)
 
     def set_payload(self, payload, charset=None):
-        if charset == 'utf-8':
+        if charset == 'utf-8' and not isinstance(charset, Charset.Charset):
             has_long_lines = any(
                 len(l.encode()) > RFC5322_EMAIL_LINE_LENGTH_LIMIT
                 for l in payload.splitlines()

tests/mail/tests.py

@@ -8 +8 @@
 import sys
 import tempfile
 import threading
-from email import message_from_binary_file, message_from_bytes
+from email import message_from_binary_file, message_from_bytes, charset
 from email.header import Header
 from email.mime.text import MIMEText
 from email.utils import parseaddr
@@ -686 +686 @@
         # The child message header is not base64 encoded
         self.assertIn('Child Subject', parent_s)
 
+    def test_respect_body_encoding(self):
+        """Test that a custom body encoding is respected."""
+        body = 'Body with latin characters: àáä.'
+        msg = EmailMessage(
+            'Subject', body, 'bounce@example.com', ['to@example.com'],
+            headers={'From': 'from@example.com'},
+        )
+        encoding = charset.Charset("utf-8")
+        encoding.body_encoding = charset.QP
+        msg.encoding = encoding
+        message = msg.message()
+        self.assertMessageHasHeaders(message, {
+            ('MIME-Version', '1.0'),
+            ('Content-Type', 'text/plain; charset="utf-8"'),
+            ('Content-Transfer-Encoding', 'quoted-printable'),
+        })
+        self.assertEqual(message.get_payload(), encoding.body_encode(body))
+
+        body = 'Hello World'
+        msg = EmailMessage(
+            'Subject', body, 'bounce@example.com', ['to@example.com'],
+            headers={'From': 'from@example.com'},
+        )
+        encoding = charset.Charset("iso-8859-1")
+        encoding.body_encoding = charset.BASE64
+        msg.encoding = encoding
+        message = msg.message()
+        self.assertMessageHasHeaders(message, {
+            ('MIME-Version', '1.0'),
+            ('Content-Type', 'text/plain; charset="iso-8859-1"'),
+            ('Content-Transfer-Encoding', 'base64'),
+        })
+        self.assertEqual(message.get_payload(), encoding.body_encode(body))
+
     def test_sanitize_address(self):
         """
         Email addresses are properly sanitized.