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

test for ApplySettings.use_passphrase

Browse Source
This commit is contained in:
Pavol Rusnak 2014-12-13 19:07:30 +01:00
parent 74a1e18e4c
commit 63ee124254
3 changed files with 39 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/common.py
View File

@ -30,6 +30,9 @@ class TrezorTest(unittest.TestCase):
def setup_mnemonic_nopin_nopassphrase(self): def setup_mnemonic_nopin_nopassphrase(self):
self.client.load_device_by_mnemonic(mnemonic=self.mnemonic12, pin='', passphrase_protection=False, label='test', language='english') self.client.load_device_by_mnemonic(mnemonic=self.mnemonic12, pin='', passphrase_protection=False, label='test', language='english')
def setup_mnemonic_pin_nopassphrase(self):
self.client.load_device_by_mnemonic(mnemonic=self.mnemonic12, pin=self.pin4, passphrase_protection=False, label='test', language='english')
def setup_mnemonic_pin_passphrase(self): def setup_mnemonic_pin_passphrase(self):
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')

33
tests/test_msg_applysettings.py
View File

@ -5,6 +5,7 @@ import common
from trezorlib import messages_pb2 as proto from trezorlib import messages_pb2 as proto
class TestMsgApplysettings(common.TrezorTest): class TestMsgApplysettings(common.TrezorTest):
def test_apply_settings(self): def test_apply_settings(self):
self.setup_mnemonic_pin_passphrase() self.setup_mnemonic_pin_passphrase()
self.assertEqual(self.client.features.label, 'test') self.assertEqual(self.client.features.label, 'test')
@ -31,5 +32,37 @@ class TestMsgApplysettings(common.TrezorTest):
self.assertEqual(self.client.features.language, 'english') self.assertEqual(self.client.features.language, 'english')
def test_apply_settings_passphrase(self):
self.setup_mnemonic_pin_nopassphrase()
self.assertEqual(self.client.features.passphrase_protection, False)
with self.client:
self.client.set_expected_responses([proto.ButtonRequest(),
proto.PinMatrixRequest(),
proto.Success(),
proto.Features()])
self.client.apply_settings(use_passphrase=True)
self.assertEqual(self.client.features.passphrase_protection, True)
with self.client:
self.client.set_expected_responses([proto.ButtonRequest(),
proto.PinMatrixRequest(),
proto.Success(),
proto.Features()])
self.client.apply_settings(use_passphrase=False)
self.assertEqual(self.client.features.passphrase_protection, False)
with self.client:
self.client.set_expected_responses([proto.ButtonRequest(),
proto.PinMatrixRequest(),
proto.Success(),
proto.Features()])
self.client.apply_settings(use_passphrase=True)
self.assertEqual(self.client.features.passphrase_protection, True)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

4
trezorlib/client.py
View File

@ -414,12 +414,14 @@ class ProtocolMixin(object):
@field('message') @field('message')
@expect(proto.Success) @expect(proto.Success)
def apply_settings(self, label=None, language=None): def apply_settings(self, label=None, language=None, use_passphrase=None):
settings = proto.ApplySettings() settings = proto.ApplySettings()
if label != None: if label != None:
settings.label = label settings.label = label
if language: if language:
settings.language = language settings.language = language
if use_passphrase != None:
settings.use_passphrase = use_passphrase
out = self.call(settings) out = self.call(settings)
self.init_device() # Reload Features self.init_device() # Reload Features