tobias/docker-offlineimap
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reformat offlineimap/test/OLItest/TestRunner.py

Browse Source 
Add some spaces, remove lines,... now format is better (lintian).
This commit is contained in:
Rodolfo García Peñas (kix) 2020-08-29 20:26:29 +02:00
parent 6d4298dca7
commit d75cf8d80c
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
test/OLItest/TestRunner.py
View File

@ -23,6 +23,7 @@ import shutil
import subprocess import subprocess
import tempfile import tempfile
import random import random
random.seed() random.seed()
from offlineimap.CustomConfig import CustomConfigParser from offlineimap.CustomConfig import CustomConfigParser
@ -36,7 +37,7 @@ class OLITestLib():
cmd = None cmd = None
"""command that will be executed to invoke offlineimap""" """command that will be executed to invoke offlineimap"""
def __init__(self, cred_file = None, cmd='offlineimap'): def __init__(self, cred_file=None, cmd='offlineimap'):
""" """
:param cred_file: file of the configuration :param cred_file: file of the configuration
@ -45,7 +46,7 @@ class OLITestLib():
OLITestLib.cred_file = cred_file OLITestLib.cred_file = cred_file
if not os.path.isfile(cred_file): if not os.path.isfile(cred_file):
raise UserWarning("Please copy 'credentials.conf.sample' to '%s' " raise UserWarning("Please copy 'credentials.conf.sample' to '%s' "
"and set your credentials there." % cred_file) "and set your credentials there." % cred_file)
OLITestLib.cmd = cmd OLITestLib.cmd = cmd
@classmethod @classmethod
@ -59,7 +60,7 @@ class OLITestLib():
assert cls.cred_file != None assert cls.cred_file != None
# creating temporary dir for testing in same dir as credentials.conf # creating temporary dir for testing in same dir as credentials.conf
cls.testdir = os.path.abspath( cls.testdir = os.path.abspath(
tempfile.mkdtemp(prefix='tmp_%s_'%suffix, tempfile.mkdtemp(prefix='tmp_%s_' % suffix,
dir=os.path.dirname(cls.cred_file))) dir=os.path.dirname(cls.cred_file)))
cls.write_config_file() cls.write_config_file()
return cls.testdir return cls.testdir
@ -70,12 +71,12 @@ class OLITestLib():
The returned config can be manipulated and then saved with The returned config can be manipulated and then saved with
write_config_file()""" write_config_file()"""
#TODO, only do first time and cache then for subsequent calls? # TODO, only do first time and cache then for subsequent calls?
assert cls.cred_file != None assert cls.cred_file != None
assert cls.testdir != None assert cls.testdir != None
config = CustomConfigParser() config = CustomConfigParser()
config.readfp(default_conf) config.readfp(default_conf)
default_conf.seek(0) # rewind config_file to start default_conf.seek(0) # rewind config_file to start
config.read(cls.cred_file) config.read(cls.cred_file)
config.set("general", "metadata", cls.testdir) config.set("general", "metadata", cls.testdir)
return config return config
@ -130,7 +131,7 @@ class OLITestLib():
sections = ['Repository {0}'.format(reponame)] sections = ['Repository {0}'.format(reponame)]
else: else:
sections = [r for r in config.sections() \ sections = [r for r in config.sections() \
if r.startswith('Repository')] if r.startswith('Repository')]
sections = [s for s in sections if config.get(s, 'Type').lower() == 'imap'] sections = [s for s in sections if config.get(s, 'Type').lower() == 'imap']
for sec in sections: for sec in sections:
# Connect to each IMAP repo and delete all folders # Connect to each IMAP repo and delete all folders
@ -162,14 +163,14 @@ class OLITestLib():
folder = bytearray(m.group('dir')) folder = bytearray(m.group('dir'))
if not m.group('quote'): if not m.group('quote'):
folder = '"%s"' % folder folder = '"%s"' % folder
#folder = folder.replace(br'\"', b'"') # remove quoting # folder = folder.replace(br'\"', b'"') # remove quoting
dirs.append(folder) dirs.append(folder)
# 2) filter out those not starting with INBOX.OLItest and del... # 2) filter out those not starting with INBOX.OLItest and del...
dirs = [d for d in dirs if d.startswith(b'"INBOX.OLItest') or d.startswith(b'"INBOX/OLItest')] dirs = [d for d in dirs if d.startswith(b'"INBOX.OLItest') or d.startswith(b'"INBOX/OLItest')]
for folder in dirs: for folder in dirs:
res_t, data = imapobj.delete(folder) res_t, data = imapobj.delete(folder)
assert res_t == 'OK', "Folder deletion of {0} failed with error"\ assert res_t == 'OK', "Folder deletion of {0} failed with error" \
":\n{1} {2}".format(folder.decode('utf-8'), res_t, data) ":\n{1} {2}".format(folder.decode('utf-8'), res_t, data)
imapobj.logout() imapobj.logout()
@classmethod @classmethod
@ -179,11 +180,11 @@ class OLITestLib():
Does not fail if it already exists""" Does not fail if it already exists"""
assert cls.testdir != None assert cls.testdir != None
maildir = os.path.join(cls.testdir, 'mail', folder) maildir = os.path.join(cls.testdir, 'mail', folder)
for subdir in ('','tmp','cur','new'): for subdir in ('', 'tmp', 'cur', 'new'):
try: try:
os.makedirs(os.path.join(maildir, subdir)) os.makedirs(os.path.join(maildir, subdir))
except OSError as e: except OSError as e:
if e.errno != 17: # 'already exists' is ok. if e.errno != 17: # 'already exists' is ok.
raise raise
@classmethod @classmethod
@ -203,13 +204,13 @@ class OLITestLib():
Use some default content if not given""" Use some default content if not given"""
assert cls.testdir != None assert cls.testdir != None
while True: # Loop till we found a unique filename while True: # Loop till we found a unique filename
mailfile = '{0}:2,'.format(random.randint(0,999999999)) mailfile = '{0}:2,'.format(random.randint(0, 999999999))
mailfilepath = os.path.join(cls.testdir, 'mail', mailfilepath = os.path.join(cls.testdir, 'mail',
folder, 'new', mailfile) folder, 'new', mailfile)
if not os.path.isfile(mailfilepath): if not os.path.isfile(mailfilepath):
break break
with open(mailfilepath,"wb") as mailf: with open(mailfilepath, "wb") as mailf:
mailf.write(b'''From: test <test@offlineimap.org> mailf.write(b'''From: test <test@offlineimap.org>
Subject: Boo Subject: Boo
Date: 1 Jan 1980 Date: 1 Jan 1980
To: test@offlineimap.org To: test@offlineimap.org
@ -229,7 +230,7 @@ Content here.''')
if set(dirs) == set(['cur', 'new', 'tmp']): if set(dirs) == set(['cur', 'new', 'tmp']):
# New maildir folder # New maildir folder
boxes += 1 boxes += 1
#raise RuntimeError("%s is not Maildir" % maildir) # raise RuntimeError("%s is not Maildir" % maildir)
if dirpath.endswith(('/cur', '/new')): if dirpath.endswith(('/cur', '/new')):
mails += len(files) mails += len(files)
return boxes, mails return boxes, mails
@ -246,7 +247,7 @@ Content here.''')
ret = [] ret = []
for dirpath, dirs, files in os.walk(mailfilepath): for dirpath, dirs, files in os.walk(mailfilepath):
if not dirpath.endswith((os.path.sep + 'new', os.path.sep + 'cur')): if not dirpath.endswith((os.path.sep + 'new', os.path.sep + 'cur')):
continue # only /new /cur are interesting continue # only /new /cur are interesting
for file in files: for file in files:
m = cls.re_uidmatch.search(file) m = cls.re_uidmatch.search(file)
uid = m.group(1) if m else None uid = m.group(1) if m else None