tests: Simplify test data generation

We always want to either verify or generate new results:
don't have multiple arguments.

Also fix annotated types.

Signed-off-by: Jussi Kukkonen <jkukkonen@google.com>
This commit is contained in:
Jussi Kukkonen 2024-12-10 20:41:58 +02:00
parent 31bb232ca3
commit ec81bfa0b1
2 changed files with 8 additions and 11 deletions

View file

@ -77,15 +77,13 @@ def verify_generation(md: Metadata, path: str) -> None:
)
def generate_all_files(
dump: bool | None = False, verify: bool | None = False
) -> None:
"""Generate a new repository and optionally verify it.
def generate_all_files(dump: bool = False) -> None:
"""Generate a new repository or verify that output has not changed.
Args:
dump: Wheter to dump the newly generated files.
verify: Whether to verify the newly generated files with the
local staored.
dump: If True, new files are generated. If False, existing files
are compared to generated files and an exception is raised if
there are differences.
"""
md_root = Metadata(Root(expires=EXPIRY))
md_timestamp = Metadata(Timestamp(expires=EXPIRY))
@ -106,11 +104,10 @@ def generate_all_files(
"ed25519_metadata",
f"{md.signed.type}_with_ed25519.json",
)
if verify:
verify_generation(md, path)
if dump:
md.to_file(path, SERIALIZER)
else:
verify_generation(md, path)
if __name__ == "__main__":

View file

@ -16,7 +16,7 @@ class TestMetadataGeneration(unittest.TestCase):
@staticmethod
def test_compare_static_md_to_generated() -> None:
# md_generator = MetadataGenerator("generated_data/ed25519_metadata")
generate_all_files(dump=False, verify=True)
generate_all_files(dump=False)
# Run unit test.