def test_addr_tuple_to_addr(self):
addr = Envelope()._addr_tuple_to_addr(('*****@*****.**', 'Test'))
assert addr == 'Test <*****@*****.**>'
addr = Envelope(charset='utf-8')._addr_tuple_to_addr((
'*****@*****.**', ))
assert addr == '*****@*****.**'
def test_cc_property(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.cc == envelope._cc
msg.pop('cc_addrs')
envelope = Envelope(**msg)
assert envelope.cc == []
def test_to_addr_property(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.to_addr == envelope._to
msg.pop('to_addr')
envelope = Envelope(**msg)
assert envelope.to_addr == []
def test_bcc_addr_property(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.bcc_addr == envelope._bcc
msg.pop('bcc_addr')
envelope = Envelope(**msg)
assert envelope.bcc_addr == []
def test_add_cc(self):
msg = self._dummy_message()
msg.pop('cc_addrs')
envelope = Envelope(**msg)
envelope.add_cc('*****@*****.**')
assert envelope.cc == ['*****@*****.**']
def test_add_bcc_addr(self):
msg = self._dummy_message()
msg.pop('bcc_addr')
envelope = Envelope(**msg)
envelope.add_bcc_addr('*****@*****.**')
assert envelope.bcc_addr == ['*****@*****.**']
def test_add_header(self):
msg = self._dummy_message()
msg.pop('headers')
envelope = Envelope(**msg)
envelope.add_header('X-Spam', 'eggs')
assert envelope.headers == {'X-Spam': 'eggs'}
def test_to_mime_message_with_data(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
mime_msg = envelope.to_mime_message()
assert mime_msg is not None
assert mime_msg['Subject'] == msg['subject']
assert mime_msg['To'] == 'Example To <*****@*****.**>'
assert mime_msg['From'] == 'Example From <*****@*****.**>'
cc_header = (
'[email protected],'
'Example CC2 <*****@*****.**>,'
'Example CC3 <*****@*****.**>'
)
assert mime_msg['CC'] == cc_header
assert 'BCC' not in mime_msg
assert mime_msg['Reply-To'] == msg['headers']['Reply-To']
assert mime_msg['X-Mailer'] == msg['headers']['X-Mailer']
mime_msg_parts = [part for part in mime_msg.walk()]
assert len(mime_msg_parts) == 3
text_part, html_part = mime_msg_parts[1:]
assert text_part.get_content_type() == 'text/plain'
assert text_part.get_payload(decode=True) == msg['text_body'].encode('utf-8')
assert html_part.get_content_type() == 'text/html'
assert html_part.get_payload(decode=True) == msg['html_body'].encode('utf-8')
def test_repr(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.__repr__() == (
u"""<Envelope from="Example From <*****@*****.**>" """
u"""to="Example To <*****@*****.**>" """
u"""subject="I'm a helicopter!">""")
def test_send(self):
envelope = Envelope(from_addr='*****@*****.**',
to_addr='*****@*****.**',
subject='Testing envelopes!',
text_body='Just a testy test.')
conn, result = envelope.send(host='localhost')
assert conn._conn is not None
assert len(conn._conn._call_stack.get('sendmail', [])) == 1
def test_addrs_to_header(self):
addrs = [
'*****@*****.**',
'Test2 <*****@*****.**>',
('*****@*****.**', 'Test3'),
]
header = Envelope()._addrs_to_header(addrs)
ok_header = ('[email protected],'
'Test2 <*****@*****.**>,'
'Test3 <*****@*****.**>')
assert header == ok_header
try:
header = Envelope()._addrs_to_header([1])
except MessageEncodeError, exc:
assert exc.message == '1 is not a valid address'
def test_raise(self):
try:
Envelope()._raise(RuntimeError, u'ęóąśłżźćń')
except RuntimeError as exc:
assert exc.args[0] == encoded(u'ęóąśłżźćń', 'utf-8')
except:
raise
else:
assert 'RuntimeError not raised'
def test_addrs_to_header(self):
addrs = [
'*****@*****.**',
'Test2 <*****@*****.**>',
('*****@*****.**', 'Test3'),
]
header = Envelope()._addrs_to_header(addrs)
ok_header = ('[email protected],'
'Test2 <*****@*****.**>,'
'Test3 <*****@*****.**>')
assert header == ok_header
try:
header = Envelope()._addrs_to_header([1])
except MessageEncodeError as exc:
assert exc.args[0] == '1 is not a valid address'
except:
raise
else:
assert False, "MessageEncodeError not raised"
def test_to_mime_message_with_no_data(self):
envelope = Envelope()
mime_msg = envelope.to_mime_message()
assert mime_msg['Subject'] == ''
assert mime_msg['To'] == ''
assert mime_msg['From'] == ''
assert 'CC' not in mime_msg
assert 'BCC' not in mime_msg
mime_msg_parts = [part for part in mime_msg.walk()]
assert len(mime_msg_parts) == 1
def test_to_mime_message_unicode(self):
msg = {
'to_addr': ('*****@*****.**', u'ęóąśłżźćń'),
'from_addr': ('*****@*****.**', u'ęóąśłżźćń'),
'subject': u'ęóąśłżźćń',
'html_body': u'ęóąśłżźćń',
'text_body': u'ęóąśłżźćń',
'cc_addr': [('*****@*****.**', u'ęóąśłżźćń')],
'bcc_addr': [u'ęóąśłżźćń <*****@*****.**>'],
'headers': {
'X-Test': u'ęóąśłżźćń'
},
'charset': 'utf-8'
}
envelope = Envelope(**msg)
def enc_addr_header(name, email):
header = Header(name)
header.append(email)
return header.encode()
mime_msg = envelope.to_mime_message()
assert mime_msg is not None
assert mime_msg['Subject'] == Header(msg['subject'], 'utf-8').encode()
assert mime_msg['To'] == enc_addr_header(u'ęóąśłżźćń',
'<*****@*****.**>')
assert mime_msg['From'] == enc_addr_header(u'ęóąśłżźćń',
'<*****@*****.**>')
assert mime_msg['CC'] == enc_addr_header(u'ęóąśłżźćń',
'<*****@*****.**>')
assert mime_msg['BCC'] == enc_addr_header(u'ęóąśłżźćń',
'<*****@*****.**>')
assert mime_msg['X-Test'] == Header(msg['headers']['X-Test'],
'utf-8').encode()
mime_msg_parts = [part for part in mime_msg.walk()]
assert len(mime_msg_parts) == 3
text_part, html_part = mime_msg_parts[1:]
assert text_part.get_content_type() == 'text/plain'
assert text_part.get_payload(
decode=True) == msg['text_body'].encode('utf-8')
assert html_part.get_content_type() == 'text/html'
assert html_part.get_payload(
decode=True) == msg['html_body'].encode('utf-8')
def test_to_mime_message_with_many_to_addresses(self):
msg = self._dummy_message()
msg['to_addr'] = [
'*****@*****.**', 'Example To2 <*****@*****.**>',
('*****@*****.**', 'Example To3')
]
envelope = Envelope(**msg)
mime_msg = envelope.to_mime_message()
assert mime_msg is not None
to_header = ('[email protected],'
'Example To2 <*****@*****.**>,'
'Example To3 <*****@*****.**>')
assert mime_msg['To'] == to_header
def test_send(self):
conn = SMTP('localhost')
msg = self._dummy_message()
envelope = Envelope(**msg)
mime_msg = envelope.to_mime_message()
conn.send(envelope)
assert conn._conn is not None
assert len(conn._conn._call_stack.get('sendmail', [])) == 1
call_args = conn._conn._call_stack['sendmail'][0][0]
assert len(call_args) == 3
assert call_args[0] == mime_msg['From']
assert call_args[1] == mime_msg['To']
assert call_args[2] != ''
def test_add_attachment(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
_jpg = self._tempfile(suffix='.jpg')
envelope.add_attachment(_jpg)
_mp3 = self._tempfile(suffix='.mp3')
envelope.add_attachment(_mp3)
_pdf = self._tempfile(suffix='.pdf')
envelope.add_attachment(_pdf)
_something = self._tempfile(suffix='.something', prefix=u'ęóąśłżźćń')
envelope.add_attachment(_something)
_octet = self._tempfile(suffix='.txt')
envelope.add_attachment(_octet, mimetype='application/octet-stream')
assert len(envelope._parts) == 7
assert envelope._parts[0][0] == 'text/plain'
assert envelope._parts[1][0] == 'text/html'
assert envelope._parts[2][0] == 'image/jpeg'
assert envelope._parts[2][1]['Content-Disposition'] ==\
'attachment; filename="%s"' % os.path.basename(_jpg)
assert envelope._parts[3][0] == 'audio/mpeg'
assert envelope._parts[3][1]['Content-Disposition'] ==\
'attachment; filename="%s"' % os.path.basename(_mp3)
assert envelope._parts[4][0] == 'application/pdf'
assert envelope._parts[4][1]['Content-Disposition'] ==\
'attachment; filename="%s"' % os.path.basename(_pdf)
assert envelope._parts[5][0] == 'application/octet-stream'
assert envelope._parts[5][1]['Content-Disposition'] ==\
'attachment; filename="%s"' %\
os.path.basename(encoded(_something, 'utf-8'))
assert envelope._parts[6][0] == 'application/octet-stream'
assert envelope._parts[6][1]['Content-Disposition'] ==\
'attachment; filename="%s"' % os.path.basename(_octet)
def test_to_mime_message_unicode(self):
msg = {
'to_addr': '*****@*****.**',
'to_name': u'ęóąśłżźćń',
'from_addr': '*****@*****.**',
'from_name': u'ęóąśłżźćń',
'subject': u'ęóąśłżźćń',
'html_body': u'ęóąśłżźćń',
'text_body': u'ęóąśłżźćń',
'cc_addrs': [('*****@*****.**', u'ęóąśłżźćń')],
'bcc_addrs': [('*****@*****.**', u'ęóąśłżźćń')],
'headers': {
'X-Test': u'ęóąśłżźćń'
},
'charset': 'utf-8'
}
envelope = Envelope(**msg)
mime_msg = envelope.to_mime_message()
assert mime_msg is not None
assert mime_msg['Subject'] == msg['subject'].encode('utf-8')
assert mime_msg['To'] == u'ęóąśłżźćń <*****@*****.**>'.encode('utf-8')
assert mime_msg['From'] == u'ęóąśłżźćń <*****@*****.**>'.\
encode('utf-8')
cc_header = u'ęóąśłżźćń <*****@*****.**>'
assert mime_msg['CC'] == cc_header.encode('utf-8')
bcc_header = u'ęóąśłżźćń <*****@*****.**>'
assert mime_msg['BCC'] == bcc_header.encode('utf-8')
assert mime_msg['X-Test'] == msg['headers']['X-Test'].encode('utf-8')
mime_msg_parts = [part for part in mime_msg.walk()]
assert len(mime_msg_parts) == 3
text_part, html_part = mime_msg_parts[1:]
assert text_part.get_content_type() == 'text/plain'
assert text_part.get_payload() == msg['text_body'].encode('utf-8')
assert html_part.get_content_type() == 'text/html'
assert html_part.get_payload() == msg['html_body'].encode('utf-8')
def test_constructor(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope._to == [msg['to_addr']]
assert envelope._from == msg['from_addr']
assert envelope._subject == msg['subject']
assert len(envelope._parts) == 2
text_part = envelope._parts[0]
assert text_part[0] == 'text/plain'
assert text_part[1] == msg['text_body']
assert text_part[2] == msg['charset']
html_part = envelope._parts[1]
assert html_part[0] == 'text/html'
assert html_part[1] == msg['html_body']
assert html_part[2] == msg['charset']
assert envelope._cc == msg['cc_addr']
assert envelope._bcc == msg['bcc_addr']
assert envelope._headers == msg['headers']
assert envelope._charset == msg['charset']
def test_clear_headers(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_headers()
assert envelope.headers == {}
def test_headers_property(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.headers == msg['headers']
def test_charset_property(self):
envelope = Envelope()
assert envelope.charset == envelope._charset
envelope.charset = 'latin2'
assert envelope._charset == 'latin2'
def test_from_addr_property(self):
envelope = Envelope(**self._dummy_message())
assert envelope.from_addr == envelope._from
envelope.from_addr = '*****@*****.**'
assert envelope.from_addr == '*****@*****.**'
def test_from_name_property(self):
envelope = Envelope(**self._dummy_message())
assert envelope.from_name == envelope._from[1]
envelope.from_name = '*****@*****.**'
assert envelope.from_name == '*****@*****.**'
def test_clear_cc(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_cc()
assert envelope.cc == []
def test_clear_bcc_addr(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_bcc_addr()
assert envelope.bcc_addr == []
def test_add_attachment(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
_jpg = self._tempfile(suffix='.jpg')
envelope.add_attachment(_jpg)
_mp3 = self._tempfile(suffix='.mp3')
envelope.add_attachment(_mp3)
_pdf = self._tempfile(suffix='.pdf')
envelope.add_attachment(_pdf)
_something = self._tempfile(suffix='.something', prefix=u'ęóąśłżźćń')
envelope.add_attachment(_something)
_octet = self._tempfile(suffix='.txt')
envelope.add_attachment(_octet, mimetype='application/octet-stream')
# Attach from string
envelope.add_attachment('file1.txt', data=LOREM, mimetype='text/plain')
# Attach from stream
sio = io.BytesIO(LOREM)
envelope.add_attachment('file2.txt', data=sio, mimetype='text/plain')
assert len(envelope._parts) == 9
assert envelope._parts[0][0] == 'text/plain'
assert envelope._parts[1][0] == 'text/html'
assert envelope._parts[2][0] == 'image/jpeg'
assert envelope._parts[2][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename(_jpg)
assert envelope._parts[3][0] == 'audio/mpeg'
assert envelope._parts[3][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename(_mp3)
assert envelope._parts[4][0] == 'application/pdf'
assert envelope._parts[4][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename(_pdf)
assert envelope._parts[5][0] == 'application/octet-stream'
assert envelope._parts[5][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
quote(os.path.basename(encoded(_something, 'utf-8')))
assert envelope._parts[6][0] == 'application/octet-stream'
assert envelope._parts[6][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename(_octet)
assert envelope._parts[6][0] == 'application/octet-stream'
assert envelope._parts[6][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename(_octet)
assert envelope._parts[7][0] == 'text/plain'
assert envelope._parts[7][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename('file1.txt')
assert envelope._parts[7][1].get_payload(decode=True) == LOREM
assert envelope._parts[8][0] == 'text/plain'
assert envelope._parts[8][1]['Content-Disposition'] ==\
"attachment; filename*=utf-8''%s" %\
os.path.basename('file2.txt')
assert envelope._parts[8][1].get_payload(decode=True) == LOREM
def test_raise(self):
try:
Envelope()._raise(RuntimeError, u'ęóąśłżźćń')
except RuntimeError, exc:
assert exc.message == u'ęóąśłżźćń'.encode('utf-8')
