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_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_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_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_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_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_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_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_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_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_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_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_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_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):
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_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_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_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_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_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 'BCC' not in mime_msg
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_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_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_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_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] == [
envelope._addrs_to_header([addr]) for addr in envelope._to + envelope._cc + envelope._bcc
]
assert call_args[2] != ""
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_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_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 "BCC" not in mime_msg
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_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_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_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_from_addr_property(self):
envelope = Envelope(**self._dummy_message())
assert envelope.from_addr == envelope._from
envelope.from_addr = '*****@*****.**'
assert envelope.from_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="%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)
assert envelope._parts[6][0] == 'application/octet-stream'
assert envelope._parts[6][1]['Content-Disposition'] ==\
'attachment; filename="%s"' % os.path.basename(_octet)
assert envelope._parts[7][0] == 'text/plain'
assert envelope._parts[7][1]['Content-Disposition'] ==\
'attachment; filename="%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="%s"' % os.path.basename('file2.txt')
assert envelope._parts[8][1].get_payload(decode=True) == LOREM
def test_clear_headers(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_headers()
assert envelope.headers == {}
def test_charset_property(self):
envelope = Envelope()
assert envelope.charset == envelope._charset
envelope.charset = 'latin2'
assert envelope._charset == 'latin2'
def test_headers_property(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
assert envelope.headers == msg['headers']
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_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')
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_clear_cc(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_cc()
assert envelope.cc == []
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_clear_bcc_addr(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_bcc_addr()
assert envelope.bcc_addr == []
def test_charset_property(self):
envelope = Envelope()
assert envelope.charset == envelope._charset
envelope.charset = "latin2"
assert envelope._charset == "latin2"
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')
def test_clear_headers(self):
msg = self._dummy_message()
envelope = Envelope(**msg)
envelope.clear_headers()
assert envelope.headers == {}
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 == '*****@*****.**'
