Merge: 27988: Make sure iter_attachments does not mutate the payload list.
This commit is contained in:
commit
91afe7ec9f
|
@ -1022,7 +1022,7 @@ class MIMEPart(Message):
|
|||
maintype, subtype = self.get_content_type().split('/')
|
||||
if maintype != 'multipart' or subtype == 'alternative':
|
||||
return
|
||||
parts = self.get_payload()
|
||||
parts = self.get_payload().copy()
|
||||
if maintype == 'multipart' and subtype == 'related':
|
||||
# For related, we treat everything but the root as an attachment.
|
||||
# The root may be indicated by 'start'; if there's no start or we
|
||||
|
|
|
@ -732,6 +732,16 @@ class TestEmailMessageBase:
|
|||
m.set_param('filename', 'abc.png', 'Content-Disposition')
|
||||
self.assertTrue(m.is_attachment())
|
||||
|
||||
def test_iter_attachments_mutation(self):
|
||||
# We had a bug where iter_attachments was mutating the list.
|
||||
m = self._make_message()
|
||||
m.set_content('arbitrary text as main part')
|
||||
m.add_related('more text as a related part')
|
||||
m.add_related('yet more text as a second "attachment"')
|
||||
orig = m.get_payload().copy()
|
||||
self.assertEqual(len(list(m.iter_attachments())), 2)
|
||||
self.assertEqual(m.get_payload(), orig)
|
||||
|
||||
|
||||
class TestEmailMessage(TestEmailMessageBase, TestEmailBase):
|
||||
message = EmailMessage
|
||||
|
|
Loading…
Reference in New Issue