arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-06 20:52:40 +00:00
Code Issues Releases Wiki Activity

tests: fix recovery device tests due to additional screen

Browse Source
This commit is contained in:
Tomas Susanka 2019-06-26 10:58:53 +02:00
parent 4fbac3ec63
commit daa473c34b
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_shamir.py
View File

@ -76,6 +76,11 @@ class TestMsgRecoveryDeviceShamir(TrezorTest):
self.client.debug.press_yes() self.client.debug.press_yes()
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")
@ -162,6 +167,11 @@ class TestMsgRecoveryDeviceShamir(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")