From f3f0673c3c6ecd48e7f4bba770cffd70c86cb19c Mon Sep 17 00:00:00 2001 From: Quentin Dufour Date: Tue, 5 Jul 2022 12:04:04 +0200 Subject: Fix test IMAP parsing --- tests/send-to-imap.py | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) (limited to 'tests/send-to-imap.py') diff --git a/tests/send-to-imap.py b/tests/send-to-imap.py index e583ea4..ccbebda 100644 --- a/tests/send-to-imap.py +++ b/tests/send-to-imap.py @@ -3,6 +3,17 @@ from os import listdir from os.path import isfile, join import sys +def rebuild_body_res(b): + bb = b'' + for e in b: + if type(e) is tuple: + bb += b'\r\n'.join([p for p in e]) + else: + bb += e + + f = bb[bb.find(b'('):] + return f + path = sys.argv[1] onlyfiles = [join(path, f) for f in listdir(path) if isfile(join(path, f)) and len(f) > 4 and f[-4:] == ".eml"] @@ -20,24 +31,15 @@ with IMAP4_SSL(host="localhost") as M: seq = (f"{idx+1}:{idx+1}").encode() (r, b) = M.fetch(seq, "(BODY)") assert r == 'OK' - if type(b[0]) is tuple: - bb = b'\r\n'.join([p for p in b[0]]) - else: - bb = b[0] - f = bb[bb.find(b'('):] + + with open(f_noext + ".body", 'w+b') as w: - w.write(f) + w.write(rebuild_body_res(b)) (r, b) = M.fetch(seq, "(BODYSTRUCTURE)") assert r == 'OK' - if type(b[0]) is tuple: - bb = b'\r\n'.join([p for p in b[0]]) - else: - bb = b[0] - - f = bb[bb.find(b'('):] with open(f_noext + ".bodystructure", 'w+b') as w: - w.write(f) + w.write(rebuild_body_res(b)) M.close() M.logout() -- cgit v1.2.3