import time import unittest import common from trezorlib import messages_pb2 as proto from trezorlib import types_pb2 as types from trezorlib.client import PinException # FIXME TODO Add passphrase tests class TestProtectCall(common.TrezorTest): def _some_protected_call(self, expected_buttonrequests): # This method perform any call which have protection in the device res = self.client.ping('random data', pin_protection=True, passphrase_protection=True) self.assertEqual(res, 'random data') msg = proto.Ping(message='random data', button_protection=True, pin_protection=True, passphrase_protection=True) return self.client.call(msg, expected=proto.Success, expected_buttonrequests=expected_buttonrequests).message def test_no_protection(self): self.client.wipe_device() self.client.load_device_by_mnemonic( mnemonic=self.mnemonic1, pin='', passphrase_protection=False, label='test', language='english', ) self.assertEqual(self.client.debuglink.read_pin()[0], '') self._some_protected_call([]) ''' def test_pin(self): self.client.wipe_device() self.client.load_device_by_mnemonic(mnemonic=self.mnemonic1, pin=self.pin2, passphrase_protection=False, label='test', language='english') self.assertEqual(self.client.debuglink.read_pin()[0], self.pin2) self._some_protected_call() def test_incorrect_pin(self): self.client.setup_debuglink(button=True, pin_correct=False) self.assertRaises(PinException, self._some_protected_call) def test_cancelled_pin(self): self.client.setup_debuglink(button=True, pin_correct=-1) # PIN cancel self.assertRaises(PinException, self._some_protected_call) def test_exponential_backoff_with_reboot(self): self.client.setup_debuglink(button=True, pin_correct=False) def test_backoff(attempts, start): expected = 1.8 ** attempts got = time.time() - start msg = "Pin delay expected to be at least %s seconds, got %s" % (expected, got) print msg self.assertLessEqual(expected, got, msg) for attempt in range(1, 4): start = time.time() self.assertRaises(PinException, self._some_protected_call) test_backoff(attempt, start) ''' ''' # Unplug Trezor now self.client.debuglink.stop() self.client.close() # Give it some time to reboot (it may take some time on RPi) boot_delay = 20 time.sleep(boot_delay) # Connect to Trezor again start = time.time() self.setUp() expected = 1.8 ** attempt / 2 # This test isn't accurate, let's expect at least some delay took = time.time() - start print "Expected reboot time at least %s seconds" % expected print "Rebooted in %s seconds" % took self.assertLessEqual(expected, time.time() - start, "Bootup took %s seconds, expected %s seconds or more!" % (took, expected)) ''' if __name__ == '__main__': unittest.main()