summaryrefslogtreecommitdiffstats
path: root/test/discover_tests.py
diff options
context:
space:
mode:
authorKlement Sekera <ksekera@cisco.com>2017-08-17 07:38:42 +0200
committerFlorin Coras <florin.coras@gmail.com>2017-08-25 18:05:17 +0000
commitfcbf44448b85519b85616a6b87310462249c1c63 (patch)
tree34eb9d341ffb07f80f40393331fdf3c9ed0a0f55 /test/discover_tests.py
parent6d157c2f91a5cf334968df5a6ac4963fedb8c706 (diff)
make test: separate test discovery code
Separating test discovery code to it's own script file has the advantage of easily doing e.g. listing of all existing tests. Change-Id: I80dc280263cc7e33e7e13cb0d48b39bf08ece24d Signed-off-by: Klement Sekera <ksekera@cisco.com>
Diffstat (limited to 'test/discover_tests.py')
-rwxr-xr-xtest/discover_tests.py57
1 files changed, 57 insertions, 0 deletions
diff --git a/test/discover_tests.py b/test/discover_tests.py
new file mode 100755
index 00000000000..eea594107b6
--- /dev/null
+++ b/test/discover_tests.py
@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+import sys
+import os
+import unittest
+import importlib
+import argparse
+
+
+def discover_tests(directory, callback):
+ do_insert = True
+ for _f in os.listdir(directory):
+ f = "%s/%s" % (directory, _f)
+ if os.path.isdir(f):
+ discover_tests(f, callback)
+ 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])
+ if name in sys.modules:
+ raise Exception("Duplicate test module `%s' found!" % name)
+ 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":
+ 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 = "."
+
+ suite = unittest.TestSuite()
+ for d in args.dir:
+ discover_tests(d, print_callback)