From 3c19e3167d69902305a27f10e43abb5fc7a0254d Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Tue, 6 Aug 2019 11:33:45 +0200 Subject: [PATCH] python: fix reset device test for t1 --- .../tests/device_tests/test_msg_resetdevice.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/trezorlib/tests/device_tests/test_msg_resetdevice.py b/python/trezorlib/tests/device_tests/test_msg_resetdevice.py index 298ebc548..e09b53959 100644 --- a/python/trezorlib/tests/device_tests/test_msg_resetdevice.py +++ b/python/trezorlib/tests/device_tests/test_msg_resetdevice.py @@ -110,10 +110,17 @@ class TestMsgResetDevice(TrezorTest): ) ) + # Do you want ... ? assert isinstance(ret, proto.ButtonRequest) self.client.debug.press_yes() ret = self.client.call_raw(proto.ButtonAck()) + # Entropy screen #1 + assert isinstance(ret, proto.ButtonRequest) + self.client.debug.press_yes() + ret = self.client.call_raw(proto.ButtonAck()) + + # Entropy screen #2 assert isinstance(ret, proto.ButtonRequest) self.client.debug.press_yes() ret = self.client.call_raw(proto.ButtonAck()) @@ -191,10 +198,17 @@ class TestMsgResetDevice(TrezorTest): ) ) + # Do you want ... ? + assert isinstance(ret, proto.ButtonRequest) + self.client.debug.press_yes() + ret = self.client.call_raw(proto.ButtonAck()) + + # Entropy screen #1 assert isinstance(ret, proto.ButtonRequest) self.client.debug.press_yes() ret = self.client.call_raw(proto.ButtonAck()) + # Entropy screen #2 assert isinstance(ret, proto.ButtonRequest) self.client.debug.press_yes() ret = self.client.call_raw(proto.ButtonAck())