mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
d8b23f1d55
Following parallel merges of #1700 (added new test method), and #1710 (started running mypy on tests), ci/cd fails in the develop branch. This is fixed in this patch. Signed-off-by: Lukas Puehringer <lukas.puehringer@nyu.edu>
134 lines
4.1 KiB
Python
134 lines
4.1 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 sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from typing import ClassVar, List
|
|
|
|
from tests import utils
|
|
|
|
|
|
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.
|
|
|
|
"""
|
|
|
|
repo_examples_dir: ClassVar[Path]
|
|
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
"""Locate and cache 'repo_example' dir."""
|
|
base = Path(__file__).resolve().parents[1]
|
|
cls.repo_examples_dir = base / "examples" / "repo_example"
|
|
|
|
def setUp(self) -> None:
|
|
"""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) -> None:
|
|
"""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: str, filenames_created: List[str]
|
|
) -> None:
|
|
"""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) -> None:
|
|
"""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) -> None:
|
|
"""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__":
|
|
utils.configure_test_logging(sys.argv)
|
|
unittest.main()
|