src/tools/testwrap

   1 #!/usr/bin/env python3
   2
   3 import argparse
   4 import shutil
   5 import subprocess
   6 import os
   7 import sys
   8
   9 parser = argparse.ArgumentParser()
  10
  11 parser.add_argument('--srcdir', help='source directory of test', type=str)
  12 parser.add_argument('--basedir', help='base directory of test', type=str)
  13 parser.add_argument('--testgroup', help='test group', type=str)
  14 parser.add_argument('--testname', help='test name', type=str)
  15 parser.add_argument('--skip', help='skip test (with reason)', type=str)
  16 parser.add_argument('test_command', nargs='*')
  17
  18 args = parser.parse_args()
  19
  20 testdir = '{}/testrun/{}/{}'.format(
  21     args.basedir, args.testgroup, args.testname)
  22
  23 print('# executing test in {} group {} test {}'.format(
  24     testdir, args.testgroup, args.testname))
  25 sys.stdout.flush()
  26
  27 if args.skip is not None:
  28     print('1..0 # Skipped: ' + args.skip)
  29     sys.exit(0)
  30
  31 if os.path.exists(testdir) and os.path.isdir(testdir):
  32     shutil.rmtree(testdir)
  33 os.makedirs(testdir)
  34
  35 os.chdir(args.srcdir)
  36
  37 # mark test as having started
  38 open(os.path.join(testdir, 'test.start'), 'x')
  39
  40 env_dict = {**os.environ,
  41             'TESTDATADIR': os.path.join(testdir, 'data'),
  42             'TESTLOGDIR': os.path.join(testdir, 'log')}
  43
  44 sp = subprocess.run(args.test_command, env=env_dict)
  45
  46 if sp.returncode == 0:
  47     print('# test succeeded')
  48     open(os.path.join(testdir, 'test.success'), 'x')
  49 else:
  50     print('# test failed')
  51     open(os.path.join(testdir, 'test.fail'), 'x')
  52 sys.exit(sp.returncode)