1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 22:18:16 +00:00
trezor-firmware/ci/hardware_tests/device/core.py
2024-02-20 19:49:41 +01:00

31 lines
860 B
Python

from .device import Device
class TrezorCore(Device):
def update_firmware(self, file=None, model_name="Trezor T"):
if not file:
raise ValueError(
"Uploading production firmware will replace the bootloader, it is not allowed!"
)
# reset to enter bootloader again
self.power_off()
self.wait(5)
self.power_on()
self.wait(10)
self.check_model("bootloader")
self.run_trezorctl("device wipe --bootloader || true")
self.wait(5)
self.power_off()
self.power_on()
self.wait(5)
self.log(f"[software] Updating the firmware to {file}")
self.run_trezorctl(f"firmware-update -s -f {file}")
# after firmware-update finishes wait for reboot
self.wait(15)
return self.check_model(model_name)