fix error in mail parsing: not all 'inline' elements ara attachments
All checks were successful
/ test (push) Successful in 2m43s
All checks were successful
/ test (push) Successful in 2m43s
This commit is contained in:
parent
598f758332
commit
3635a55e39
2 changed files with 78 additions and 2 deletions
|
@ -129,8 +129,11 @@ def parse_email_body(raw, log=None):
|
||||||
ctype = part.get_content_type()
|
ctype = part.get_content_type()
|
||||||
cdispo = str(part.get('Content-Disposition'))
|
cdispo = str(part.get('Content-Disposition'))
|
||||||
|
|
||||||
|
|
||||||
|
if ctype == 'multipart/mixed':
|
||||||
|
log.debug("Ignoring Multipart %s %s", ctype, cdispo)
|
||||||
# skip any text/plain (txt) attachments
|
# skip any text/plain (txt) attachments
|
||||||
if ctype == 'text/plain' and 'attachment' not in cdispo:
|
elif ctype == 'text/plain' and 'attachment' not in cdispo:
|
||||||
segment = part.get_payload()
|
segment = part.get_payload()
|
||||||
if not segment:
|
if not segment:
|
||||||
continue
|
continue
|
||||||
|
@ -209,7 +212,8 @@ def receive_email(envelope, log=None):
|
||||||
|
|
||||||
email = Email.objects.create(
|
email = Email.objects.create(
|
||||||
sender=sender, recipient=recipient, body=body, subject=subject, reference=header_message_id,
|
sender=sender, recipient=recipient, body=body, subject=subject, reference=header_message_id,
|
||||||
in_reply_to=header_in_reply_to, raw_file=ContentFile(envelope.content, name=random_filename), event=target_event,
|
in_reply_to=header_in_reply_to, raw_file=ContentFile(envelope.content, name=random_filename),
|
||||||
|
event=target_event,
|
||||||
issue_thread=active_issue_thread)
|
issue_thread=active_issue_thread)
|
||||||
for attachment in attachments:
|
for attachment in attachments:
|
||||||
email.attachments.add(attachment)
|
email.attachments.add(attachment)
|
||||||
|
|
|
@ -939,3 +939,75 @@ hello =C3=A4=C3=B6=C3=BC'''
|
||||||
states = StateChange.objects.filter(issue_thread=IssueThread.objects.all()[0])
|
states = StateChange.objects.filter(issue_thread=IssueThread.objects.all()[0])
|
||||||
self.assertEqual(1, len(states))
|
self.assertEqual(1, len(states))
|
||||||
self.assertEqual('pending_new', states[0].state)
|
self.assertEqual('pending_new', states[0].state)
|
||||||
|
|
||||||
|
def test_text_with_attachment2(self):
|
||||||
|
from aiosmtpd.smtp import Envelope
|
||||||
|
from asgiref.sync import async_to_sync
|
||||||
|
import aiosmtplib
|
||||||
|
aiosmtplib.send = make_mocked_coro()
|
||||||
|
handler = LMTPHandler()
|
||||||
|
server = mock.Mock()
|
||||||
|
session = mock.Mock()
|
||||||
|
envelope = Envelope()
|
||||||
|
envelope.mail_from = 'test1@test'
|
||||||
|
envelope.rcpt_tos = ['test2@test']
|
||||||
|
envelope.content = b'''Subject: test
|
||||||
|
From: test1@test
|
||||||
|
To: test2@test
|
||||||
|
Message-ID: <1@test>
|
||||||
|
Content-Type: multipart/mixed; boundary="abc"
|
||||||
|
Content-Disposition: inline
|
||||||
|
Content-Transfer-Encoding: 8bit
|
||||||
|
|
||||||
|
--abc
|
||||||
|
Content-Type: text/plain; charset=utf-8
|
||||||
|
Content-Disposition: inline
|
||||||
|
Content-Transfer-Encoding: 8bit
|
||||||
|
|
||||||
|
test1
|
||||||
|
|
||||||
|
--abc
|
||||||
|
Content-Type: image/jpeg; name="test.jpg"
|
||||||
|
Content-Disposition: attachment; filename="test.jpg"
|
||||||
|
Content-Transfer-Encoding: base64
|
||||||
|
Content-ID: <1>
|
||||||
|
X-Attachment-Id: 1
|
||||||
|
|
||||||
|
dGVzdGltYWdl
|
||||||
|
|
||||||
|
--abc--'''
|
||||||
|
|
||||||
|
result = async_to_sync(handler.handle_DATA)(server, session, envelope)
|
||||||
|
self.assertEqual(result, '250 Message accepted for delivery')
|
||||||
|
self.assertEqual(len(Email.objects.all()), 2)
|
||||||
|
self.assertEqual(len(IssueThread.objects.all()), 1)
|
||||||
|
aiosmtplib.send.assert_called_once()
|
||||||
|
self.assertEqual('test', Email.objects.all()[0].subject)
|
||||||
|
self.assertEqual('test1@test', Email.objects.all()[0].sender)
|
||||||
|
self.assertEqual('test2@test', Email.objects.all()[0].recipient)
|
||||||
|
self.assertEqual('test1\n', Email.objects.all()[0].body)
|
||||||
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
||||||
|
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
||||||
|
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
||||||
|
self.assertEqual(expected_auto_reply_subject.format('test', IssueThread.objects.all()[0].short_uuid()),
|
||||||
|
Email.objects.all()[1].subject)
|
||||||
|
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
||||||
|
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
||||||
|
self.assertEqual(expected_auto_reply.format(IssueThread.objects.all()[0].short_uuid()),
|
||||||
|
Email.objects.all()[1].body)
|
||||||
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
||||||
|
self.assertTrue(Email.objects.all()[1].reference.startswith("<"))
|
||||||
|
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
||||||
|
self.assertEqual("<1@test>", Email.objects.all()[1].in_reply_to)
|
||||||
|
self.assertEqual('test', IssueThread.objects.all()[0].name)
|
||||||
|
self.assertEqual('pending_new', IssueThread.objects.all()[0].state)
|
||||||
|
self.assertEqual(None, IssueThread.objects.all()[0].assigned_to)
|
||||||
|
states = StateChange.objects.filter(issue_thread=IssueThread.objects.all()[0])
|
||||||
|
self.assertEqual(1, len(states))
|
||||||
|
self.assertEqual('pending_new', states[0].state)
|
||||||
|
self.assertEqual(1, len(EmailAttachment.objects.all()))
|
||||||
|
self.assertEqual(1, EmailAttachment.objects.all()[0].id)
|
||||||
|
self.assertEqual('image/jpeg', EmailAttachment.objects.all()[0].mime_type)
|
||||||
|
self.assertEqual('test.jpg', EmailAttachment.objects.all()[0].name)
|
||||||
|
file_content = EmailAttachment.objects.all()[0].file.read()
|
||||||
|
self.assertEqual(b'testimage', file_content)
|
Loading…
Add table
Reference in a new issue