1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 11:39:03 +00:00

update get_public_key and get_address tests to reflect reality

This commit is contained in:
Pavol Rusnak 2016-05-20 16:34:49 +02:00
parent fdc3cff1d6
commit 52ad4e9f0d
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D

View File

@ -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', [])