arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-21 12:02:19 +00:00
Code Issues Releases Wiki Activity

device_tests: smarter device selection

Browse Source 
skips bridge, dies somewhat more cleanly when no device found
This commit is contained in:
matejcik 2018-08-21 16:06:18 +02:00
parent abf0e82d80
commit e2d45397d4
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
trezorlib/tests/device_tests/conftest.py
View File

@ -18,14 +18,23 @@ import functools
import os import os
import pytest import pytest
from trezorlib.transport import get_transport from trezorlib.transport import get_transport, enumerate_devices
from trezorlib.client import TrezorClient, TrezorClientDebugLink from trezorlib.client import TrezorClient, TrezorClientDebugLink
from trezorlib import log, coins from trezorlib import log, coins
TREZOR_VERSION = None
def get_device(): def get_device():
path = os.environ.get("TREZOR_PATH") path = os.environ.get("TREZOR_PATH")
return get_transport(path) if path:
return get_transport(path)
else:
devices = enumerate_devices()
for device in devices:
if hasattr(device, "find_debug"):
return device
raise RuntimeError("No debuggable device found")
def device_version(): def device_version():
@ -39,9 +48,6 @@ def device_version():
return 1 return 1
TREZOR_VERSION = device_version()
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def client(): def client():
wirelink = get_device() wirelink = get_device()
@ -86,6 +92,9 @@ def setup_client(mnemonic=None, pin="", passphrase=False):
def pytest_configure(config): def pytest_configure(config):
global TREZOR_VERSION
TREZOR_VERSION = device_version()
if config.getoption("verbose"): if config.getoption("verbose"):
log.enable_debug_output() log.enable_debug_output()