This allows running different subsets of tests, e.g.

$ ./tools/testing/kunit/kunit.py build
$ ./tools/testing/kunit/kunit.py exec 'list*'
$ ./tools/testing/kunit/kunit.py exec 'kunit*'

This passes the "kunit_filter.glob" commandline option to the UML
kernel, which currently only supports filtering by suite name.

Signed-off-by: Daniel Latypov <dlaty...@google.com>
Reviewed-by: Brendan Higgins <brendanhigg...@google.com>
---
 tools/testing/kunit/kunit.py        | 21 ++++++++++++++++-----
 tools/testing/kunit/kunit_kernel.py |  4 +++-
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/tools/testing/kunit/kunit.py b/tools/testing/kunit/kunit.py
index e808a47c839b..a15ee33bb1f5 100755
--- a/tools/testing/kunit/kunit.py
+++ b/tools/testing/kunit/kunit.py
@@ -28,12 +28,12 @@ KunitBuildRequest = namedtuple('KunitBuildRequest',
                               ['jobs', 'build_dir', 'alltests',
                                'make_options'])
 KunitExecRequest = namedtuple('KunitExecRequest',
-                             ['timeout', 'build_dir', 'alltests'])
+                             ['timeout', 'build_dir', 'alltests', 
'filter_glob'])
 KunitParseRequest = namedtuple('KunitParseRequest',
                               ['raw_output', 'input_data', 'build_dir', 
'json'])
 KunitRequest = namedtuple('KunitRequest', ['raw_output','timeout', 'jobs',
-                                          'build_dir', 'alltests', 'json',
-                                          'make_options'])
+                                          'build_dir', 'alltests', 
'filter_glob',
+                                          'json', 'make_options'])
 
 KernelDirectoryPath = sys.argv[0].split('tools/testing/kunit/')[0]
 
@@ -93,6 +93,7 @@ def exec_tests(linux: kunit_kernel.LinuxSourceTree,
        test_start = time.time()
        result = linux.run_kernel(
                timeout=None if request.alltests else request.timeout,
+                filter_glob=request.filter_glob,
                build_dir=request.build_dir)
 
        test_end = time.time()
@@ -149,7 +150,7 @@ def run_tests(linux: kunit_kernel.LinuxSourceTree,
                return build_result
 
        exec_request = KunitExecRequest(request.timeout, request.build_dir,
-                                       request.alltests)
+                                       request.alltests, request.filter_glob)
        exec_result = exec_tests(linux, exec_request)
        if exec_result.status != KunitStatus.SUCCESS:
                return exec_result
@@ -197,6 +198,14 @@ def add_exec_opts(parser) -> None:
                            type=int,
                            default=300,
                            metavar='timeout')
+       parser.add_argument('filter_glob',
+                           help='maximum number of seconds to allow for all 
tests '
+                           'to run. This does not include time taken to build 
the '
+                           'tests.',
+                           type=str,
+                           nargs='?',
+                           default='',
+                           metavar='filter_glob')
 
 def add_parse_opts(parser) -> None:
        parser.add_argument('--raw_output', help='don\'t format output from 
kernel',
@@ -263,6 +272,7 @@ def main(argv, linux=None):
                                       cli_args.jobs,
                                       cli_args.build_dir,
                                       cli_args.alltests,
+                                      cli_args.filter_glob,
                                       cli_args.json,
                                       cli_args.make_options)
                result = run_tests(linux, request)
@@ -304,7 +314,8 @@ def main(argv, linux=None):
 
                exec_request = KunitExecRequest(cli_args.timeout,
                                                cli_args.build_dir,
-                                               cli_args.alltests)
+                                               cli_args.alltests,
+                                               cli_args.filter_glob)
                exec_result = exec_tests(linux, exec_request)
                parse_request = KunitParseRequest(cli_args.raw_output,
                                                  exec_result.result,
diff --git a/tools/testing/kunit/kunit_kernel.py 
b/tools/testing/kunit/kunit_kernel.py
index 2076a5a2d060..71b1942f5ccd 100644
--- a/tools/testing/kunit/kunit_kernel.py
+++ b/tools/testing/kunit/kunit_kernel.py
@@ -199,8 +199,10 @@ class LinuxSourceTree(object):
                        return False
                return self.validate_config(build_dir)
 
-       def run_kernel(self, args=[], build_dir='', timeout=None) -> 
Iterator[str]:
+       def run_kernel(self, args=[], build_dir='', filter_glob='', 
timeout=None) -> Iterator[str]:
                args.extend(['mem=1G', 'console=tty'])
+               if filter_glob:
+                       args.append('kunit.filter_glob='+filter_glob)
                self._ops.linux_bin(args, timeout, build_dir)
                outfile = get_outfile_path(build_dir)
                subprocess.call(['stty', 'sane'])
-- 
2.30.0.478.g8a0d178c01-goog

Reply via email to