1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-30 03:18:20 +00:00
trezor-firmware/tests/ui_tests/reporting/download.py
matejcik fbc2c828ec fix(tests): do not place recorded.zip into the destdir
instead, download it to a temp location, as the Prophecy (and authors of
urlretrieve) foretold
2024-11-12 16:55:17 +01:00

34 lines
924 B
Python

from __future__ import annotations
import urllib.error
import urllib.request
import zipfile
from pathlib import Path
from typing import Any
import requests
RECORDS_WEBSITE = "https://data.trezor.io/dev/firmware/ui_tests/"
FIXTURES_MASTER = "https://raw.githubusercontent.com/trezor/trezor-firmware/master/tests/ui_tests/fixtures.json"
FIXTURES_CURRENT = Path(__file__).resolve().parent.parent / "fixtures.json"
def fetch_recorded(hash: str, path: Path) -> None:
zip_src = RECORDS_WEBSITE + hash + ".zip"
try:
dest, _ = urllib.request.urlretrieve(zip_src)
except urllib.error.HTTPError:
raise RuntimeError(f"No such recorded collection was found on '{zip_src}'.")
with zipfile.ZipFile(dest, "r") as z:
z.extractall(path)
Path(dest).unlink()
def fetch_fixtures_master() -> dict[str, Any]:
r = requests.get(FIXTURES_MASTER)
r.raise_for_status()
return r.json()