def broken_headers_test():
if six.PY2:
message = create.from_string(MAILFORMED_HEADERS)
else:
message = create.from_string(MAILFORMED_HEADERS.decode('utf-8', 'replace'))
ok_(message.headers['Subject'])
eq_(six.text_type, type(message.headers['Subject']))
def message_content_dispositions_test():
message = create.from_string(IPHONE)
parts = list(message.walk())
eq_((None, {}), parts[2].content_disposition)
eq_(('inline', {'filename': 'photo.JPG'}), parts[1].content_disposition)
message = create.from_string("Content-Disposition: Нельзя распарсить")
parts = list(message.walk(with_self=True))
eq_((None, {}), parts[0].content_disposition)
def broken_headers_test_2():
if six.PY2:
message = create.from_string(SPAM_BROKEN_HEADERS)
else:
message = create.from_string(SPAM_BROKEN_HEADERS.decode('utf-8', 'replace'))
ok_(message.headers['Subject'])
eq_(six.text_type, type(message.headers['Subject']))
eq_(('text/plain', {'charset': 'iso-8859-1'}),
message.headers['Content-Type'])
eq_(six.text_type, type(message.body))
def bilingual_test():
message = create.from_string(BILINGUAL)
eq_(u"Simple text. How are you? Как ты поживаешь?",
message.headers['Subject'])
message.headers['Subject'] = u"Да все ок!"
eq_(u"Да все ок!", message.headers['Subject'])
message = create.from_string(message.to_string())
eq_(u"Да все ок!", message.headers['Subject'])
eq_("", message.headers.get("SashaNotExists", ""))
def message_content_dispositions_test():
message = create.from_string(IPHONE)
parts = list(message.walk())
eq_((None, {}), parts[2].content_disposition)
eq_(('inline', {'filename': 'photo.JPG'}), parts[1].content_disposition)
message = create.from_string("Content-Disposition: Нельзя распарсить")
parts = list(message.walk(with_self=True))
# content disposition value is anything (including unicode chars) up to the first space, tab or semicolon
# but non-ascii value will raise DecodeError
eq_(('Нельзя', {}), parts[0].content_disposition)
def message_content_dispositions_test():
message = create.from_string(IPHONE)
parts = list(message.walk())
eq_((None, {}), parts[2].content_disposition)
eq_(('inline', {'filename': 'photo.JPG'}), parts[1].content_disposition)
message = create.from_string("Content-Disposition: Нельзя распарсить")
parts = list(message.walk(with_self=True))
# content disposition value is anything (including unicode chars) up to the first space, tab or semicolon
# but non-ascii value will raise DecodeError
# FIXME: In python 2 the returned value is binary, should it be unicode?
expected_cd = u'нельзя' if six.PY3 else 'Нельзя'
eq_((expected_cd, {}), parts[0].content_disposition)
def clean_subject_test():
# Given
message = create.from_string(ENCLOSED)
message.headers['Subject'] = 'FWD: RE: FW: Foo Bar'
# When/Then
eq_('Foo Bar', message.clean_subject)
def broken_headers_test_2():
message = create.from_string(SPAM_BROKEN_HEADERS)
ok_(message.headers['Subject'])
eq_(unicode, type(message.headers['Subject']))
eq_(('text/plain', {'charset': 'iso-8859-1'}),
message.headers['Content-Type'])
eq_(unicode, type(message.body))
def message_headers_transform_test():
"""
Header transformation is reflected in the underlying Python standard
library email object.
"""
# Given
orig = create.from_string(ENCLOSED)
# When
orig.headers.transform(lambda k, v: ('X-{}'.format(k), v[:1]))
restored = create.from_python(orig.to_python_message())
# Then
eq_([('X-Delivered-To', 'b'),
('X-Received', 'b'),
('X-Received', 'b'),
('X-Return-Path', '<'),
('X-Received', 'f'),
('X-Received-Spf', 'p'),
('X-Authentication-Results', 'm'),
('X-Dkim-Signature', 'a'),
('X-Domainkey-Signature', 'a'),
('X-Content-Type', 'm'),
('X-Mime-Version', '1'),
('X-Received', 'b'),
('X-Received', 'f'),
('X-Message-Id', '<'),
('X-Date', 'S'),
('X-From', 'B'),
('X-User-Agent', 'M'),
('X-To', u'"'),
('X-Subject', 'W'),
('X-X-Example-Sid', 'W')],
restored.headers.items())
def test_walk():
message = create.from_string(ENCLOSED)
expected = [
'multipart/mixed', 'text/plain', 'message/rfc822',
'multipart/alternative', 'text/plain', 'text/html'
]
eq_(expected[1:], [str(p.content_type) for p in message.walk()])
eq_(expected, [str(p.content_type) for p in message.walk(with_self=True)])
eq_(['text/plain', 'message/rfc822'],
[str(p.content_type) for p in message.walk(skip_enclosed=True)])
def message_headers_equivalence_test():
"""
FallbackMimePart headers match MimePart headers exactly for the same input.
"""
# Given
message = scan(ENCLOSED)
fallback_message = create.from_string(ENCLOSED)
# When
eq_(message.headers.items(), fallback_message.headers.items())
def detected_fields_test():
# Given
message = create.from_string(MAILGUN_PIC)
attachment = message.parts[1]
# When/Then
eq_('mailgun.png', attachment.detected_file_name)
eq_('png', attachment.detected_subtype)
eq_('image', attachment.detected_format)
ok_(not attachment.is_body())
def set_message_id_test():
# Given
message = create.from_string(ENCLOSED)
# When
message.message_id = '*****@*****.**'
# Then
eq_('*****@*****.**', message.message_id)
eq_('<*****@*****.**>', message.headers['Message-Id'])
def bounce_test():
# When
message = create.from_string(BOUNCE)
# Then
ok_(message.is_bounce())
eq_('5.1.1', message.bounce.status)
eq_(
'smtp; 550-5.1.1 The email account that you tried to reach does '
'not exist. Please try 550-5.1.1 double-checking the recipient\'s email '
'address for typos or 550-5.1.1 unnecessary spaces. Learn more at '
'550 5.1.1 http://mail.google.com/support/bin/answer.py?answer=6596 '
'17si20661415yxe.22', message.bounce.diagnostic_code)
def message_alter_body_and_serialize_test():
message = create.from_string(IPHONE)
parts = list(message.walk())
eq_(3, len(parts))
eq_(u'\n\n\n~Danielle', parts[2].body)
eq_((None, {}), parts[2].content_disposition)
eq_(('inline', {'filename': 'photo.JPG'}), parts[1].content_disposition)
part = list(message.walk())[2]
part.body = u'Привет, Danielle!\n\n'
with closing(StringIO()) as out:
message.to_stream(out)
message1 = create.from_string(out.getvalue())
message2 = create.from_string(message.to_string())
parts = list(message1.walk())
eq_(3, len(parts))
eq_(u'Привет, Danielle!\n\n', parts[2].body)
parts = list(message2.walk())
eq_(3, len(parts))
eq_(u'Привет, Danielle!\n\n', parts[2].body)
def message_headers_append_test():
"""
`prepend` and `add` both insert a header at the beginning and the end of the
header list.
"""
# Given
orig = create.from_string(ENCLOSED)
# When
orig.headers.prepend('foo-bar', 'hello')
orig.headers.add('blah', 'kitty')
restored = create.from_python(orig.to_python_message())
# Then
eq_(('Blah', 'kitty'), restored.headers.items()[0])
eq_(('Foo-Bar', 'hello'), restored.headers.items()[1])
def bad_string_test():
mime = "Content-Type: multipart/broken\n\n"
message = create.from_string("Content-Type:multipart/broken")
eq_(mime, message.to_string())
with closing(StringIO()) as out:
message.to_stream(out)
eq_(mime, out.getvalue())
list(message.walk())
message.remove_headers()
assert_false(message.is_attachment())
assert_false(message.is_inline())
assert_false(message.is_delivery_notification())
assert_false(message.is_bounce())
ok_(message.to_python_message())
eq_(None, message.get_attached_message())
ok_(str(message))
def message_from_python_test():
message = create.from_string(ENCLOSED)
eq_(2, len(message.parts))
eq_('multipart/alternative', message.parts[1].enclosed.content_type)
eq_('multipart/mixed', message.content_type)
assert_false(message.body)
message.headers['Sasha'] = 'Hello!'
message.parts[1].enclosed.headers['Yo'] = u'Man'
ok_(message.to_string())
ok_(str(message))
ok_(str(message.parts[0]))
eq_('*****@*****.**', message.message_id)
eq_('Wow', message.subject)
m = message.get_attached_message()
eq_('multipart/alternative', str(m.content_type))
eq_('Thanks!', m.subject)
def test_binary_attachment():
"""
A text part body is returned as a unicode string, but any other part type
body is returned as a binary string.
"""
# Given
message = create.from_string(ATTACHED_PDF)
# When
parts = [p for p in message.walk(with_self=True)]
# Then
def part_spec(p):
return str(p.content_type), str(type(p.body))
eq_(('multipart/mixed', "<type 'NoneType'>"), part_spec(parts[0]))
eq_(('multipart/alternative', "<type 'NoneType'>"), part_spec(parts[1]))
eq_(('text/plain', "<type 'unicode'>"), part_spec(parts[2]))
eq_(('application/pdf', "<type 'str'>"), part_spec(parts[3]))
def bounce_test():
# When
message = create.from_string(BOUNCE)
# Then
ok_(message.is_bounce())
eq_('5.1.1', message.bounce.status)
expected_code = (
'smtp; 550-5.1.1 The email account that you tried to reach does '
'not exist. Please try 550-5.1.1 double-checking the recipient\'s email '
'address for typos or 550-5.1.1 unnecessary spaces. Learn more at '
'550 5.1.1 http://mail.google.com/support/bin/answer.py?answer=6596 '
'17si20661415yxe.22')
# In Python 2 email.message.Message used to truncate leading spaces, but
# in Python 3 leading spaces are preserved.
if six.PY2:
expected_code = expected_code.replace(' ', ' ').replace(' ', ' ')
eq_(expected_code, message.bounce.diagnostic_code)
def message_headers_mutation_test():
"""
FallbackMimePart headers match MimePart headers exactly for the same input.
"""
# Given
orig = create.from_string(ENCLOSED)
eq_('Wow', orig.headers['Subject'])
eq_(ContentType('multipart', 'mixed', {'boundary': u'===============6195527458677812340=='}),
orig.headers['Content-Type'])
# When
del orig.headers['Subject']
orig.headers['Content-Type'] = ContentType('text', 'foo')
orig.headers.prepend('foo-bar', 'hello')
orig.headers.add('blah', 'kitty')
restored = create.from_python(orig.to_python_message())
# Then
ok_('Subject' not in restored.headers)
eq_('', restored.subject)
eq_(ContentType('text', 'foo'), restored.headers['Content-Type'])
def bad_string_test_2():
mime = "Content-Mype: multipart/broken\n\n"
message = create.from_string(mime)
ok_(message.content_type)
eq_((None, {}), message.content_disposition)
def broken_headers_test():
message = create.from_string(MAILFORMED_HEADERS)
ok_(message.headers['Subject'])
eq_(unicode, type(message.headers['Subject']))
def text_only_test():
message = create.from_string(TEXT_ONLY)
eq_(u"Hello,\nI'm just testing message parsing\n\nBR,\nBob",
message.body)
eq_(None, message.bounce)
eq_(None, message.get_attached_message())
def torture_test():
message = create.from_string(TORTURE)
ok_(list(message.walk(with_self=True)))
ok_(message.size)
message.parts[0].content_encoding
message.parts[0].content_encoding = 'blablac'
