third_party/typ/run

   1 #!/usr/bin/env python
   2
   3 from __future__ import print_function
   4
   5 import argparse
   6 import os
   7 import subprocess
   8 import sys
   9
  10 from tools import cov
  11
  12
  13 is_python3 = bool(sys.version_info.major == 3)
  14 has_python34 = False
  15 verbose = False
  16 repo_dir = os.path.abspath(os.path.dirname(__file__))
  17 path_to_cov = os.path.join(repo_dir, 'tools', 'cov.py')
  18 path_to_runner = os.path.join(repo_dir, 'typ', 'runner.py')
  19
  20
  21 def call(*args, **kwargs):
  22     if verbose:
  23         print(' '.join(args[0]))
  24     ret = subprocess.call(*args, **kwargs)
  25     if ret != 0:
  26         sys.exit(ret)
  27
  28
  29 def main(argv):
  30     parser = argparse.ArgumentParser(prog='run')
  31     parser.add_argument('--no3', action='store_true',
  32                         help='Do not run the tests under Python 3.')
  33     parser.add_argument('-v', '--verbose', action='store_true')
  34     subps = parser.add_subparsers()
  35
  36     subp = subps.add_parser('clean', help='Remove any local files.')
  37     subp.set_defaults(func=run_clean)
  38
  39     subp = subps.add_parser('coverage',
  40                             help='Run the tests and report code coverage.')
  41     subp.set_defaults(func=run_coverage)
  42     cov.add_arguments(subp)
  43
  44     subp = subps.add_parser('help',
  45                             help='Get help on a subcommand.')
  46     subp.add_argument(nargs='?', action='store', dest='subcommand',
  47                       help='The command to get help for.')
  48     subp.set_defaults(func=run_help)
  49
  50     subp = subps.add_parser('lint',
  51                             help='run lint over the source')
  52     subp.set_defaults(func=run_lint)
  53
  54     subp = subps.add_parser('tests',
  55                             help='run the tests')
  56     subp.set_defaults(func=run_tests)
  57
  58     args = parser.parse_args(argv)
  59
  60     global verbose
  61     if args.verbose:
  62         verbose = True
  63     global has_python34
  64     if not args.no3:
  65         try:
  66             ver = subprocess.check_output(['python3', '--version'])
  67             has_python34 = ver.split()[1] >= '3.4'
  68         except:
  69             pass
  70     args.func(args)
  71
  72
  73 def run_clean(args):
  74     call(['git', 'clean', '-fxd'])
  75
  76
  77 def run_coverage(args):
  78     if not args.path:
  79         args.path = [repo_dir]
  80     if not args.source:
  81         args.source = [os.path.join(repo_dir, 'typ')]
  82     argv = cov.argv_from_args(args)
  83     cov_args = [path_to_runner, '-j', '1']
  84     print('Running coverage of unit tests for Python 2.7.')
  85     call(['python', path_to_cov] + argv + cov_args)
  86     if has_python34:
  87         print('Running coverage of unit tests for Python 3.4.')
  88         call(['python3', path_to_cov] + argv + cov_args)
  89
  90
  91 def run_help(args):
  92     if args.subcommand:
  93         main([args.subcommand, '--help'])
  94     main(['--help'])
  95
  96
  97 def run_lint(args):
  98     call('pylint --rcfile=pylintrc */*.py */*/*.py', shell=True)
  99
 100
 101 def run_tests(args):
 102     print('Testing running the typ module directly if it is in sys.path.')
 103     call(['python', '-m', 'typ', 'typ.tests.main_test.TestMain.test_basic'])
 104
 105     print('Testing running the runner directly if nothing is in sys.path.')
 106     home_dir = os.environ['HOME']
 107     call(['python', path_to_runner, 'typ.tests.main_test.TestMain.test_basic'],
 108          cwd=home_dir)
 109
 110     # Now run all the tests under Python2 and Python3.
 111     print('Running the unit tests under Python 2.')
 112     call(['python', path_to_runner])
 113     if has_python34:
 114         print('Running the unit tests under Python 3.4.')
 115         call(['python3', path_to_runner])
 116
 117
 118 if __name__ == '__main__':
 119     sys.exit(main(sys.argv[1:]))