8 from framework import VppTestRunner
11 def add_from_dir(suite, directory):
13 for _f in os.listdir(directory):
14 f = "%s/%s" % (directory, _f)
16 add_from_dir(suite, f)
18 if not os.path.isfile(f):
21 sys.path.insert(0, directory)
23 if not _f.startswith("test_") or not _f.endswith(".py"):
25 name = "".join(f.split("/")[-1].split(".")[:-1])
26 if name in sys.modules:
27 raise Exception("Duplicate test module `%s' found!" % name)
28 module = importlib.import_module(name)
29 for name, cls in module.__dict__.items():
30 if not isinstance(cls, type):
32 if not issubclass(cls, unittest.TestCase):
34 if name == "VppTestCase":
36 for method in dir(cls):
37 if not callable(getattr(cls, method)):
39 if method.startswith("test_"):
40 suite.addTest(cls(method))
42 if __name__ == '__main__':
44 verbose = int(os.getenv("V", 0))
48 parser = argparse.ArgumentParser(description="VPP unit tests")
49 parser.add_argument("-f", "--failfast", action='count',
50 help="fast failure flag")
51 parser.add_argument("-d", "--dir", action='append', type=str,
52 help="directory containing test files "
53 "(may be specified multiple times)")
54 args = parser.parse_args()
55 failfast = True if args.failfast == 1 else False
57 suite = unittest.TestSuite()
59 print("Adding tests from directory tree %s" % d)
60 add_from_dir(suite, d)
61 sys.exit(not VppTestRunner(verbosity=verbose,
62 failfast=failfast).run(suite).wasSuccessful())