def test_sent_message_same_as_original(self):
"""
Tests that an ``email.message.Message`` instance sent by using the
``CustomEmailMessage`` class is the same as the original message.
"""
msg = self.create_multipart()
custom_message = CustomEmailMessage(msg=msg, to=['recipient'])
custom_message.send()
self.assertEqual(msg.as_string(), mail.outbox[0].message().as_string())
def test_attachment_included(self):
"""
Tests that an attachment included in the ``CustomEmailMessage``
instance is sent with the rest of the message.
"""
msg = self.create_multipart()
attachment = self.prepare_part(b'new_data')
msg.attach(attachment)
custom_message = CustomEmailMessage(msg=msg, to=['recipient'])
custom_message.send()
self.assertIn(attachment, mail.outbox[0].message().get_payload())