1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-18 12:28:09 +00:00

device_tests: wrap tests in a session

This commit is contained in:
Jan Pochyla 2017-10-31 17:04:27 +01:00
parent fcd793e6e4
commit cc9ddcbc12

View File

@ -104,6 +104,7 @@ class TrezorTest(unittest.TestCase):
self.pin8 = '45678978' self.pin8 = '45678978'
self.client.wipe_device() self.client.wipe_device()
self.client.transport.session_begin()
print("Setup finished") print("Setup finished")
print("--------------") print("--------------")
@ -121,6 +122,7 @@ class TrezorTest(unittest.TestCase):
self.client.load_device_by_mnemonic(mnemonic=self.mnemonic12, pin=self.pin4, passphrase_protection=True, label='test', language='english') self.client.load_device_by_mnemonic(mnemonic=self.mnemonic12, pin=self.pin4, passphrase_protection=True, label='test', language='english')
def tearDown(self): def tearDown(self):
self.client.transport.session_end()
self.client.close() self.client.close()