From b7f5849ecca928e30438091bae5bd4fd90d5ba34 Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Mon, 28 Apr 2014 14:39:05 +0200 Subject: [PATCH] fix ButtonRequestType in Ping and GetEntropy --- tests/test_msg_getentropy.py | 2 +- tests/test_msg_ping.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_msg_getentropy.py b/tests/test_msg_getentropy.py index 5eae42d..8cf7a65 100644 --- a/tests/test_msg_getentropy.py +++ b/tests/test_msg_getentropy.py @@ -23,7 +23,7 @@ class TestMsgGetentropy(common.TrezorTest): def test_entropy(self): for l in [0, 1, 2, 3, 4, 5, 8, 9, 16, 17, 32, 33, 64, 65, 128, 129, 256, 257, 512, 513, 1024]: with self.client: - self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_Other), proto.Entropy()]) + self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_ProtectCall), proto.Entropy()]) ent = self.client.get_entropy(l) self.assertTrue(len(ent) >= l) print 'entropy = ', entropy(ent) diff --git a/tests/test_msg_ping.py b/tests/test_msg_ping.py index 8fc5326..f5df25d 100644 --- a/tests/test_msg_ping.py +++ b/tests/test_msg_ping.py @@ -16,7 +16,7 @@ class TestPing(common.TrezorTest): self.assertEqual(res, 'random data') with self.client: - self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_Other), proto.Success()]) + self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_ProtectCall), proto.Success()]) res = self.client.ping('random data', button_protection=True) self.assertEqual(res, 'random data') @@ -34,13 +34,13 @@ class TestPing(common.TrezorTest): self.setup_mnemonic_pin_passphrase() with self.client: - self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_Other), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()]) + self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_ProtectCall), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()]) res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True) self.assertEqual(res, 'random data') with self.client: # pin and passphrase are cached - self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_Other), proto.Success()]) + self.client.set_expected_responses([proto.ButtonRequest(code=proto_types.ButtonRequest_ProtectCall), proto.Success()]) res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True) self.assertEqual(res, 'random data')