From 0f7f69491453e360210449c968f3a670cafbfc64 Mon Sep 17 00:00:00 2001 From: matejcik Date: Tue, 2 Oct 2018 17:18:38 +0200 Subject: [PATCH] debug: fix test_msg_applysettings --- .../device_tests/test_msg_applysettings.py | 75 +++++-------------- 1 file changed, 17 insertions(+), 58 deletions(-) diff --git a/trezorlib/tests/device_tests/test_msg_applysettings.py b/trezorlib/tests/device_tests/test_msg_applysettings.py index 050277d48..4ae177632 100644 --- a/trezorlib/tests/device_tests/test_msg_applysettings.py +++ b/trezorlib/tests/device_tests/test_msg_applysettings.py @@ -21,6 +21,15 @@ import pytest from trezorlib import device, messages as proto from .common import TrezorTest +from .conftest import TREZOR_VERSION + +EXPECTED_RESPONSES_NOPIN = [proto.ButtonRequest(), proto.Success(), proto.Features()] +EXPECTED_RESPONSES_PIN = [proto.PinMatrixRequest()] + EXPECTED_RESPONSES_NOPIN + +if TREZOR_VERSION >= 2: + EXPECTED_RESPONSES = EXPECTED_RESPONSES_NOPIN +else: + EXPECTED_RESPONSES = EXPECTED_RESPONSES_PIN class TestMsgApplysettings(TrezorTest): @@ -29,16 +38,7 @@ class TestMsgApplysettings(TrezorTest): assert self.client.features.label == "test" with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) - if self.client.features.major_version >= 2: - self.client.expected_responses.pop(0) # skip PinMatrixRequest + self.client.set_expected_responses(EXPECTED_RESPONSES) device.apply_settings(self.client, label="new label") assert self.client.features.label == "new label" @@ -49,14 +49,7 @@ class TestMsgApplysettings(TrezorTest): assert self.client.features.language == "english" with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES) device.apply_settings(self.client, language="nonexistent") assert self.client.features.language == "english" @@ -67,32 +60,19 @@ class TestMsgApplysettings(TrezorTest): assert self.client.features.passphrase_protection is False with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) - if self.client.features.major_version >= 2: - self.client.expected_responses.pop(0) # skip PinMatrixRequest + self.client.set_expected_responses(EXPECTED_RESPONSES) device.apply_settings(self.client, use_passphrase=True) assert self.client.features.passphrase_protection is True with self.client: - self.client.set_expected_responses( - [proto.ButtonRequest(), proto.Success(), proto.Features()] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES_NOPIN) device.apply_settings(self.client, use_passphrase=False) assert self.client.features.passphrase_protection is False with self.client: - self.client.set_expected_responses( - [proto.ButtonRequest(), proto.Success(), proto.Features()] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES_NOPIN) device.apply_settings(self.client, use_passphrase=True) assert self.client.features.passphrase_protection is True @@ -104,14 +84,7 @@ class TestMsgApplysettings(TrezorTest): img = b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"\x00\x00\x00\x00\x04\x80\x00\x00\x00\x00\x00\x00\x00\x00\x04\x88\x02\x00\x00\x00\x02\x91\x00\x00\x00\x00\x00\x00\x80\x00\x00\x00\x00\x90@\x00\x11@\x00\x00\x00\x00\x00\x00\x08\x00\x10\x92\x12\x04\x00\x00\x05\x12D\x00\x00\x00\x00\x00 \x00\x00\x08\x00Q\x00\x00\x02\xc0\x00\x00\x00\x00\x00\x00\x00\x10\x02 \x01\x04J\x00)$\x00\x00\x00\x00\x80\x00\x00\x00\x00\x08\x10\xa1\x00\x00\x02\x81 \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x08\x00\x00\tP\x00\x00\x00\x00\x00\x00 \x00\x00\xa0\x00\xa0R \x12\x84\x80\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\t\x08\x00\tP\x00\x00\x00\x00 \x00\x04 \x00\x80\x02\x00@\x02T\xc2 \x00\x00\x00\x00\x00\x00\x00\x10@\x00)\t@\n\xa0\x80\x00\x00\x00\x00\x00\x00\x00\x00\x10\x00\x80@\x14\xa9H\x04\x00\x00\x88@\x00\x00\x00\x00\x00\x02\x02$\x00\x15B@\x00\nP\x00\x00\x00\x00\x00\x80\x00\x00\x91\x01UP\x00\x00 \x02\x00\x00\x00\x00\x00\x00\x02\x08@ Z\xa5 \x00\x00\x80\x00\x00\x00\x00\x00\x00\x08\xa1%\x14*\xa0\x00\x00\x02\x08\x00\x00\x00\x00\x00\x00\x00\x00\x00@\xaa\x91 \x00\x05E\x80\x00\x00\x00\x00\x00\x02*T\x05-D\x00\x00\x05 @\x00\x00\x00\x00\x00%@\x80\x11V\xa0\x88\x00\x05@\xb0\x00\x00\x00\x00\x00\x818$\x04\xabD \x00\x06\xa1T\x00\x00\x00\x00\x02\x03\xb8\x01R\xd5\x01\x00\x00\x05AP\x00\x00\x00\x00\x08\xadT\x00\x05j\xa4@\x00\x87ah\x00\x00\x00\x00\x02\x8d\xb8\x08\x00.\x01\x00\x00\x02\xa5\xa8\x10\x00\x00\x00*\xc1\xec \n\xaa\x88 \x02@\xf6\xd0\x02\x00\x00\x00\x0bB\xb6\x14@U"\x80\x00\x01{`\x00\x00\x00\x00M\xa3\xf8 \x15*\x00\x00\x00\x10n\xc0\x04\x00\x00\x02\x06\xc2\xa8)\x00\x96\x84\x80\x00\x00\x1b\x00\x00\x80@\x10\x87\xa7\xf0\x84\x10\xaa\x10\x00\x00D\x00\x00\x02 \x00\x8a\x06\xfa\xe0P\n-\x02@\x00\x12\x00\x00\x00\x00\x10@\x83\xdf\xa0\x00\x08\xaa@\x00\x00\x01H\x00\x05H\x04\x12\x01\xf7\x81P\x02T\t\x00\x00\x00 \x00\x00\x84\x10\x00\x00z\x00@)* \x00\x00\x01\n\xa0\x02 \x05\n\x00\x00\x05\x10\x84\xa8\x84\x80\x00\x00@\x14\x00\x92\x10\x80\x00\x04\x11@\tT\x00\x00\x00\x00\n@\x00\x08\x84@$\x00H\x00\x12Q\x02\x00\x00\x00\x00\x90\x02A\x12\xa8\n\xaa\x92\x10\x04\xa8\x10@\x00\x00\x04\x04\x00\x04I\x00\x04\x14H\x80"R\x01\x00\x00\x00!@\x00\x00$\xa0EB\x80\x08\x95hH\x00\x00\x00\x84\x10 \x05Z\x00\x00(\x00\x02\x00\xa1\x01\x00\x00\x04\x00@\x82\x00\xadH*\x92P\x00\xaaP\x00\x00\x00\x00\x11\x02\x01*\xad\x01\x00\x01\x01"\x11D\x08\x00\x00\x10\x80 \x00\x81W\x80J\x94\x04\x08\xa5 !\x00\x00\x00\x02\x00B*\xae\xa1\x00\x80\x10\x01\x08\xa4\x00\x00\x00\x00\x00\x84\x00\t[@"HA\x04E\x00\x84\x00\x00\x00\x10\x00\x01J\xd5\x82\x90\x02\x00!\x02\xa2\x00\x00\x00\x00\x00\x00\x00\x05~\xa0\x00 \x10\n)\x00\x11\x00\x00\x00\x00\x00\x00!U\x80\xa8\x88\x82\x80\x01\x00\x00\x00\x00\x00\x00H@\x11\xaa\xc0\x82\x00 *\n\x00\x00\x00\x00\x00\x00\x00\x00\n\xabb@ \x04\x00! \x84\x00\x00\x00\x00\x02@\xa5\x15A$\x04\x81(\n\x00\x00\x00\x00\x00\x00 \x01\x10\x02\xe0\x91\x02\x00\x00\x04\x00\x00\x00\x00\x00\x00\x01 \xa9\tQH@\x91 P\x00\x00\x00\x00\x00\x00\x08\x00\x00\xa0T\xa5\x00@\x80\x08\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"\x00\x00\x00\x00\xa2\x00\x00\x00\x00\x00\x00\x00\x08\x00\x00 T\xa0\t\x00\x08\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00@\x02\xa0\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00*\x10\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x10\x00\x00\x10\x02\x80\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\t\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x08\x00\x00@\x04\x80\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00@\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x08@\x10\x00\x00\x00\x00' with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES) device.apply_settings(self.client, homescreen=img) @pytest.mark.skip_t2 @@ -119,14 +92,7 @@ class TestMsgApplysettings(TrezorTest): self.setup_mnemonic_pin_passphrase() with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES_PIN) device.apply_settings(self.client, auto_lock_delay_ms=int(10e3)) # 10 secs time.sleep(0.1) # sleep less than auto-lock delay @@ -151,14 +117,7 @@ class TestMsgApplysettings(TrezorTest): self.setup_mnemonic_pin_passphrase() with self.client: - self.client.set_expected_responses( - [ - proto.PinMatrixRequest(), - proto.ButtonRequest(), - proto.Success(), - proto.Features(), - ] - ) + self.client.set_expected_responses(EXPECTED_RESPONSES_PIN) # Note: the actual delay will be 10 secs (see above). device.apply_settings(self.client, auto_lock_delay_ms=int(1e3))