test/discover_tests.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import os
   5 import unittest
   6 import importlib
   7 import argparse
   8
   9
  10 def discover_tests(directory, callback):
  11     do_insert = True
  12     for _f in os.listdir(directory):
  13         f = "%s/%s" % (directory, _f)
  14         if os.path.isdir(f):
  15             discover_tests(f, callback)
  16             continue
  17         if not os.path.isfile(f):
  18             continue
  19         if do_insert:
  20             sys.path.insert(0, directory)
  21             do_insert = False
  22         if not _f.startswith("test_") or not _f.endswith(".py"):
  23             continue
  24         name = "".join(f.split("/")[-1].split(".")[:-1])
  25         if name in sys.modules:
  26             raise Exception("Duplicate test module `%s' found!" % name)
  27         module = importlib.import_module(name)
  28         for name, cls in module.__dict__.items():
  29             if not isinstance(cls, type):
  30                 continue
  31             if not issubclass(cls, unittest.TestCase):
  32                 continue
  33             if name == "VppTestCase" or name.startswith("Template"):
  34                 continue
  35             for method in dir(cls):
  36                 if not callable(getattr(cls, method)):
  37                     continue
  38                 if method.startswith("test_"):
  39                     callback(_f, cls, method)
  40
  41
  42 def print_callback(file_name, cls, method):
  43     print("%s.%s.%s" % (file_name, cls.__name__, method))
  44
  45
  46 if __name__ == '__main__':
  47     parser = argparse.ArgumentParser(description="Discover VPP unit tests")
  48     parser.add_argument("-d", "--dir", action='append', type=str,
  49                         help="directory containing test files "
  50                              "(may be specified multiple times)")
  51     args = parser.parse_args()
  52     if args.dir is None:
  53         args.dir = "."
  54
  55     suite = unittest.TestSuite()
  56     for d in args.dir:
  57         discover_tests(d, print_callback)