trezor-firmware/ci/prepare_ui_artifacts.py

import hashlib
import json
import shutil
from pathlib import Path


def _hash_files(path):
    files = path.iterdir()
    hasher = hashlib.sha256()
    for file in sorted(files):
        hasher.update(file.read_bytes())

    return hasher.digest().hex()


root = Path(__file__).parent / ".."
screens = root / "tests/ui_tests/screens"
fixtures = root / "tests/ui_tests/fixtures.json"

hashes = json.loads(fixtures.read_text())

for test_case in hashes.keys():
    recorded_dir = screens / test_case / "recorded"
    expected_hash = hashes[test_case]
    actual_hash = _hash_files(recorded_dir)
    assert expected_hash == actual_hash
    shutil.make_archive(root / "ci/ui_test_records" / actual_hash, "zip", recorded_dir)