def setUp(self):
raw_message = read_file(
f"{settings.BASE_DIR}/apps/restricted/fixtures/emails/valid.txt")
parser = EmailParser(raw_message, "*****@*****.**",
ParserMessageType.STRING)
self.message = parser.parse()
self.processor = MessageProcessor("*****@*****.**", self.message, {})
def test_rewrite_message(self):
"""Remove headers that may cause problems. Only keep a few required headers."""
raw_message = read_file(
f'{settings.BASE_DIR}/apps/restricted/fixtures/emails/clean_headers.txt'
)
parser = EmailParser(raw_message, '*****@*****.**',
ParserMessageType.STRING)
message = parser.parse()
new_message = self.processor.rewrite_message(message, '*****@*****.**')
self.assertCountEqual(
['From', 'Subject', 'Content-Type', 'MIME-Version', 'Sender'],
new_message.keys())
def test_send(self):
"""Test the outbox and make sure we have correct headers in the messages."""
raw_message = read_file(
f'{settings.BASE_DIR}/apps/restricted/fixtures/emails/clean_headers.txt'
)
parser = EmailParser(raw_message, '*****@*****.**',
ParserMessageType.STRING)
message = parser.parse()
new_message = self.processor.rewrite_message(message, '*****@*****.**')
messages = self.processor.send(['*****@*****.**', '*****@*****.**'],
'*****@*****.**', new_message)
self.assertEquals(2, messages)
outbox = mail.outbox
first_message = outbox[0].message()
self.assertSequenceEqual([
'Subject', 'Content-Type', 'MIME-Version', 'Sender', 'From', 'To'
], first_message.keys())
def test_send(self):
"""Test the outbox and make sure we have correct headers in the messages."""
raw_message = read_file(
f"{settings.BASE_DIR}/apps/restricted/fixtures/emails/clean_headers.txt"
)
parser = EmailParser(raw_message, "*****@*****.**",
ParserMessageType.STRING)
message = parser.parse()
new_message = self.processor.rewrite_message(message, "*****@*****.**")
messages = self.processor.send(["*****@*****.**", "*****@*****.**"],
"*****@*****.**", new_message)
self.assertEquals(2, messages)
outbox = mail.outbox
first_message = outbox[0].message()
self.assertSequenceEqual(
[
"Subject", "Content-Type", "MIME-Version", "Sender", "From",
"To"
],
first_message.keys(),
)
