arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-19 14:08:11 +00:00
Code Issues Releases Wiki Activity

tests: fix input_flow of recovery abort

Browse Source 
[no changelog]
This commit is contained in:
obrusvit 2024-07-28 18:54:42 +02:00 committed by Vít Obrusník
parent 71b2c986f8
commit 37edfcb974
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
tests/input_flows_helpers.py
View File

@ -126,20 +126,31 @@ class RecoveryFlow:
yield yield
if self.client.model is models.T2B1: if self.client.model is models.T2B1:
TR.assert_in(self._text_content(), "recovery__num_of_words") 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: elif self.client.model is models.T3T1:
TR.assert_in(self._text_content(), "recovery__enter_each_word") 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: else:
TR.assert_in(self._text_content(), "recovery__enter_any_share") 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() 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: def input_number_of_words(self, num_words: int) -> BRGeneratorType:
br = yield br = yield