|
|
@ -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()
|
|
|
|