trezor-firmware/ci/hardware_tests/device/device.py

import datetime
import sys
import time
from subprocess import run


class Device:
    def __init__(self, uhub_location, device_port):
        self.uhub_location = uhub_location
        self.device_port = device_port

    @staticmethod
    def log(msg):
        print(msg, flush=True, file=sys.stderr)

    def run_trezorctl(self, cmd: str, **kwargs):
        full_cmd = "trezorctl "
        full_cmd += cmd
        self.log("[software/trezorctl] Running '{}'".format(full_cmd))
        return run(full_cmd, shell=True, check=True, **kwargs)

    def check_model(self, model=None):
        res = self.run_trezorctl("list", capture_output=True, text=True)
        self.log(res.stdout)
        self.log(res.stderr)
        self.run_trezorctl("get-features | grep version")
        lines = res.stdout.splitlines()
        if len(lines) != 1:
            raise RuntimeError("{} trezors connected".format(len(lines)))
        if model and model not in lines[0]:
            raise RuntimeError(
                "invalid trezor model connected (expected {})".format(model)
            )
        return lines[0].split()[0]

    def reboot(self):
        self.power_off()
        self.power_on()

    def power_on(self):
        self.now()
        self.log("[hardware/usb] Turning power on...")
        run(
            "uhubctl -l {} -p {} -a on".format(self.uhub_location, self.device_port),
            shell=True,
            check=True,
        )
        self.wait(3)

    def power_off(self):
        self.now()
        self.log("[hardware/usb] Turning power off...")
        run(
            "uhubctl -l {} -p {} -r 100 -a off".format(
                self.uhub_location, self.device_port
            ),
            shell=True,
            check=True,
        )
        self.wait(3)

    def touch(self, location, action):
        raise NotImplementedError

    @staticmethod
    def wait(seconds):
        Device.now()
        Device.log("[software] Waiting for {} seconds...".format(seconds))
        time.sleep(seconds)

    @staticmethod
    def now():
        Device.log("\n[timestamp] {}".format(datetime.datetime.now()))