def test_message(self):
text = 'This is some text: 高兴'
html = '<h1>Heading</h1>\n<p>This is a paragraph: 高兴</p>'
msg = message(body=text)
self.eq(message(body=text), MIMEText(text.encode('utf-8'), 'plain', 'utf-8'))
self.eq(message(html=html), MIMEText(html.encode('utf-8'), 'html', 'utf-8'))
msg = MIMEMultipart('alternative')
msg.attach(MIMEText(text.encode('utf-8'), 'plain', 'utf-8'))
msg.attach(MIMEText(html.encode('utf-8'), 'html', 'utf-8'))
self.eq(message(body=text, html=html), msg)
def test_images(self):
html = '高兴: <img src="cid:logo">'
img = os.path.join(folder, 'small-image.jpg')
msg = MIMEMultipart('related')
msg.attach(MIMEText(html.encode('utf-8'), 'html', 'utf-8'))
with open(img, 'rb') as handle:
img_part = MIMEImage(handle.read())
img_part.add_header('Content-ID', '<logo>')
msg.attach(img_part)
self.eq(message(html=html, images={'logo': img}), msg)
def check_attachment(self, img):
msg = message(body='text', attachments=[img])
img_part = list(msg.walk())[-1]
eq_(img_part.get_content_type(), 'image/jpeg')
if 'filename' in img:
eq_(img_part.get_filename(), os.path.basename(img['filename']))