From 513e6aae086b9ea62828ee4abdd0988b60c38129 Mon Sep 17 00:00:00 2001
From: matejcik <ja@matejcik.cz>
Date: Fri, 2 Mar 2018 18:25:39 +0100
Subject: [PATCH] better way for test suite to search for the right device,
 that also respects TREZOR_PATH

---
 trezorlib/tests/device_tests/common.py | 77 +++-----------------------
 1 file changed, 7 insertions(+), 70 deletions(-)

diff --git a/trezorlib/tests/device_tests/common.py b/trezorlib/tests/device_tests/common.py
index e702fdd99..0fffb7ae9 100644
--- a/trezorlib/tests/device_tests/common.py
+++ b/trezorlib/tests/device_tests/common.py
@@ -23,86 +23,23 @@ import pytest
 import os
 
 from trezorlib.client import TrezorClient, TrezorClientDebugLink
+from trezorlib.transport import get_transport
 from trezorlib import tx_api
 
 tests_dir = os.path.dirname(os.path.abspath(__file__))
 tx_api.cache_dir = os.path.join(tests_dir, '../txcache')
 
 
-try:
-    from trezorlib.transport_hid import HidTransport
-    HID_ENABLED = True
-except ImportError as e:
-    print('HID transport disabled:', e)
-    HID_ENABLED = False
+def get_device():
+    path = os.environ.get('TREZOR_PATH')
+    return get_transport(path)
 
-try:
-    from trezorlib.transport_webusb import WebUsbTransport
-    WEBUSB_ENABLED = True
-except ImportError as e:
-    print('WebUSB transport disabled:', e)
-    WEBUSB_ENABLED = False
 
-try:
-    from trezorlib.transport_pipe import PipeTransport
-    PIPE_ENABLED = True
-except ImportError as e:
-    print('PIPE transport disabled:', e)
-    PIPE_ENABLED = False
-
-try:
-    from trezorlib.transport_udp import UdpTransport
-    UDP_ENABLED = True
-except ImportError as e:
-    print('UDP transport disabled:', e)
-    UDP_ENABLED = False
-
-
-def pipe_exists(path):
-    import os
-    import stat
-    try:
-        return stat.S_ISFIFO(os.stat(path).st_mode)
-    except:
-        return False
-
-
-def get_transport():
-    if HID_ENABLED and HidTransport.enumerate():
-        devices = HidTransport.enumerate()
-        wirelink = devices[0]
-        debuglink = devices[0].find_debug()
-
-    elif WEBUSB_ENABLED and WebUsbTransport.enumerate():
-        devices = WebUsbTransport.enumerate()
-        wirelink = devices[0]
-        debuglink = devices[0].find_debug()
-
-    elif PIPE_ENABLED and pipe_exists('/tmp/pipe.trezor.to'):
-        wirelink = PipeTransport('/tmp/pipe.trezor', False)
-        debuglink = PipeTransport('/tmp/pipe.trezor_debug', False)
-
-    elif UDP_ENABLED:
-        wirelink = UdpTransport('127.0.0.1:21324')
-        debuglink = UdpTransport('127.0.0.1:21325')
-
-    return wirelink, debuglink
-
-
-if HID_ENABLED and HidTransport.enumerate():
-    print('Using TREZOR')
-elif WEBUSB_ENABLED and WebUsbTransport.enumerate():
-    print('Using TREZOR via WebUSB')
-elif PIPE_ENABLED and pipe_exists('/tmp/pipe.trezor.to'):
-    print('Using Emulator (v1=pipe)')
-elif UDP_ENABLED:
-    print('Using Emulator (v2=udp)')
-
-
-class TrezorTest(object):
+class TrezorTest:
 
     def setup_method(self, method):
-        wirelink, debuglink = get_transport()
+        wirelink = get_device()
+        debuglink = wirelink.find_debug()
         self.client = TrezorClientDebugLink(wirelink)
         self.client.set_debuglink(debuglink)
         self.client.set_tx_api(tx_api.TxApiBitcoin)