From 52ad4e9f0d194094b17bf1c9e6b0f0632f5a8702 Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Fri, 20 May 2016 16:34:49 +0200 Subject: [PATCH] update get_public_key and get_address tests to reflect reality --- tests/test_protection_levels.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_protection_levels.py b/tests/test_protection_levels.py index 2fcd353..e0b9faf 100644 --- a/tests/test_protection_levels.py +++ b/tests/test_protection_levels.py @@ -52,14 +52,16 @@ class TestProtectionLevels(common.TrezorTest): def test_get_public_key(self): with self.client: self.setup_mnemonic_pin_passphrase() - self.client.set_expected_responses([proto.PassphraseRequest(), + self.client.set_expected_responses([proto.PinMatrixRequest(), + proto.PassphraseRequest(), proto.PublicKey()]) self.client.get_public_node([]) def test_get_address(self): with self.client: self.setup_mnemonic_pin_passphrase() - self.client.set_expected_responses([proto.PassphraseRequest(), + self.client.set_expected_responses([proto.PinMatrixRequest(), + proto.PassphraseRequest(), proto.Address()]) self.client.get_address('Bitcoin', [])