diff options
| author | Jake Howard <git@theorangeone.net> | 2024-06-09 09:09:07 +0100 |
|---|---|---|
| committer | Sarah Boyce <42296566+sarahboyce@users.noreply.github.com> | 2024-06-20 09:43:40 +0200 |
| commit | aba0e541caaa086f183197eaaca0ac20a730bbe4 (patch) | |
| tree | 7178521b84f461c95903cbbb5c03b76b9b39a86f /tests/mail | |
| parent | 9691a00d5839e6137a2716526277013af9ee97ff (diff) | |
Fixed #35537 -- Changed EmailMessage.attachments and EmailMultiAlternatives.alternatives to use namedtuples.
This makes it more descriptive to pull out the named fields.
Diffstat (limited to 'tests/mail')
| -rw-r--r-- | tests/mail/tests.py | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/tests/mail/tests.py b/tests/mail/tests.py index 4db752df59..3746ede338 100644 --- a/tests/mail/tests.py +++ b/tests/mail/tests.py @@ -550,6 +550,18 @@ class MailTests(HeadersCheckMixin, SimpleTestCase): msg.attach("example.txt", "Text file content", "text/plain") self.assertIn(html_content, msg.message().as_string()) + def test_alternatives(self): + msg = EmailMultiAlternatives() + html_content = "<p>This is <strong>html</strong></p>" + mime_type = "text/html" + msg.attach_alternative(html_content, mime_type) + + self.assertEqual(msg.alternatives[0][0], html_content) + self.assertEqual(msg.alternatives[0].content, html_content) + + self.assertEqual(msg.alternatives[0][1], mime_type) + self.assertEqual(msg.alternatives[0].mimetype, mime_type) + def test_none_body(self): msg = EmailMessage("subject", None, "from@example.com", ["to@example.com"]) self.assertEqual(msg.body, "") @@ -626,6 +638,22 @@ class MailTests(HeadersCheckMixin, SimpleTestCase): ) def test_attachments(self): + msg = EmailMessage() + file_name = "example.txt" + file_content = "Text file content" + mime_type = "text/plain" + msg.attach(file_name, file_content, mime_type) + + self.assertEqual(msg.attachments[0][0], file_name) + self.assertEqual(msg.attachments[0].filename, file_name) + + self.assertEqual(msg.attachments[0][1], file_content) + self.assertEqual(msg.attachments[0].content, file_content) + + self.assertEqual(msg.attachments[0][2], mime_type) + self.assertEqual(msg.attachments[0].mimetype, mime_type) + + def test_decoded_attachments(self): """Regression test for #9367""" headers = {"Date": "Fri, 09 Nov 2001 01:08:47 -0000", "Message-ID": "foo"} subject, from_email, to = "hello", "from@example.com", "to@example.com" @@ -645,14 +673,14 @@ class MailTests(HeadersCheckMixin, SimpleTestCase): self.assertEqual(payload[0].get_content_type(), "multipart/alternative") self.assertEqual(payload[1].get_content_type(), "application/pdf") - def test_attachments_two_tuple(self): + def test_decoded_attachments_two_tuple(self): msg = EmailMessage(attachments=[("filename1", "content1")]) filename, content, mimetype = self.get_decoded_attachments(msg)[0] self.assertEqual(filename, "filename1") self.assertEqual(content, b"content1") self.assertEqual(mimetype, "application/octet-stream") - def test_attachments_MIMEText(self): + def test_decoded_attachments_MIMEText(self): txt = MIMEText("content1") msg = EmailMessage(attachments=[txt]) payload = msg.message().get_payload() |
