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(), )