arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-05-29 20:28:45 +00:00
Code Issues Releases Wiki Activity

tests: fix recovery device tests also in bip39 backup

Browse Source
This commit is contained in:
Tomas Susanka 2019-06-26 11:57:59 +02:00
parent b4d0dc8da4
commit c8b9beaba2
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py
View File

@ -66,6 +66,11 @@ class TestMsgRecoverydeviceT2(TrezorTest):
self.client.debug.input("654") self.client.debug.input("654")
ret = self.client.call_raw(proto.ButtonAck()) 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 # Workflow succesfully ended
assert ret == proto.Success(message="Device recovered") assert ret == proto.Success(message="Device recovered")
@ -107,6 +112,11 @@ class TestMsgRecoverydeviceT2(TrezorTest):
self.client.debug.input(word) self.client.debug.input(word)
ret = self.client.transport.read() 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 # Workflow succesfully ended
assert ret == proto.Success(message="Device recovered") assert ret == proto.Success(message="Device recovered")