From a4102cb57d87ad0aa4bed52fc9c579d3506c1bf7 Mon Sep 17 00:00:00 2001 From: gabor-mezei-arm Date: Thu, 24 Jun 2021 09:53:26 +0200 Subject: [PATCH] Use different subslasses for the test data files Signed-off-by: gabor-mezei-arm --- tests/scripts/generate_psa_tests.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/tests/scripts/generate_psa_tests.py b/tests/scripts/generate_psa_tests.py index eeffcebb2d..af1cb533ec 100644 --- a/tests/scripts/generate_psa_tests.py +++ b/tests/scripts/generate_psa_tests.py @@ -461,6 +461,17 @@ class StorageFormat: return [self.make_test_case(key) for key in keys] +class StorageFormatForward(StorageFormat): + """Storage format stability test cases for forward compatibility.""" + + def __init__(self, info: Information, version: int) -> None: + super().__init__(info, version, True) + +class StorageFormatV0(StorageFormat): + """Storage format stability test cases for version 0 compatibility.""" + + def __init__(self, info: Information) -> None: + super().__init__(info, 0, False) class TestGenerator: """Generate test data.""" @@ -492,9 +503,9 @@ class TestGenerator: 'test_suite_psa_crypto_not_supported.generated': lambda info: NotSupported(info).test_cases_for_not_supported(), 'test_suite_psa_crypto_storage_format.current': - lambda info: StorageFormat(info, 0, True).all_test_cases(), + lambda info: StorageFormatForward(info, 0).all_test_cases(), 'test_suite_psa_crypto_storage_format.v0': - lambda info: StorageFormat(info, 0, False).all_test_cases(), + lambda info: StorageFormatV0(info).all_test_cases(), } #type: Dict[str, Callable[[Information], Iterable[test_case.TestCase]]] def generate_target(self, name: str) -> None: