mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
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:
parent
31bb232ca3
commit
ec81bfa0b1
2 changed files with 8 additions and 11 deletions
|
|
@ -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__":
|
||||
|
|
|
|||
|
|
@ -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.
|
||||
|
|
|
|||
Loading…
Reference in a new issue