b8c72a4a8d
instead of using build-root, use /test/venv directory for virtualenv similarly, don't pollute build-root with test-built binaries Change-Id: I1e63c04037eaee718b27b34ef16c9eb0252afa53 Signed-off-by: Klement Sekera <ksekera@cisco.com>
59 lines
1.9 KiB
Python
Executable File
59 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import os
|
|
import unittest
|
|
import importlib
|
|
import argparse
|
|
|
|
|
|
def discover_tests(directory, callback, ignore_path):
|
|
do_insert = True
|
|
for _f in os.listdir(directory):
|
|
f = "%s/%s" % (directory, _f)
|
|
if os.path.isdir(f):
|
|
if ignore_path is not None and f.startswith(ignore_path):
|
|
continue
|
|
discover_tests(f, callback, ignore_path)
|
|
continue
|
|
if not os.path.isfile(f):
|
|
continue
|
|
if do_insert:
|
|
sys.path.insert(0, directory)
|
|
do_insert = False
|
|
if not _f.startswith("test_") or not _f.endswith(".py"):
|
|
continue
|
|
name = "".join(f.split("/")[-1].split(".")[:-1])
|
|
module = importlib.import_module(name)
|
|
for name, cls in module.__dict__.items():
|
|
if not isinstance(cls, type):
|
|
continue
|
|
if not issubclass(cls, unittest.TestCase):
|
|
continue
|
|
if name == "VppTestCase" or name.startswith("Template"):
|
|
continue
|
|
for method in dir(cls):
|
|
if not callable(getattr(cls, method)):
|
|
continue
|
|
if method.startswith("test_"):
|
|
callback(_f, cls, method)
|
|
|
|
|
|
def print_callback(file_name, cls, method):
|
|
print("%s.%s.%s" % (file_name, cls.__name__, method))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description="Discover VPP unit tests")
|
|
parser.add_argument("-d", "--dir", action='append', type=str,
|
|
help="directory containing test files "
|
|
"(may be specified multiple times)")
|
|
args = parser.parse_args()
|
|
if args.dir is None:
|
|
args.dir = "."
|
|
|
|
ignore_path = os.getenv("VENV_PATH", "")
|
|
suite = unittest.TestSuite()
|
|
for d in args.dir:
|
|
discover_tests(d, print_callback, ignore_path)
|