arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-22 12:32:02 +00:00
Code Issues Releases Wiki Activity

tests: update persistence tests

Browse Source
This commit is contained in:
grdddj 2023-03-30 15:56:23 +02:00
parent 58cb524be3
commit 9bcb4b058f
2 changed files with 15 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/persistence_tests/test_shamir_persistence.py
View File

@ -50,10 +50,10 @@ def test_abort(emulator: Emulator):
device_handler.run(device.recover, pin_protection=False)
layout = debug.wait_layout()
assert layout.get_title() == "WALLET RECOVERY"
assert layout.title() == "WALLET RECOVERY"
layout = debug.click(buttons.OK, wait=True)
assert "Select number of words" in layout.get_content()
assert "select the number of words" in layout.text_content()
device_handler.restart(emulator)
debug = device_handler.debuglink()
@ -63,13 +63,13 @@ def test_abort(emulator: Emulator):
# no waiting for layout because layout doesn't change
layout = debug.read_layout()
assert "Select number of words" in layout.get_content()
assert "select the number of words" in layout.text_content()
layout = debug.click(buttons.CANCEL, wait=True)
assert layout.get_title() == "ABORT RECOVERY"
assert layout.title() == "ABORT RECOVERY"
layout = debug.click(buttons.OK, wait=True)
assert layout.text.startswith("< Homescreen")
assert layout.str_content.startswith("< Homescreen")
features = device_handler.features()
assert features.recovery_mode is False
@ -136,10 +136,10 @@ def test_recovery_on_old_wallet(emulator: Emulator):
# start entering first share
layout = debug.read_layout()
assert "Enter any share" in layout.text
assert "Enter any share" in layout.str_content
debug.press_yes()
layout = debug.wait_layout()
assert layout.text == "< MnemonicKeyboard >"
assert "MnemonicKeyboard" in layout.str_content
# enter first word
debug.input(words[0])
@ -151,12 +151,12 @@ def test_recovery_on_old_wallet(emulator: Emulator):
# try entering remaining 19 words
for word in words[1:]:
assert layout.text == "< MnemonicKeyboard >"
assert "MnemonicKeyboard" in layout.str_content
debug.input(word)
layout = debug.wait_layout()
# check that we entered the first share successfully
assert "2 more shares" in layout.text
assert "2 more shares" in layout.str_content
# try entering the remaining shares
for share in MNEMONIC_SLIP39_BASIC_20_3of6[1:3]:
@ -178,13 +178,13 @@ def test_recovery_multiple_resets(emulator: Emulator):
expected_text = "Enter any share"
remaining = len(shares)
for share in shares:
assert expected_text in layout.text
assert expected_text in layout.str_content
layout = recovery.enter_share(debug, share)
remaining -= 1
expected_text = "You have entered"
debug = _restart(device_handler, emulator)
assert "You have successfully recovered your wallet" in layout.get_content()
assert "You have finished recovering your wallet" in layout.text_content()
device_handler = BackgroundDeviceHandler(emulator.client)
debug = device_handler.debuglink()
@ -212,7 +212,7 @@ def test_recovery_multiple_resets(emulator: Emulator):
enter_shares_with_restarts(debug)
debug = device_handler.debuglink()
layout = debug.read_layout()
assert layout.text.startswith("< Homescreen")
assert layout.str_content.startswith("< Homescreen")
features = device_handler.features()
assert features.initialized is True

20
tests/persistence_tests/test_wipe_code.py
View File

@ -3,6 +3,7 @@ from trezorlib.debuglink import TrezorClientDebugLink as Client
from ..common import MNEMONIC12
from ..emulators import EmulatorWrapper
from ..input_flows import InputFlowSetupDevicePINWIpeCode
from ..upgrade_tests import core_only, legacy_only
PIN = "1234"
@ -26,24 +27,9 @@ def setup_device_core(client: Client, pin: str, wipe_code: str) -> None:
client, MNEMONIC12, pin, passphrase_protection=False, label="WIPECODE"
)
def input_flow():
yield # do you want to set/change the wipe_code?
client.debug.press_yes()
if pin is not None:
yield # enter current pin
client.debug.input(pin)
yield # enter new wipe code
client.debug.input(wipe_code)
yield # enter new wipe code again
client.debug.input(wipe_code)
yield # success
client.debug.press_yes()
with client:
client.set_expected_responses(
[messages.ButtonRequest()] * 5 + [messages.Success, messages.Features]
)
client.set_input_flow(input_flow)
IF = InputFlowSetupDevicePINWIpeCode(client, pin, wipe_code)
client.set_input_flow(IF.get())
device.change_wipe_code(client)