def test_htmlMessagesAreInjected(self):
msg = MIMEText("<html><body>Test</body></html>",'html')
msg['From'] = '*****@*****.**'
newMessage = SugarMessageProcessor(self.authUser).process(msg,"category")
self.assertEquals('text/html', newMessage.get_content_type())
self.assertTrue("</img>" in newMessage.get_payload())
def test_textMessagesConvertedToMultipart(self):
msg = MIMEText("Test")
msg['From'] = '*****@*****.**'
newMessage = SugarMessageProcessor(self.authUser).process(msg,"category")
self.assertEquals('multipart/alternative', newMessage.get_content_type())
self.assertEquals(2, len(newMessage.get_payload()))
def test_textMessagesAreInjected(self):
msg = MIMEText("Test")
msg['From'] = '*****@*****.**'
newMessage = SugarMessageProcessor(self.authUser).process(msg,"category")
self.assertEquals('multipart/alternative', newMessage.get_content_type())
last_payload = newMessage.get_payload()[1]
self.assertEquals('text/html', last_payload.get_content_type())
self.assertTrue("</img>" in last_payload.get_payload())
