Compare commits
No commits in common. "9269f2ec48ce6d1e017272a392f367ee8c24024a" and "180493940730ffd910beb320744b37a7bc497bb3" have entirely different histories.
9269f2ec48
...
1804939407
6 changed files with 26 additions and 280 deletions
|
@ -82,13 +82,7 @@ async def send_smtp(message, log):
|
||||||
await aiosmtplib.send(message, hostname="127.0.0.1", port=25, use_tls=False, start_tls=False)
|
await aiosmtplib.send(message, hostname="127.0.0.1", port=25, use_tls=False, start_tls=False)
|
||||||
|
|
||||||
|
|
||||||
def find_active_issue_thread(in_reply_to, address, subject):
|
def find_active_issue_thread(in_reply_to, subject=None):
|
||||||
from re import match
|
|
||||||
uuid_match = match(r'^ticket\+([a-f0-9-]{36})@', address)
|
|
||||||
if uuid_match:
|
|
||||||
issue = IssueThread.objects.filter(uuid=uuid_match.group(1))
|
|
||||||
if issue.exists():
|
|
||||||
return issue.first(), False
|
|
||||||
reply_to = Email.objects.filter(reference=in_reply_to)
|
reply_to = Email.objects.filter(reference=in_reply_to)
|
||||||
if reply_to.exists():
|
if reply_to.exists():
|
||||||
return reply_to.first().issue_thread, False
|
return reply_to.first().issue_thread, False
|
||||||
|
@ -123,8 +117,6 @@ def parse_email_body(raw, log=None):
|
||||||
# skip any text/plain (txt) attachments
|
# skip any text/plain (txt) attachments
|
||||||
if ctype == 'text/plain' and 'attachment' not in cdispo:
|
if ctype == 'text/plain' and 'attachment' not in cdispo:
|
||||||
segment = part.get_payload(decode=True).decode('utf-8')
|
segment = part.get_payload(decode=True).decode('utf-8')
|
||||||
if not segment:
|
|
||||||
continue
|
|
||||||
segment = unescape_and_decode_quoted_printable(segment)
|
segment = unescape_and_decode_quoted_printable(segment)
|
||||||
segment = unescape_and_decode_base64(segment)
|
segment = unescape_and_decode_base64(segment)
|
||||||
log.debug(segment)
|
log.debug(segment)
|
||||||
|
@ -146,9 +138,6 @@ def parse_email_body(raw, log=None):
|
||||||
log.info("Attachment", ctype, cdispo)
|
log.info("Attachment", ctype, cdispo)
|
||||||
else:
|
else:
|
||||||
body = parsed.get_payload(decode=True).decode('utf-8')
|
body = parsed.get_payload(decode=True).decode('utf-8')
|
||||||
body = unescape_and_decode_quoted_printable(body)
|
|
||||||
body = unescape_and_decode_base64(body)
|
|
||||||
log.debug(body)
|
|
||||||
|
|
||||||
return parsed, body, attachments
|
return parsed, body, attachments
|
||||||
|
|
||||||
|
@ -172,16 +161,17 @@ def receive_email(envelope, log=None):
|
||||||
recipient = envelope.rcpt_tos[0].lower() if envelope.rcpt_tos else header_to.lower()
|
recipient = envelope.rcpt_tos[0].lower() if envelope.rcpt_tos else header_to.lower()
|
||||||
sender = envelope.mail_from if envelope.mail_from else header_from
|
sender = envelope.mail_from if envelope.mail_from else header_from
|
||||||
subject = parsed.get('Subject')
|
subject = parsed.get('Subject')
|
||||||
if not subject:
|
|
||||||
subject = "No subject"
|
|
||||||
subject = unescape_and_decode_quoted_printable(subject)
|
subject = unescape_and_decode_quoted_printable(subject)
|
||||||
subject = unescape_and_decode_base64(subject)
|
subject = unescape_and_decode_base64(subject)
|
||||||
target_event = find_target_event(recipient)
|
target_event = find_target_event(recipient)
|
||||||
|
|
||||||
active_issue_thread, new = find_active_issue_thread(header_in_reply_to, recipient, subject)
|
active_issue_thread, new = find_active_issue_thread(header_in_reply_to, subject)
|
||||||
|
body_decoded = body
|
||||||
|
body_decoded = unescape_and_decode_quoted_printable(body_decoded)
|
||||||
|
body_decoded = unescape_and_decode_base64(body_decoded)
|
||||||
|
|
||||||
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_decoded, subject=subject, reference=header_message_id,
|
||||||
in_reply_to=header_in_reply_to, raw=envelope.content.decode('utf-8'), event=target_event,
|
in_reply_to=header_in_reply_to, raw=envelope.content.decode('utf-8'), event=target_event,
|
||||||
issue_thread=active_issue_thread)
|
issue_thread=active_issue_thread)
|
||||||
for attachment in attachments:
|
for attachment in attachments:
|
||||||
|
@ -192,20 +182,9 @@ def receive_email(envelope, log=None):
|
||||||
if new:
|
if new:
|
||||||
# auto reply if new issue
|
# auto reply if new issue
|
||||||
references = collect_references(active_issue_thread)
|
references = collect_references(active_issue_thread)
|
||||||
if not sender.startswith('noreply'):
|
|
||||||
subject = f"Re: {subject} [#{active_issue_thread.short_uuid()}]"
|
|
||||||
body = '''Your request (#{}) has been received and will be reviewed by our lost&found angels.
|
|
||||||
|
|
||||||
We are reviewing incoming requests during the event and teardown. Immediately after the event, expect a delay as the \
|
|
||||||
workload is high. We will not forget about your request and get back in touch once we have updated information on your \
|
|
||||||
request. Requests for devices, wallets, credit cards or similar items will be handled with priority.
|
|
||||||
|
|
||||||
If you happen to find your lost item or just want to add additional information, please reply to this email. Please \
|
|
||||||
do not create a new request.
|
|
||||||
|
|
||||||
Your c3lf (Cloakroom + Lost&Found) Team'''.format(active_issue_thread.short_uuid())
|
|
||||||
reply_email = Email.objects.create(
|
reply_email = Email.objects.create(
|
||||||
sender=recipient, recipient=sender, body=body, subject=subject,
|
sender=recipient, recipient=sender, body="Thank you for your message.", subject="Message received",
|
||||||
in_reply_to=header_message_id, event=target_event, issue_thread=active_issue_thread)
|
in_reply_to=header_message_id, event=target_event, issue_thread=active_issue_thread)
|
||||||
reply = make_reply(reply_email, references)
|
reply = make_reply(reply_email, references)
|
||||||
else:
|
else:
|
||||||
|
@ -244,7 +223,7 @@ class LMTPHandler:
|
||||||
"message": "email received"}
|
"message": "email received"}
|
||||||
)
|
)
|
||||||
log.info(f"Sent message to frontend")
|
log.info(f"Sent message to frontend")
|
||||||
if new and reply:
|
if new:
|
||||||
await send_smtp(reply, log)
|
await send_smtp(reply, log)
|
||||||
log.info("Sent auto reply")
|
log.info("Sent auto reply")
|
||||||
|
|
||||||
|
|
|
@ -12,19 +12,6 @@ from mail.models import Email, EventAddress, EmailAttachment
|
||||||
from mail.protocol import LMTPHandler
|
from mail.protocol import LMTPHandler
|
||||||
from tickets.models import IssueThread, StateChange
|
from tickets.models import IssueThread, StateChange
|
||||||
|
|
||||||
expected_auto_reply_subject = 'Re: {} [#{}]'
|
|
||||||
|
|
||||||
expected_auto_reply = '''Your request (#{}) has been received and will be reviewed by our lost&found angels.
|
|
||||||
|
|
||||||
We are reviewing incoming requests during the event and teardown. Immediately after the event, expect a delay as the \
|
|
||||||
workload is high. We will not forget about your request and get back in touch once we have updated information on your \
|
|
||||||
request. Requests for devices, wallets, credit cards or similar items will be handled with priority.
|
|
||||||
|
|
||||||
If you happen to find your lost item or just want to add additional information, please reply to this email. Please \
|
|
||||||
do not create a new request.
|
|
||||||
|
|
||||||
Your c3lf (Cloakroom + Lost&Found) Team'''
|
|
||||||
|
|
||||||
|
|
||||||
def make_mocked_coro(return_value=mock.sentinel, raise_exception=mock.sentinel):
|
def make_mocked_coro(return_value=mock.sentinel, raise_exception=mock.sentinel):
|
||||||
async def mock_coro(*args, **kwargs):
|
async def mock_coro(*args, **kwargs):
|
||||||
|
@ -106,12 +93,10 @@ class LMTPHandlerTestCase(TestCase): # TODO replace with less hacky test
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
||||||
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
||||||
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
||||||
self.assertEqual(expected_auto_reply_subject.format('test', IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Message received', Email.objects.all()[1].subject)
|
||||||
Email.objects.all()[1].subject)
|
|
||||||
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
||||||
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
||||||
self.assertEqual(expected_auto_reply.format(IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Thank you for your message.', Email.objects.all()[1].body)
|
||||||
Email.objects.all()[1].body)
|
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
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.startswith("<"))
|
||||||
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
||||||
|
@ -347,15 +332,13 @@ class LMTPHandlerTestCase(TestCase): # TODO replace with less hacky test
|
||||||
self.assertEqual(event, Email.objects.all()[0].event)
|
self.assertEqual(event, Email.objects.all()[0].event)
|
||||||
self.assertEqual(event, Email.objects.all()[1].event)
|
self.assertEqual(event, Email.objects.all()[1].event)
|
||||||
self.assertEqual('test', Email.objects.all()[0].subject)
|
self.assertEqual('test', Email.objects.all()[0].subject)
|
||||||
self.assertEqual(expected_auto_reply_subject.format('test', IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Message received', Email.objects.all()[1].subject)
|
||||||
Email.objects.all()[1].subject)
|
|
||||||
self.assertEqual('test1@test', Email.objects.all()[0].sender)
|
self.assertEqual('test1@test', Email.objects.all()[0].sender)
|
||||||
self.assertEqual('test_event@localhost', Email.objects.all()[0].recipient)
|
self.assertEqual('test_event@localhost', Email.objects.all()[0].recipient)
|
||||||
self.assertEqual('test_event@localhost', Email.objects.all()[1].sender)
|
self.assertEqual('test_event@localhost', Email.objects.all()[1].sender)
|
||||||
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
||||||
self.assertEqual('test', Email.objects.all()[0].body)
|
self.assertEqual('test', Email.objects.all()[0].body)
|
||||||
self.assertEqual(expected_auto_reply.format(IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Thank you for your message.', Email.objects.all()[1].body)
|
||||||
Email.objects.all()[1].body)
|
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
||||||
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
||||||
|
@ -421,12 +404,10 @@ test2
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
||||||
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
||||||
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
||||||
self.assertEqual(expected_auto_reply_subject.format('test', IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Message received', Email.objects.all()[1].subject)
|
||||||
Email.objects.all()[1].subject)
|
|
||||||
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
||||||
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
||||||
self.assertEqual(expected_auto_reply.format(IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Thank you for your message.', Email.objects.all()[1].body)
|
||||||
Email.objects.all()[1].body)
|
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
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.startswith("<"))
|
||||||
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
||||||
|
@ -489,12 +470,10 @@ dGVzdGltYWdl
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[0].issue_thread)
|
||||||
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
self.assertEqual('<1@test>', Email.objects.all()[0].reference)
|
||||||
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
self.assertEqual(None, Email.objects.all()[0].in_reply_to)
|
||||||
self.assertEqual(expected_auto_reply_subject.format('test', IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Message received', Email.objects.all()[1].subject)
|
||||||
Email.objects.all()[1].subject)
|
|
||||||
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
self.assertEqual('test2@test', Email.objects.all()[1].sender)
|
||||||
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
self.assertEqual('test1@test', Email.objects.all()[1].recipient)
|
||||||
self.assertEqual(expected_auto_reply.format(IssueThread.objects.all()[0].short_uuid()),
|
self.assertEqual('Thank you for your message.', Email.objects.all()[1].body)
|
||||||
Email.objects.all()[1].body)
|
|
||||||
self.assertEqual(IssueThread.objects.all()[0], Email.objects.all()[1].issue_thread)
|
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.startswith("<"))
|
||||||
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
self.assertTrue(Email.objects.all()[1].reference.endswith("@localhost>"))
|
||||||
|
@ -512,169 +491,3 @@ dGVzdGltYWdl
|
||||||
file_content = EmailAttachment.objects.all()[0].file.read()
|
file_content = EmailAttachment.objects.all()[0].file.read()
|
||||||
self.assertEqual(b'testimage', file_content)
|
self.assertEqual(b'testimage', file_content)
|
||||||
|
|
||||||
def test_mail_noreply(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 = 'noreply@test'
|
|
||||||
envelope.rcpt_tos = ['test2@test']
|
|
||||||
envelope.content = b'Subject: test\nFrom: noreply@test\nTo: test2@test\nMessage-ID: <1@test>\n\ntest'
|
|
||||||
result = async_to_sync(handler.handle_DATA)(server, session, envelope)
|
|
||||||
self.assertEqual(result, '250 Message accepted for delivery')
|
|
||||||
self.assertEqual(len(Email.objects.all()), 1)
|
|
||||||
self.assertEqual(len(IssueThread.objects.all()), 1)
|
|
||||||
aiosmtplib.send.assert_not_called()
|
|
||||||
self.assertEqual('test', Email.objects.all()[0].subject)
|
|
||||||
self.assertEqual('noreply@test', Email.objects.all()[0].sender)
|
|
||||||
self.assertEqual('test2@test', Email.objects.all()[0].recipient)
|
|
||||||
self.assertEqual('test', 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('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)
|
|
||||||
|
|
||||||
def test_mail_empty_subject(self):
|
|
||||||
from aiosmtpd.smtp import Envelope
|
|
||||||
from asgiref.sync import async_to_sync
|
|
||||||
import aiosmtplib
|
|
||||||
import logging
|
|
||||||
logging.disable(logging.CRITICAL)
|
|
||||||
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'From: noreply@test\nTo: test2@test\nMessage-ID: <1@test>\n\ntest'
|
|
||||||
result = async_to_sync(handler.handle_DATA)(server, session, envelope)
|
|
||||||
logging.disable(logging.NOTSET)
|
|
||||||
self.assertEqual('250 Message accepted for delivery', result)
|
|
||||||
self.assertEqual(2, len(Email.objects.all()))
|
|
||||||
self.assertEqual(1, len(IssueThread.objects.all()))
|
|
||||||
aiosmtplib.send.assert_called_once()
|
|
||||||
self.assertEqual('No subject', 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('test', 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('No subject', 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('No subject', 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)
|
|
||||||
|
|
||||||
def test_mail_empty_body(self):
|
|
||||||
from aiosmtpd.smtp import Envelope
|
|
||||||
from asgiref.sync import async_to_sync
|
|
||||||
import aiosmtplib
|
|
||||||
import logging
|
|
||||||
logging.disable(logging.CRITICAL)
|
|
||||||
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\nFrom: <test1@test>\nTo: test2@test\nMessage-ID: <1@test>\n\n'
|
|
||||||
result = async_to_sync(handler.handle_DATA)(server, session, envelope)
|
|
||||||
logging.disable(logging.NOTSET)
|
|
||||||
self.assertEqual('250 Message accepted for delivery', result)
|
|
||||||
self.assertEqual(2, len(Email.objects.all()))
|
|
||||||
self.assertEqual(1, len(IssueThread.objects.all()))
|
|
||||||
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('', 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)
|
|
||||||
|
|
||||||
def test_mail_plus_issue_thread(self):
|
|
||||||
issue_thread = IssueThread.objects.create(
|
|
||||||
name="test",
|
|
||||||
)
|
|
||||||
mail1 = Email.objects.create(
|
|
||||||
subject='test subject',
|
|
||||||
body='test',
|
|
||||||
sender='test1@test',
|
|
||||||
recipient='test2@test',
|
|
||||||
issue_thread=issue_thread,
|
|
||||||
)
|
|
||||||
mail1_reply = Email.objects.create(
|
|
||||||
subject='Message received',
|
|
||||||
body='Thank you for your message.',
|
|
||||||
sender='test2@test',
|
|
||||||
recipient='test1@test',
|
|
||||||
in_reply_to=mail1.reference,
|
|
||||||
issue_thread=issue_thread,
|
|
||||||
)
|
|
||||||
from aiosmtpd.smtp import Envelope
|
|
||||||
from asgiref.sync import async_to_sync
|
|
||||||
import aiosmtplib
|
|
||||||
import logging
|
|
||||||
logging.disable(logging.CRITICAL)
|
|
||||||
aiosmtplib.send = make_mocked_coro()
|
|
||||||
handler = LMTPHandler()
|
|
||||||
server = mock.Mock()
|
|
||||||
session = mock.Mock()
|
|
||||||
envelope = Envelope()
|
|
||||||
envelope.mail_from = '<test1@test>'
|
|
||||||
envelope.rcpt_tos = ['ticket+{}@test'.format(issue_thread.uuid)]
|
|
||||||
envelope.content = (f'Subject: foo\nFrom: <test3@test>\nTo: ticket+{issue_thread.uuid}@test\n'
|
|
||||||
f'Message-ID: <3@test>\n\nbar'.encode('utf-8'))
|
|
||||||
result = async_to_sync(handler.handle_DATA)(server, session, envelope)
|
|
||||||
logging.disable(logging.NOTSET)
|
|
||||||
self.assertEqual('250 Message accepted for delivery', result)
|
|
||||||
self.assertEqual(3, len(Email.objects.all()))
|
|
||||||
self.assertEqual(3, len(Email.objects.filter(issue_thread=issue_thread)))
|
|
||||||
self.assertEqual(1, len(IssueThread.objects.all()))
|
|
||||||
aiosmtplib.send.assert_not_called()
|
|
||||||
self.assertEqual(Email.objects.all()[2].subject, 'foo')
|
|
||||||
self.assertEqual(Email.objects.all()[2].sender, '<test1@test>')
|
|
||||||
self.assertEqual(Email.objects.all()[2].recipient, 'ticket+{}@test'.format(issue_thread.uuid))
|
|
||||||
self.assertEqual(Email.objects.all()[2].body, 'bar')
|
|
||||||
self.assertEqual(Email.objects.all()[2].issue_thread, issue_thread)
|
|
||||||
self.assertEqual(Email.objects.all()[2].reference, '<3@test>')
|
|
||||||
self.assertEqual('test', IssueThread.objects.all()[0].name)
|
|
||||||
|
|
|
@ -22,8 +22,8 @@ class IssueSerializer(serializers.ModelSerializer):
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
model = IssueThread
|
model = IssueThread
|
||||||
fields = ('id', 'timeline', 'name', 'state', 'assigned_to', 'last_activity', 'uuid')
|
fields = ('id', 'timeline', 'name', 'state', 'assigned_to', 'last_activity')
|
||||||
read_only_fields = ('id', 'timeline', 'last_activity', 'uuid')
|
read_only_fields = ('id', 'timeline', 'last_activity')
|
||||||
|
|
||||||
def to_internal_value(self, data):
|
def to_internal_value(self, data):
|
||||||
ret = super().to_internal_value(data)
|
ret = super().to_internal_value(data)
|
||||||
|
|
|
@ -1,32 +0,0 @@
|
||||||
# Generated by Django 4.2.7 on 2024-01-12 21:28
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
|
||||||
|
|
||||||
from tickets.models import IssueThread
|
|
||||||
|
|
||||||
|
|
||||||
class Migration(migrations.Migration):
|
|
||||||
|
|
||||||
dependencies = [
|
|
||||||
('tickets', '0005_remove_issuethread_last_activity'),
|
|
||||||
]
|
|
||||||
|
|
||||||
def set_uuid(apps, schema_editor):
|
|
||||||
import uuid
|
|
||||||
for issue_thread in IssueThread.objects.all():
|
|
||||||
issue_thread.uuid = str(uuid.uuid4())
|
|
||||||
issue_thread.save()
|
|
||||||
|
|
||||||
operations = [
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='issuethread',
|
|
||||||
name='uuid',
|
|
||||||
field=models.CharField(max_length=255, null=True),
|
|
||||||
),
|
|
||||||
migrations.RunPython(set_uuid),
|
|
||||||
migrations.AlterField(
|
|
||||||
model_name='issuethread',
|
|
||||||
name='uuid',
|
|
||||||
field=models.CharField(max_length=255, unique=True, null=False, blank=False),
|
|
||||||
),
|
|
||||||
]
|
|
|
@ -2,7 +2,7 @@ from django.db import models
|
||||||
from django_softdelete.models import SoftDeleteModel
|
from django_softdelete.models import SoftDeleteModel
|
||||||
|
|
||||||
from inventory.models import Event
|
from inventory.models import Event
|
||||||
from django.db.models.signals import post_save, pre_save
|
from django.db.models.signals import post_save
|
||||||
from django.dispatch import receiver
|
from django.dispatch import receiver
|
||||||
|
|
||||||
STATE_CHOICES = (
|
STATE_CHOICES = (
|
||||||
|
@ -25,14 +25,10 @@ STATE_CHOICES = (
|
||||||
|
|
||||||
class IssueThread(SoftDeleteModel):
|
class IssueThread(SoftDeleteModel):
|
||||||
id = models.AutoField(primary_key=True)
|
id = models.AutoField(primary_key=True)
|
||||||
uuid = models.CharField(max_length=255, unique=True, null=False, blank=False)
|
|
||||||
name = models.CharField(max_length=255)
|
name = models.CharField(max_length=255)
|
||||||
assigned_to = models.CharField(max_length=255, null=True)
|
assigned_to = models.CharField(max_length=255, null=True)
|
||||||
manually_created = models.BooleanField(default=False)
|
manually_created = models.BooleanField(default=False)
|
||||||
|
|
||||||
def short_uuid(self):
|
|
||||||
return self.uuid[:8]
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def state(self):
|
def state(self):
|
||||||
try:
|
try:
|
||||||
|
@ -53,13 +49,6 @@ class IssueThread(SoftDeleteModel):
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
@receiver(pre_save, sender=IssueThread)
|
|
||||||
def set_uuid(sender, instance, **kwargs):
|
|
||||||
import uuid
|
|
||||||
if instance.uuid is None or instance.uuid == '':
|
|
||||||
instance.uuid = str(uuid.uuid4())
|
|
||||||
|
|
||||||
|
|
||||||
@receiver(post_save, sender=IssueThread)
|
@receiver(post_save, sender=IssueThread)
|
||||||
def create_issue_thread(sender, instance, created, **kwargs):
|
def create_issue_thread(sender, instance, created, **kwargs):
|
||||||
if created:
|
if created:
|
||||||
|
|
|
@ -51,10 +51,7 @@ class IssueApiTest(TestCase):
|
||||||
comment="test",
|
comment="test",
|
||||||
timestamp=now + timedelta(seconds=3),
|
timestamp=now + timedelta(seconds=3),
|
||||||
)
|
)
|
||||||
self.assertEqual('pending_new', issue.state)
|
|
||||||
self.assertEqual('test issue', issue.name)
|
|
||||||
self.assertEqual(None, issue.assigned_to)
|
|
||||||
self.assertEqual(36, len(issue.uuid))
|
|
||||||
response = self.client.get('/api/2/tickets/')
|
response = self.client.get('/api/2/tickets/')
|
||||||
self.assertEqual(response.status_code, 200)
|
self.assertEqual(response.status_code, 200)
|
||||||
self.assertEqual(len(response.json()), 1)
|
self.assertEqual(len(response.json()), 1)
|
||||||
|
@ -62,7 +59,6 @@ class IssueApiTest(TestCase):
|
||||||
self.assertEqual(response.json()[0]['name'], "test issue")
|
self.assertEqual(response.json()[0]['name'], "test issue")
|
||||||
self.assertEqual(response.json()[0]['state'], "pending_new")
|
self.assertEqual(response.json()[0]['state'], "pending_new")
|
||||||
self.assertEqual(response.json()[0]['assigned_to'], None)
|
self.assertEqual(response.json()[0]['assigned_to'], None)
|
||||||
self.assertEqual(response.json()[0]['uuid'], issue.uuid)
|
|
||||||
self.assertEqual(response.json()[0]['last_activity'], comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
self.assertEqual(response.json()[0]['last_activity'], comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
||||||
self.assertEqual(len(response.json()[0]['timeline']), 4)
|
self.assertEqual(len(response.json()[0]['timeline']), 4)
|
||||||
self.assertEqual(response.json()[0]['timeline'][0]['type'], 'state')
|
self.assertEqual(response.json()[0]['timeline'][0]['type'], 'state')
|
||||||
|
@ -147,6 +143,7 @@ class IssueApiTest(TestCase):
|
||||||
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
||||||
response.json()[2]['timeline'][1]['timestamp'])
|
response.json()[2]['timeline'][1]['timestamp'])
|
||||||
|
|
||||||
|
|
||||||
def test_manual_creation(self):
|
def test_manual_creation(self):
|
||||||
response = self.client.post('/api/2/tickets/manual/',
|
response = self.client.post('/api/2/tickets/manual/',
|
||||||
{'name': 'test issue', 'sender': 'test', 'recipient': 'test', 'body': 'test'},
|
{'name': 'test issue', 'sender': 'test', 'recipient': 'test', 'body': 'test'},
|
||||||
|
|
Loading…
Reference in a new issue