merge test-user-mode and test-baremetal into test-executables

Differentiate with --mode userland vs --mode baremetal. The code was basically copied, now it's DRY.
2026-01-22 17:55:57 +01:00 · 2019-05-29 00:00:01 +00:00
parent 6994dc21af
commit fedb68c62c
11 changed files with 84 additions and 130 deletions
--- a/90
+++ b/90
@@ -0,0 +1,90 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+
+import common
+import lkmc.import_path
+import path_properties
+import thread_pool
+
+class Main(common.TestCliFunction):
+    def __init__(self, *args, **kwargs):
+        if not 'description' in kwargs:
+            kwargs['description'] = '''\
+Test userland executables in user mode, or baremetal executables in full system
+depending on the value of the --mode option. See also:
+
+* https://github.com/cirosantilli/linux-kernel-module-cheat#user-mode-tests
+* https://github.com/cirosantilli/linux-kernel-module-cheat#baremetal-tests
+* https://github.com/cirosantilli/linux-kernel-module-cheat#userland-setup-getting-started-natively
+'''
+        if not 'defaults' in kwargs:
+            kwargs['defaults'] = {}
+        if not 'mode' in kwargs['defaults']:
+            kwargs['defaults']['mode'] = 'userland'
+        super().__init__(*args, **kwargs)
+        self.add_argument(
+            'tests',
+            nargs='*',
+            help='''\
+If given, run only the given tests. Otherwise, run all tests.
+'''
+        )
+
+    def setup_one(self):
+        self.env['tests'] = self.resolve_targets(
+            [
+                self.env['baremetal_source_dir'],
+                self.env['userland_source_dir']
+            ],
+            self.env['tests']
+        )
+
+    def timed_main(self):
+        run_args = self.get_common_args()
+        if self.env['mode'] == 'userland' and self.env['emulator'] == 'gem5':
+            run_args['userland_build_id'] = 'static'
+        rootdir_abs_len = len(self.env['root_dir'])
+        with thread_pool.ThreadPool(
+            self.run_test,
+            handle_output=self.handle_output_function,
+            nthreads=self.env['nproc'],
+            thread_id_arg='thread_id',
+            submit_raise_exit=self.env['quit_on_fail'],
+        ) as my_thread_pool:
+            for test in self.env['tests']:
+                for path, in_dirnames, in_filenames in self.sh.walk(test):
+                    path_abs = os.path.abspath(path)
+                    dirpath_relative_root = path_abs[rootdir_abs_len + 1:]
+                    for in_filename in in_filenames:
+                        if os.path.splitext(in_filename)[1] in self.env['build_in_exts']:
+                            path_relative_root = os.path.join(dirpath_relative_root, in_filename)
+                            my_path_properties = path_properties.get(path_relative_root)
+                            if my_path_properties.should_be_tested(self.env):
+                                cur_run_args = run_args.copy()
+                                cur_run_args.update({
+                                    self.env['mode']: os.path.relpath(
+                                        os.path.join(path_abs, in_filename),
+                                        os.getcwd()
+                                    ),
+                                })
+                                cur_run_args.update(my_path_properties['test_run_args'])
+                                run_test_args = {
+                                    'expected_exit_status': my_path_properties['exit_status'],
+                                    'run_args': cur_run_args,
+                                    'run_obj': lkmc.import_path.import_path_main('run'),
+                                    'test_id': '{} {}'.format(self.env['mode'], path_relative_root),
+                                }
+                                signal = my_path_properties['signal_received']
+                                if signal is not None:
+                                    if self.env['mode'] == 'baremetal':
+                                        run_test_args['expected_exit_status'] = 128 + signal.value
+                                    elif self.env['mode'] == 'userland':
+                                        # Python subprocess reports signals differently from Bash's 128 + signal rule.
+                                        run_test_args['expected_exit_status'] = -signal.value
+                                my_thread_pool.submit(run_test_args)
+        return self._handle_thread_pool_errors(my_thread_pool)
+
+if __name__ == '__main__':
+    Main().cli()