mirror/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-01-29 12:32:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Pass KNOWN_TASKS as an argument to main

Browse Source 
Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine 2024-10-03 18:36:09 +02:00
parent 005dca6ad8
commit e41cde57c3
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/scripts/analyze_outcomes.py
View File

@ -761,7 +761,7 @@ KNOWN_TASKS = {
'analyze_block_cipher_dispatch': DriverVSReference_block_cipher_dispatch,
}
def main() -> None:
def main(known_tasks: typing.Dict[str, typing.Type[Task]]) -> None:
main_results = Results()
try:
@ -783,16 +783,16 @@ def main() -> None:
options = parser.parse_args()
if options.list:
for task_name in KNOWN_TASKS:
for task_name in known_tasks:
print(task_name)
sys.exit(0)
if options.specified_tasks == 'all':
tasks_list = list(KNOWN_TASKS.keys())
tasks_list = list(known_tasks.keys())
else:
tasks_list = re.split(r'[, ]+', options.specified_tasks)
for task_name in tasks_list:
if task_name not in KNOWN_TASKS:
if task_name not in known_tasks:
sys.stderr.write('invalid task: {}\n'.format(task_name))
sys.exit(2)
@ -805,7 +805,7 @@ def main() -> None:
sys.exit(2)
task_name = tasks_list[0]
task_class = KNOWN_TASKS[task_name]
task_class = known_tasks[task_name]
if not issubclass(task_class, DriverVSReference):
sys.stderr.write("please provide valid outcomes file for {}.\n".format(task_name))
sys.exit(2)
@ -824,7 +824,7 @@ def main() -> None:
outcomes = read_outcome_file(options.outcomes)
for task_name in tasks_list:
task_constructor = KNOWN_TASKS[task_name]
task_constructor = known_tasks[task_name]
task_instance = task_constructor(options)
main_results.new_section(task_instance.section_name())
task_instance.run(main_results, outcomes)
@ -840,4 +840,4 @@ def main() -> None:
sys.exit(120)
if __name__ == '__main__':
main()
main(KNOWN_TASKS)