mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-21 05:48:23 +00:00
4264e87319
This enables to run device tests against an actual device connected to our CI. It uses https://github.com/mmahut/tpmb to enter bootloader and then it uploads a debug firmware to the device. The debug mode is the used to "click" the buttons but we might improve this and actually use the hardware buttons instead.
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import os
|
|
|
|
from .device import Device
|
|
|
|
|
|
class TrezorOne(Device):
|
|
def touch(self, location, action):
|
|
self.now()
|
|
print(
|
|
"[hardware/trezor] Touching the {} button by {}...".format(location, action)
|
|
)
|
|
self.serial.write(("{} {}\n".format(location, action)).encode())
|
|
|
|
def update_firmware(self, file=None):
|
|
if file:
|
|
unofficial = True
|
|
trezorctlcmd = "trezorctl firmware-update -s -f {} &".format(file)
|
|
print("[software/trezorctl] Updating the firmware to {}...".format(file))
|
|
else:
|
|
unofficial = False
|
|
trezorctlcmd = "trezorctl firmware-update &"
|
|
print("[software/trezorctl] Updating the firmware to latest...")
|
|
self._enter_bootloader()
|
|
|
|
os.system(trezorctlcmd)
|
|
self.wait(3)
|
|
self.touch("right", "click")
|
|
self.wait(20)
|
|
if unofficial:
|
|
self.touch("right", "click")
|
|
self.wait(10)
|
|
self.power_off()
|
|
self.power_on()
|
|
if unofficial:
|
|
self.touch("right", "click")
|
|
self.wait(5)
|
|
self.touch("right", "click")
|
|
self.wait(5)
|
|
os.system("trezorctl get-features|grep version")
|
|
|
|
def _enter_bootloader(self):
|
|
self.power_off()
|
|
self.touch("all", "press")
|
|
self.wait(2)
|
|
self.power_on()
|
|
self.wait(2)
|
|
self.touch("all", "unpress")
|