yuzu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2025-02-24 22:46:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extend description generation

Browse source 
Add an extra optional parameter to generate a more detailed description
for test cases.

Signed-off-by: gabor-mezei-arm <gabor.mezei@arm.com>
This commit is contained in:
gabor-mezei-arm 2021-06-24 09:42:02 +02:00
parent 0996080652
commit d71659f447
No known key found for this signature in database
GPG key ID: 106F5A41ECC305BD
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/scripts/generate_psa_tests.py
View file

@ -334,13 +334,15 @@ class StorageFormat:
def key_for_usage_flags( def key_for_usage_flags(
self, self,
usage_flags: List[str], usage_flags: List[str],
short: Optional[str] = None short: Optional[str] = None,
extra_desc: Optional[str] = None
) -> StorageKey: ) -> StorageKey:
"""Construct a test key for the given key usage.""" """Construct a test key for the given key usage."""
usage = ' | '.join(usage_flags) if usage_flags else '0' usage = ' | '.join(usage_flags) if usage_flags else '0'
if short is None: if short is None:
short = re.sub(r'\bPSA_KEY_USAGE_', r'', usage) short = re.sub(r'\bPSA_KEY_USAGE_', r'', usage)
description = 'usage: ' + short extra_desc = ' ' + extra_desc if extra_desc is not None and len(extra_desc) > 0 else ''
description = 'usage' + extra_desc + ': ' + short
return self.key_builder.build(version=self.version, return self.key_builder.build(version=self.version,
id=1, lifetime=0x00000001, id=1, lifetime=0x00000001,
type='PSA_KEY_TYPE_RAW_DATA', bits=8, type='PSA_KEY_TYPE_RAW_DATA', bits=8,
@ -348,14 +350,17 @@ class StorageFormat:
material=b'K', material=b'K',
description=description) description=description)
def all_keys_for_usage_flags(self) -> List[StorageKey]: def all_keys_for_usage_flags(
self,
extra_desc: Optional[str] = None
) -> List[StorageKey]:
"""Generate test keys covering usage flags.""" """Generate test keys covering usage flags."""
known_flags = sorted(self.constructors.key_usage_flags) known_flags = sorted(self.constructors.key_usage_flags)
keys = [] #type List[StorageKey] keys = [] #type List[StorageKey]
keys.append(self.key_for_usage_flags(['0'])) keys.append(self.key_for_usage_flags(['0'], extra_desc=extra_desc))
keys += [self.key_for_usage_flags([usage_flag]) keys += [self.key_for_usage_flags([usage_flag], extra_desc=extra_desc)
for usage_flag in known_flags] for usage_flag in known_flags]
keys += [self.key_for_usage_flags([flag1, flag2]) keys += [self.key_for_usage_flags([flag1, flag2], extra_desc=extra_desc)
for flag1, flag2 in zip(known_flags, for flag1, flag2 in zip(known_flags,
known_flags[1:] + [known_flags[0]])] known_flags[1:] + [known_flags[0]])]
keys.append(self.key_for_usage_flags(known_flags, short='all known')) keys.append(self.key_for_usage_flags(known_flags, short='all known'))