From 37edfcb974ace94bc841ea84d0f8826e06ea7783 Mon Sep 17 00:00:00 2001 From: obrusvit Date: Sun, 28 Jul 2024 18:54:42 +0200 Subject: [PATCH] tests: fix input_flow of recovery abort [no changelog] --- tests/input_flows_helpers.py | 29 ++++++++++++++++++++--------- 1 file changed, 20 insertions(+), 9 deletions(-) diff --git a/tests/input_flows_helpers.py b/tests/input_flows_helpers.py index 5135549e0b..8d59db2b9a 100644 --- a/tests/input_flows_helpers.py +++ b/tests/input_flows_helpers.py @@ -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