| # SPDX-License-Identifier: GPL-2.0 |
| |
| import os |
| import string |
| |
| TEMPLATE_DIR = os.path.dirname(os.path.abspath(__file__)) |
| TEST_TEMPLATE_PATH = os.path.join(TEMPLATE_DIR, 'test_template.c') |
| KCONFIG_TEMPLATE_PATH = os.path.join(TEMPLATE_DIR, 'test_template.Kconfig') |
| MAKEFILE_TEMPLATE_PATH = os.path.join(TEMPLATE_DIR, 'test_template.Makefile') |
| |
| def create_skeleton_from_template(template_path, test_prefix, test_object_file): |
| with open(template_path, 'r') as f: |
| return string.Template(f.read()).safe_substitute( |
| test_prefix=test_prefix, |
| caps_test_prefix=test_prefix.upper(), |
| test_object_file=test_object_file) |
| |
| |
| class Skeletons(object): |
| """ |
| Represents the KUnit skeletons for a test, Kconfig entry, and Makefile |
| entry. |
| """ |
| def __init__(self, test_skeleton, kconfig_skeleton, makefile_skeleton): |
| self.test_skeleton = test_skeleton |
| self.kconfig_skeleton = kconfig_skeleton |
| self.makefile_skeleton = makefile_skeleton |
| |
| |
| def create_skeletons(namespace_prefix, test_object_file): |
| test_prefix = namespace_prefix + '_test' |
| return Skeletons( |
| test_skeleton=create_skeleton_from_template( |
| TEST_TEMPLATE_PATH, |
| test_prefix, |
| test_object_file), |
| kconfig_skeleton=create_skeleton_from_template( |
| KCONFIG_TEMPLATE_PATH, |
| test_prefix, |
| test_object_file), |
| makefile_skeleton=create_skeleton_from_template( |
| MAKEFILE_TEMPLATE_PATH, |
| test_prefix, |
| test_object_file) |
| ) |
| |
| def namespace_prefix_from_path(path): |
| file_name = os.path.basename(path) |
| return os.path.splitext(file_name) |
| |
| def create_skeletons_from_path(path, namespace_prefix=None, print_test_only=False): |
| dir_name, file_name = os.path.split(path) |
| file_prefix, _ = os.path.splitext(file_name) |
| test_path = os.path.join(dir_name, file_prefix + '-test.c') |
| test_object_file = file_prefix + '-test.o' |
| if not namespace_prefix: |
| namespace_prefix = file_prefix.replace('-', '_') |
| skeletons = create_skeletons(namespace_prefix, test_object_file) |
| print('### In ' + test_path) |
| print(skeletons.test_skeleton) |
| if print_test_only: |
| return |
| print('### In Kconfig') |
| print(skeletons.kconfig_skeleton) |
| print('### In Makefile') |
| print(skeletons.makefile_skeleton) |
| |
