mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-18 13:38:12 +00:00
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
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 tests for more models separately,
|
|
# there will always be the other models missing.
|
|
# Therefore, choosing just the cases for our model.
|
|
if len(sys.argv) > 1:
|
|
model = sys.argv[1].upper()
|
|
else:
|
|
model = "TT"
|
|
if model == "TT" and 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
|
|
)
|