From af357c5cfbadb82e9884d086b2d6a84b2820f0db Mon Sep 17 00:00:00 2001 From: Manuel Traut Date: Mon, 4 Dec 2017 15:54:53 +0100 Subject: add --class parameter to specify classname of a testcase, if not given it defaults to 'sh' Signed-off-by: Manuel Traut --- pyjutest | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/pyjutest b/pyjutest index b3ca92d..5c3f60a 100755 --- a/pyjutest +++ b/pyjutest @@ -21,6 +21,7 @@ from pyju.version import pyju_version oparser = OptionParser(usage="usage: %prog [options] \'mycmd param1 param2\'") oparser.add_option("--version", dest="version", action="store_true", default=False) +oparser.add_option("--class", dest="uclass", default="sh") (opt,args) = oparser.parse_args(sys.argv) @@ -32,6 +33,8 @@ if len(args) != 2: oparser.print_help() sys.exit(-1) +cmd = args[1] + if os.path.exists('pyjutest.dat'): with open('pyjutest.dat', 'rb') as dat: tss = load(dat) @@ -40,12 +43,12 @@ else: start = time.time() -p = Popen(sys.argv[1], shell=True, stdout=PIPE, stderr=PIPE) +p = Popen(cmd, shell=True, stdout=PIPE, stderr=PIPE) out,err = p.communicate() duration = time.time() - start -tc = TestCase(sys.argv[1], 'sh', duration, out[:-1].decode('utf-8'), +tc = TestCase(cmd, opt.uclass, duration, out[:-1].decode('utf-8'), err[:-1].decode('utf-8')) if p.returncode: @@ -59,9 +62,9 @@ with open('pyjutest.dat', 'wb') as dat: with open('pyjutest.xml', 'w') as xml: TestSuite.to_file(xml, tss, prettyprint=True) -print("%s STDOUT:" % sys.argv[1]) +print("%s STDOUT:" % cmd) for l in out.decode().split("\n"): print(l) -print("%s STDERR:" % sys.argv[1]) +print("%s STDERR:" % cmd) for l in err.decode().split("\n"): print(l, file=sys.stderr) -- cgit v1.2.3