update get_public_key and get_address tests to reflect reality

8 years ago · 52ad4e9f0d
parent fdc3cff1d6
commit 52ad4e9f0d
1 changed files with 4 additions and 2 deletions
--- 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', [])