Rework TestGenerator to add file targets
BaseTarget-derived targets are now added to TestGenerator.targets in
initialization. This reduces repeated code in generate_xxx_tests.py
scripts which use this framework.
Signed-off-by: Werner Lewis <werner.lewis@arm.com>
diff --git a/scripts/mbedtls_dev/test_generation.py b/scripts/mbedtls_dev/test_generation.py
index 1adf8e2..57eb2be 100644
--- a/scripts/mbedtls_dev/test_generation.py
+++ b/scripts/mbedtls_dev/test_generation.py
@@ -133,6 +133,11 @@
"""Generate test data."""
def __init__(self, options) -> None:
self.test_suite_directory = getattr(options, 'directory')
+ # Add file Targets which have been declared in other modules
+ self.targets.update({
+ subclass.target_basename: subclass.generate_tests
+ for subclass in BaseTarget.__subclasses__()
+ })
def filename_for(self, basename: str) -> str:
"""The location of the data file with the specified base name."""
@@ -149,7 +154,7 @@
# Note that targets whose names contain 'test_format' have their content
# validated by `abi_check.py`.
- TARGETS = {} # type: Dict[str, Callable[..., Iterable[test_case.TestCase]]]
+ targets = {} # type: Dict[str, Callable[..., Iterable[test_case.TestCase]]]
def generate_target(self, name: str, *target_args) -> None:
"""Generate cases and write to data file for a target.
@@ -157,7 +162,7 @@
For target callables which require arguments, override this function
and pass these arguments using super() (see PSATestGenerator).
"""
- test_cases = self.TARGETS[name](*target_args)
+ test_cases = self.targets[name](*target_args)
self.write_test_data_file(name, test_cases)
def main(args, generator_class: Type[TestGenerator] = TestGenerator):
@@ -170,25 +175,27 @@
parser.add_argument('--directory', default="tests/suites", metavar='DIR',
help='Output directory (default: tests/suites)')
parser.add_argument('targets', nargs='*', metavar='TARGET',
- default=sorted(generator_class.TARGETS),
help='Target file to generate (default: all; "-": none)')
options = parser.parse_args(args)
build_tree.chdir_to_root()
generator = generator_class(options)
if options.list:
- for name in sorted(generator.TARGETS):
+ for name in sorted(generator.targets):
print(generator.filename_for(name))
return
# List in a cmake list format (i.e. ';'-separated)
if options.list_for_cmake:
print(';'.join(generator.filename_for(name)
- for name in sorted(generator.TARGETS)), end='')
+ for name in sorted(generator.targets)), end='')
return
- # Allow "-" as a special case so you can run
- # ``generate_xxx_tests.py - $targets`` and it works uniformly whether
- # ``$targets`` is empty or not.
- options.targets = [os.path.basename(re.sub(r'\.data\Z', r'', target))
- for target in options.targets
- if target != '-']
+ if options.targets:
+ # Allow "-" as a special case so you can run
+ # ``generate_xxx_tests.py - $targets`` and it works uniformly whether
+ # ``$targets`` is empty or not.
+ options.targets = [os.path.basename(re.sub(r'\.data\Z', r'', target))
+ for target in options.targets
+ if target != '-']
+ else:
+ options.targets = sorted(generator.targets)
for target in options.targets:
generator.generate_target(target)