def test_to_file_from_file():
mb = mailbox.mbox("tests/spam")
msg = encoding.from_message(mb[0])
outfile = "run/encoding_test.msg"
with open(outfile, 'w') as outfp:
encoding.to_file(msg, outfp)
with open(outfile) as outfp:
msg2 = encoding.from_file(outfp)
outdata = open(outfile).read()
assert_equal(len(msg), len(msg2))
os.unlink(outfile)
def test_to_file_from_file():
mb = mailbox.mbox("tests/spam")
msg = encoding.from_message(mb[0])
outfile = "run/encoding_test.msg"
with open(outfile, 'w') as outfp:
encoding.to_file(msg, outfp)
with open(outfile) as outfp:
msg2 = encoding.from_file(outfp)
outdata = open(outfile).read()
assert_equal(len(msg), len(msg2))
os.unlink(outfile)
def test_content_encoding_headers_are_maintained():
inmail = encoding.from_file(open("tests/signed.msg"))
ctype, ctype_params = inmail.content_encoding['Content-Type']
assert_equal(ctype, 'multipart/signed')
# these have to be maintained
for key in ['protocol', 'micalg']:
assert key in ctype_params
# these get removed
for key in encoding.CONTENT_ENCODING_REMOVED_PARAMS:
assert key not in ctype_params
outmsg = encoding.to_message(inmail)
ctype, ctype_params = encoding.parse_parameter_header(outmsg, 'Content-Type')
for key in ['protocol', 'micalg']:
assert key in ctype_params, key
def test_content_encoding_headers_are_maintained():
inmail = encoding.from_file(open("tests/signed.msg"))
ctype, ctype_params = inmail.content_encoding['Content-Type']
assert_equal(ctype, 'multipart/signed')
# these have to be maintained
for key in ['protocol', 'micalg']:
assert key in ctype_params
# these get removed
for key in encoding.CONTENT_ENCODING_REMOVED_PARAMS:
assert key not in ctype_params
outmsg = encoding.to_message(inmail)
ctype, ctype_params = encoding.parse_parameter_header(outmsg, 'Content-Type')
for key in ['protocol', 'micalg']:
assert key in ctype_params, key
def test_to_message_encoding_error(mp_init):
mp_init.side_effect = raises_TypeError
test = encoding.from_file(open("tests/borked.msg"))
msg = encoding.to_message(test)
def test_specially_borked_lua_message():
assert encoding.from_file(open("tests/borked.msg"))
def test_to_message_encoding_error(mp_init):
mp_init.side_effect = raises_TypeError
test = encoding.from_file(open("tests/borked.msg"))
msg = encoding.to_message(test)
def test_specially_borked_lua_message():
assert encoding.from_file(open("tests/borked.msg"))
