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
1 changed files with 2 additions and 0 deletions

View File

@ -104,6 +104,7 @@ class TrezorTest(unittest.TestCase):
self.pin8 = '45678978'
self.client.wipe_device()
self.client.transport.session_begin()
print("Setup finished")
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')
def tearDown(self):
self.client.transport.session_end()
self.client.close()