1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 15:38:11 +00:00
trezor-firmware/ci/hardware_tests/device/tt.py
2021-10-13 11:53:17 +02:00

30 lines
821 B
Python

from .device import Device
class TrezorT(Device):
def update_firmware(self, file=None):
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.power_on()
self.wait(5)
self.check_model("Trezor T 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("Trezor T")