Support multiple private key formats in DriverGenerator

Signed-off-by: Gilles Peskine <[email protected]>
This commit is contained in:
Gilles Peskine
2026-07-08 18:26:05 +02:00
parent 73b5c8ee6c
commit 98fc1f179b
@@ -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()):