mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
e90acfe123
As 'repository_tool' and 'repository_lib' are being deprecated,
hash bin delegation interfaces are no longer available in this
implementation. The example code in this file demonstrates how to
easily implement those interfaces, and how to use them together
with the TUF metadata API, to perform hash bin delegation.
Note, the hash bin delegation logic in this example is largely
copied from repository_{lib, tool}, and modernized and simplified
for this purpose.
Signed-off-by: Lukas Puehringer <lukas.puehringer@nyu.edu>
125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
# Copyright 2020, New York University and the TUF contributors
|
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
|
""" Unit tests for 'examples' scripts.
|
|
|
|
"""
|
|
import glob
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
class TestRepoExamples(unittest.TestCase):
|
|
"""Unit test class for 'repo_example' scripts.
|
|
|
|
Provides a '_run_example_script' method to run (exec) a script located in
|
|
the 'repo_example' directory.
|
|
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
"""Locate and cache 'repo_example' dir."""
|
|
base = Path(__file__).resolve().parents[1]
|
|
cls.repo_examples_dir = base / "examples" / "repo_example"
|
|
|
|
def setUp(self):
|
|
"""Create and change into test dir.
|
|
NOTE: Test scripts are expected to create dirs/files in new CWD."""
|
|
self.original_cwd = os.getcwd()
|
|
self.base_test_dir = os.path.realpath(tempfile.mkdtemp())
|
|
os.chdir(self.base_test_dir)
|
|
|
|
def tearDown(self):
|
|
"""Change back to original dir and remove test dir, which may contain
|
|
dirs/files the test created at test-time CWD."""
|
|
os.chdir(self.original_cwd)
|
|
shutil.rmtree(self.base_test_dir)
|
|
|
|
def _run_script_and_assert_files(self, script_name, filenames_created):
|
|
"""Run script in 'repo_example' dir and assert that it created the
|
|
files corresponding to the passed filenames inside a 'tmp*' test dir at
|
|
CWD."""
|
|
script_path = str(self.repo_examples_dir / script_name)
|
|
with open(script_path, "rb") as f:
|
|
# pylint: disable=exec-used
|
|
exec(
|
|
compile(f.read(), script_path, "exec"),
|
|
{"__file__": script_path},
|
|
)
|
|
|
|
test_dirs = glob.glob("tmp*")
|
|
self.assertTrue(
|
|
len(test_dirs) == 1, f"expected 1 'tmp*' test dir, got {test_dirs}"
|
|
)
|
|
|
|
test_dir = test_dirs.pop()
|
|
for name in filenames_created:
|
|
metadata_path = Path(test_dir) / f"{name}"
|
|
self.assertTrue(
|
|
metadata_path.exists(), f"missing '{metadata_path}' file"
|
|
)
|
|
|
|
def test_basic_repo(self):
|
|
"""Run 'basic_repo.py' and assert creation of metadata files."""
|
|
self._run_script_and_assert_files(
|
|
"basic_repo.py",
|
|
[
|
|
"1.python-scripts.json",
|
|
"1.root.json",
|
|
"1.snapshot.json",
|
|
"1.targets.json",
|
|
"2.root.json",
|
|
"2.snapshot.json",
|
|
"2.targets.json",
|
|
"timestamp.json",
|
|
],
|
|
)
|
|
|
|
def test_hashed_bin_delegation(self):
|
|
"""Run 'hashed_bin_delegation.py' and assert creation of metadata files."""
|
|
self._run_script_and_assert_files(
|
|
"hashed_bin_delegation.py",
|
|
[
|
|
"bins.json",
|
|
"00-07.json",
|
|
"08-0f.json",
|
|
"10-17.json",
|
|
"18-1f.json",
|
|
"20-27.json",
|
|
"28-2f.json",
|
|
"30-37.json",
|
|
"38-3f.json",
|
|
"40-47.json",
|
|
"48-4f.json",
|
|
"50-57.json",
|
|
"58-5f.json",
|
|
"60-67.json",
|
|
"68-6f.json",
|
|
"70-77.json",
|
|
"78-7f.json",
|
|
"80-87.json",
|
|
"88-8f.json",
|
|
"90-97.json",
|
|
"98-9f.json",
|
|
"a0-a7.json",
|
|
"a8-af.json",
|
|
"b0-b7.json",
|
|
"b8-bf.json",
|
|
"c0-c7.json",
|
|
"c8-cf.json",
|
|
"d0-d7.json",
|
|
"d8-df.json",
|
|
"e0-e7.json",
|
|
"e8-ef.json",
|
|
"f0-f7.json",
|
|
"f8-ff.json",
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|