arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-16 17:42:02 +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 pytest
from trezorlib.transport import get_transport
from trezorlib.transport import get_transport, enumerate_devices
from trezorlib.client import TrezorClient, TrezorClientDebugLink
from trezorlib import log, coins
TREZOR_VERSION = None
def get_device():
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():
@ -39,9 +48,6 @@ def device_version():
return 1
TREZOR_VERSION = device_version()
@pytest.fixture(scope="function")
def client():
wirelink = get_device()
@ -86,6 +92,9 @@ def setup_client(mnemonic=None, pin="", passphrase=False):
def pytest_configure(config):
global TREZOR_VERSION
TREZOR_VERSION = device_version()
if config.getoption("verbose"):
log.enable_debug_output()