arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 11:39:03 +00:00
Code Issues Releases Wiki Activity

Test of PIN cancel function

Browse Source
This commit is contained in:
slush0 2013-10-10 17:18:02 +02:00
parent 4e5c3740aa
commit b8ddd0279a
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_protect_call.py
View File

@ -27,5 +27,9 @@ class TestProtectCall(common.TrezorTest):
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)
if __name__ == '__main__':
unittest.main()

5
trezorlib/client.py
View File

@ -102,11 +102,14 @@ class TrezorClient(object):
if isinstance(resp, proto.PinMatrixRequest):
if self.debuglink:
if self.debug_pin:
if self.debug_pin == 1:
pin = self.debuglink.read_pin_encoded()
msg2 = proto.PinMatrixAck(pin=pin)
elif self.debug_pin == -1:
msg2 = proto.PinMatrixCancel()
else:
msg2 = proto.PinMatrixAck(pin='444444222222')
else:
pin = self.pin_func("PIN required: ", resp.message)
msg2 = proto.PinMatrixAck(pin=pin)