lb: remove api boilerplate
[vpp.git] / 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, ignore_path):
11     do_insert = True
12     for _f in os.listdir(directory):
13         f = "%s/%s" % (directory, _f)
14         if os.path.isdir(f):
15             if ignore_path is not None and f.startswith(ignore_path):
16                 continue
17             discover_tests(f, callback, ignore_path)
18             continue
19         if not os.path.isfile(f):
20             continue
21         if do_insert:
22             sys.path.insert(0, directory)
23             do_insert = False
24         if not _f.startswith("test_") or not _f.endswith(".py"):
25             continue
26         name = "".join(f.split("/")[-1].split(".")[:-1])
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     ignore_path = os.getenv("VENV_PATH", "")
56     suite = unittest.TestSuite()
57     for d in args.dir:
58         discover_tests(d, print_callback, ignore_path)