src/git_stats/tests.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import os
   5
   6 from optparse import OptionParser
   7
   8 from git_stats import testing
   9 from git_stats import config_tests
  10
  11 all_modules = {
  12     "config" : config_tests
  13     }
  14
  15 def dispatch(*args):
  16   """Dispatches test related commands
  17   """
  18
  19   progname = os.path.basename(sys.argv[0]) + " test"
  20   parser = OptionParser(prog=progname)
  21
  22   parser.add_option(
  23       "-a", "--all",
  24       action="store_true",
  25       help="run tests for all known modules")
  26
  27   parser.add_option(
  28       "-m", "--module",
  29       help="run the tests defined in the specified module")
  30
  31   parser.add_option(
  32       "-s", "--single",
  33       metavar="TEST",
  34       help="run only the tests for the specified module")
  35
  36   parser.add_option(
  37       "-t", "--tests",
  38       metavar="TESTS",
  39       action="store_true",
  40       help="run only the specified tests")
  41
  42   parser.add_option(
  43       "-v", "--verbose",
  44       action="store_true",
  45       help="enable verbose output")
  46
  47   if not args:
  48     args.append("--help")
  49
  50   options, args = parser.parse_args(list(args))
  51
  52   tests = []
  53   modules = []
  54
  55   # Add all modules we know about
  56   if options.all:
  57     tests.extend(all_modules.keys())
  58
  59   # Add only one test
  60   if options.single:
  61     tests.append(options.single)
  62
  63   # Also add all tests as specified as command arguments
  64   if options.tests:
  65     tests.extend(args)
  66
  67   if options.module:
  68     try:
  69       module = __import__("git_stats." + options.module, fromlist=True)
  70     except ImportError, e:
  71       parser.error("Could not import module '%s'.\n%s" % (
  72         options.module, str(e)))
  73
  74     modules.append(module)
  75
  76   for test in tests:
  77     if not test in all_modules:
  78       parser.error("Unknown module '%s'." % test)
  79
  80     module = all_modules[test]
  81     modules.append(module)
  82
  83   if not modules:
  84     parser.error("Please specify which modules to test")
  85
  86   classes = []
  87
  88   for module in modules:
  89     module_classes = module.getTestClasses()
  90     classes.extend(module_classes)
  91
  92   runner = testing.GitTestRunner()
  93   res = runner.runTests(options, classes)
  94
  95   return res
  96