3diy
/
klipper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test_klippy: Add option to not remove temporary files 

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2018-06-30 12:20:42 -04:00
parent d725160706
commit 0cbfcfd3d1
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/test_klippy.py
View File

@ -18,11 +18,12 @@ class error(Exception):
pass pass
class TestCase: class TestCase:
def __init__(self, fname, dictdir, tempdir, verbose): def __init__(self, fname, dictdir, tempdir, verbose, keepfiles):
self.fname = fname self.fname = fname
self.dictdir = dictdir self.dictdir = dictdir
self.tempdir = tempdir self.tempdir = tempdir
self.verbose = verbose self.verbose = verbose
self.keepfiles = keepfiles
def relpath(self, fname, rel='test'): def relpath(self, fname, rel='test'):
if rel == 'dict': if rel == 'dict':
reldir = self.dictdir reldir = self.dictdir
@ -104,6 +105,8 @@ class TestCase:
raise error("Test failed to raise an error") raise error("Test failed to raise an error")
raise error("Error during test") raise error("Error during test")
# Do cleanup # Do cleanup
if self.keepfiles:
return
for fname in os.listdir(self.tempdir): for fname in os.listdir(self.tempdir):
if fname.startswith(TEMP_OUTPUT_FILE): if fname.startswith(TEMP_OUTPUT_FILE):
os.unlink(fname) os.unlink(fname)
@ -141,6 +144,8 @@ def main():
help="directory for dictionary files") help="directory for dictionary files")
opts.add_option("-t", "--tempdir", dest="tempdir", default=".", opts.add_option("-t", "--tempdir", dest="tempdir", default=".",
help="directory for temporary files") help="directory for temporary files")
opts.add_option("-k", action="store_true", dest="keepfiles",
help="do not remove temporary files")
opts.add_option("-v", action="store_true", dest="verbose", opts.add_option("-v", action="store_true", dest="verbose",
help="show all output from tests") help="show all output from tests")
options, args = opts.parse_args() options, args = opts.parse_args()
@ -150,7 +155,8 @@ def main():
# Run each test # Run each test
for fname in args: for fname in args:
tc = TestCase(fname, options.dictdir, options.tempdir, options.verbose) tc = TestCase(fname, options.dictdir, options.tempdir, options.verbose,
options.keepfiles)
res = tc.run() res = tc.run()
if res != 'success': if res != 'success':
sys.stderr.write("\n\nTest case %s FAILED (%s)!\n\n" % (fname, res)) sys.stderr.write("\n\nTest case %s FAILED (%s)!\n\n" % (fname, res))