From ebe41c69f853309d3e40931a6ac3ba25613840b4 Mon Sep 17 00:00:00 2001 From: Tomas Susanka Date: Wed, 13 Jun 2018 13:40:58 +0200 Subject: [PATCH] tests/stellar: trezor core currently throws Firmware err --- .../tests/device_tests/test_msg_stellar_get_address.py | 8 ++++++-- .../tests/device_tests/test_msg_stellar_get_public_key.py | 8 ++++++-- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/trezorlib/tests/device_tests/test_msg_stellar_get_address.py b/trezorlib/tests/device_tests/test_msg_stellar_get_address.py index 5f75eff099..a2ab69db20 100644 --- a/trezorlib/tests/device_tests/test_msg_stellar_get_address.py +++ b/trezorlib/tests/device_tests/test_msg_stellar_get_address.py @@ -49,5 +49,9 @@ class TestMsgStellarGetAddress(TrezorTest): with pytest.raises(CallException) as exc: self.client.stellar_get_address(parse_path('m/0/1')) - assert exc.value.args[0] == proto.FailureType.ProcessError - assert exc.value.args[1].endswith('Failed to derive private key') + if TREZOR_VERSION == 1: + assert exc.value.args[0] == proto.FailureType.ProcessError + assert exc.value.args[1].endswith('Failed to derive private key') + else: + assert exc.value.args[0] == proto.FailureType.FirmwareError + assert exc.value.args[1].endswith('Firmware error') diff --git a/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py b/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py index 1d07312d9a..e8bf8e5a39 100644 --- a/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py +++ b/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py @@ -41,5 +41,9 @@ class TestMsgStellarGetPublicKey(TrezorTest): with pytest.raises(CallException) as exc: self.client.stellar_get_public_key(parse_path('m/0/1')) - assert exc.value.args[0] == proto.FailureType.ProcessError - assert exc.value.args[1].endswith('Failed to derive private key') + if TREZOR_VERSION == 1: + assert exc.value.args[0] == proto.FailureType.ProcessError + assert exc.value.args[1].endswith('Failed to derive private key') + else: + assert exc.value.args[0] == proto.FailureType.FirmwareError + assert exc.value.args[1].endswith('Firmware error')