arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-11 07:50:57 +00:00
Code Issues Releases Wiki Activity

tests/click: refactor reading lines to a function

Browse Source
This commit is contained in:
Tomas Susanka 2019-11-05 10:40:06 +00:00
parent e30245b96b
commit 2aab4e5955
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/click_tests/reset.py
View File

@ -26,6 +26,9 @@ def set_selection(debug, button, diff):
def read_words(debug, is_advanced=False): def read_words(debug, is_advanced=False):
def read_word(line: str):
return line.split()[1]
words = [] words = []
layout = debug.read_layout() layout = debug.read_layout()
if is_advanced: if is_advanced:
@ -35,17 +38,17 @@ def read_words(debug, is_advanced=False):
for i in range(6): for i in range(6):
lines = debug.read_layout().lines lines = debug.read_layout().lines
if i == 0: if i == 0:
words.append(lines[3].split()[1]) words.append(read_word(lines[3]))
words.append(lines[4].split()[1]) words.append(read_word(lines[4]))
debug.input(swipe=messages.DebugSwipeDirection.UP, wait=True) debug.input(swipe=messages.DebugSwipeDirection.UP, wait=True)
elif i == 5: elif i == 5:
words.append(lines[1].split()[1]) words.append(read_word(lines[1]))
words.append(lines[2].split()[1]) words.append(read_word(lines[2]))
else: else:
words.append(lines[1].split()[1]) words.append(read_word(lines[1]))
words.append(lines[2].split()[1]) words.append(read_word(lines[2]))
words.append(lines[3].split()[1]) words.append(read_word(lines[3]))
words.append(lines[4].split()[1]) words.append(read_word(lines[4]))
debug.input(swipe=messages.DebugSwipeDirection.UP, wait=True) debug.input(swipe=messages.DebugSwipeDirection.UP, wait=True)
debug.press_yes() debug.press_yes()
@ -53,7 +56,6 @@ def read_words(debug, is_advanced=False):
def confirm_words(debug, words): def confirm_words(debug, words):
# confirm words
layout = debug.wait_layout() layout = debug.wait_layout()
layout.text.startswith("Check share") layout.text.startswith("Check share")
for _ in range(3): for _ in range(3):