mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-19 14:08:11 +00:00
tests: fix input_flow of recovery abort
[no changelog]
This commit is contained in:
parent
71b2c986f8
commit
37edfcb974
@ -126,20 +126,31 @@ class RecoveryFlow:
|
||||
yield
|
||||
if self.client.model is models.T2B1:
|
||||
TR.assert_in(self._text_content(), "recovery__num_of_words")
|
||||
self.debug.press_no()
|
||||
yield
|
||||
TR.assert_in(self._text_content(), "recovery__wanna_cancel_recovery")
|
||||
self.debug.press_right()
|
||||
if confirm:
|
||||
self.debug.press_yes()
|
||||
else:
|
||||
self.debug.press_no()
|
||||
elif self.client.model is models.T3T1:
|
||||
TR.assert_in(self._text_content(), "recovery__enter_each_word")
|
||||
self.debug.click(buttons.CORNER_BUTTON, wait=True)
|
||||
self.debug.synchronize_at("VerticalMenu")
|
||||
if confirm:
|
||||
self.debug.click(buttons.VERTICAL_MENU[0], wait=True)
|
||||
else:
|
||||
self.debug.click(buttons.CORNER_BUTTON, wait=True)
|
||||
else:
|
||||
TR.assert_in(self._text_content(), "recovery__enter_any_share")
|
||||
self.debug.press_no()
|
||||
|
||||
yield
|
||||
TR.assert_in(self._text_content(), "recovery__wanna_cancel_recovery")
|
||||
if self.client.model is models.T2B1:
|
||||
self.debug.press_right()
|
||||
if confirm:
|
||||
self.debug.press_yes()
|
||||
else:
|
||||
self.debug.press_no()
|
||||
yield
|
||||
TR.assert_in(self._text_content(), "recovery__wanna_cancel_recovery")
|
||||
if confirm:
|
||||
self.debug.press_yes()
|
||||
else:
|
||||
self.debug.press_no()
|
||||
|
||||
def input_number_of_words(self, num_words: int) -> BRGeneratorType:
|
||||
br = yield
|
||||
|
Loading…
Reference in New Issue
Block a user