mirror of
https://github.com/Mbed-TLS/mbedtls-framework.git
synced 2026-10-14 04:37:22 +00:00
Support multiple private key formats in DriverGenerator
Signed-off-by: Gilles Peskine <[email protected]>
This commit is contained in:
@@ -394,8 +394,13 @@ class DriverGenerator(Generator):
|
||||
yield self.one_multipart('verify', key, lengths,
|
||||
tweak_signature=tweak)
|
||||
|
||||
def gen_all(self, multipart: bool = False) -> Iterator[test_case.TestCase]:
|
||||
def gen_all(self,
|
||||
multipart: bool = False,
|
||||
private_key_formats: Optional[Sequence[PrivateKeyFormat]] = None,
|
||||
) -> Iterator[test_case.TestCase]:
|
||||
"""Generate all the tests for this API."""
|
||||
if private_key_formats is not None:
|
||||
self.private_key_formats = private_key_formats
|
||||
yield from super().gen_all()
|
||||
if multipart:
|
||||
for kl in sorted(KEYS.keys()):
|
||||
|
||||
Reference in New Issue
Block a user