arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-05-06 17:09:11 +00:00
Code Issues Releases Wiki Activity

chore(tests): accounting for the extra T2B1 buton request in input flows

Browse Source 
[no changelog]
This commit is contained in:
grdddj 2023-09-14 17:43:08 +02:00 committed by Jiří Musil
parent 8a9354cd85
commit 2ca3c4c83b
1 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/input_flows.py
View File

@ -499,7 +499,7 @@ class InputFlowSignTxHighFee(InputFlowBase):
self.finished = True self.finished = True
def input_flow_tt(self) -> BRGeneratorType: def input_flow_common(self) -> BRGeneratorType:
screens = [ screens = [
B.ConfirmOutput, B.ConfirmOutput,
B.ConfirmOutput, B.ConfirmOutput,
@ -508,14 +508,6 @@ class InputFlowSignTxHighFee(InputFlowBase):
] ]
yield from self.go_through_all_screens(screens) yield from self.go_through_all_screens(screens)
def input_flow_tr(self) -> BRGeneratorType:
screens = [
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
yield from self.go_through_all_screens(screens)
def sign_tx_go_to_info(client: Client) -> Generator[None, None, str]: def sign_tx_go_to_info(client: Client) -> Generator[None, None, str]:
yield # confirm output yield # confirm output
@ -540,11 +532,12 @@ def sign_tx_go_to_info(client: Client) -> Generator[None, None, str]:
def sign_tx_go_to_info_tr( def sign_tx_go_to_info_tr(
client: Client, client: Client,
) -> Generator[None, None, str]: ) -> Generator[None, None, str]:
yield # confirm output yield # confirm address
client.debug.wait_layout() client.debug.wait_layout()
client.debug.press_right() # CONTINUE client.debug.press_yes() # CONTINUE
yield # confirm amount
client.debug.wait_layout() client.debug.wait_layout()
client.debug.press_right() # CONFIRM client.debug.press_yes() # CONFIRM
screen_texts: list[str] = [] screen_texts: list[str] = []
@ -686,9 +679,10 @@ def lock_time_input_flow_tt(
def lock_time_input_flow_tr( def lock_time_input_flow_tr(
debug: DebugLink, layout_assert_func: Callable[[DebugLink], None] debug: DebugLink, layout_assert_func: Callable[[DebugLink], None]
) -> BRGeneratorType: ) -> BRGeneratorType:
yield # confirm output yield # confirm address
debug.wait_layout() debug.wait_layout()
debug.swipe_up() debug.press_yes()
yield # confirm amount
debug.wait_layout() debug.wait_layout()
debug.press_yes() debug.press_yes()