From c8b9beaba240abfd04eb4c9a903e552c6aac0692 Mon Sep 17 00:00:00 2001 From: Tomas Susanka Date: Wed, 26 Jun 2019 11:57:59 +0200 Subject: [PATCH] tests: fix recovery device tests also in bip39 backup --- .../tests/device_tests/test_msg_recoverydevice_t2.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py b/python/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py index 823a4c6ffd..5cf5719cdb 100644 --- a/python/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py +++ b/python/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py @@ -66,6 +66,11 @@ class TestMsgRecoverydeviceT2(TrezorTest): self.client.debug.input("654") ret = self.client.call_raw(proto.ButtonAck()) + # Confirm success + assert isinstance(ret, proto.ButtonRequest) + self.client.debug.press_yes() + ret = self.client.call_raw(proto.ButtonAck()) + # Workflow succesfully ended assert ret == proto.Success(message="Device recovered") @@ -107,6 +112,11 @@ class TestMsgRecoverydeviceT2(TrezorTest): self.client.debug.input(word) ret = self.client.transport.read() + # Confirm success + assert isinstance(ret, proto.ButtonRequest) + self.client.debug.press_yes() + ret = self.client.call_raw(proto.ButtonAck()) + # Workflow succesfully ended assert ret == proto.Success(message="Device recovered")