trezor-firmware/ci/prepare_ui_artifacts.py

import os
import shutil
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(ROOT))
# Needed for setup purposes, filling the FILE_HASHES dict
from tests.ui_tests import read_fixtures  # isort:skip

read_fixtures()
from tests.ui_tests import _hash_files, FILE_HASHES, SCREENS_DIR  # isort:skip

# As in CI we are running T1 and TT tests separately, there will
# always be the other model missing.
# Therefore, choosing just the cases for our model.
if len(sys.argv) > 1 and sys.argv[1].upper() == "T1":
    model = "T1"
else:
    model = "TT"
if os.getenv("UI2") == "1":
    model += "ui2"
model_file_hashes = {k: v for k, v in FILE_HASHES.items() if k.startswith(f"{model}_")}

for test_case, expected_hash in model_file_hashes.items():
    recorded_dir = SCREENS_DIR / test_case / "recorded"
    actual_hash = _hash_files(recorded_dir)
    assert expected_hash == actual_hash
    shutil.make_archive(
        str(ROOT / "ci/ui_test_records" / actual_hash), "zip", recorded_dir
    )