arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-18 05:28:40 +00:00
Code Issues Releases Wiki Activity

tests: fix client init

Browse Source
This commit is contained in:
Jan Pochyla 2017-09-04 11:22:41 +02:00
parent 02437d166a
commit c805284a86
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/device_tests/common.py
View File

@ -91,13 +91,19 @@ elif UDP_ENABLED:
DEBUG_TRANSPORT_KWARGS = {}
def get_transport():
return TRANSPORT(*TRANSPORT_ARGS, **TRANSPORT_KWARGS)
def get_debug_transport():
return DEBUG_TRANSPORT(*DEBUG_TRANSPORT_ARGS, **DEBUG_TRANSPORT_KWARGS)
class TrezorTest(unittest.TestCase):
def setUp(self):
transport = TRANSPORT(*TRANSPORT_ARGS, **TRANSPORT_KWARGS)
debug_transport = DEBUG_TRANSPORT(*DEBUG_TRANSPORT_ARGS, **DEBUG_TRANSPORT_KWARGS)
self.client = TrezorClientDebugLink(transport)
self.client.set_debuglink(debug_transport)
self.client = TrezorClientDebugLink(get_transport)
self.client.set_debuglink(get_debug_transport())
self.client.set_tx_api(tx_api.TxApiBitcoin)
# self.client.set_buttonwait(3)