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_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"))