mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-15 19:08:07 +00:00
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
import pytest
|
|
|
|
from trezorlib import device, messages
|
|
|
|
from .. import buttons
|
|
from ..common import MNEMONIC_SLIP39_BASIC_20_3of6
|
|
|
|
|
|
def enter_word(debug, word):
|
|
word = word[:4]
|
|
for coords in buttons.type_word(word):
|
|
debug.click(coords)
|
|
return " ".join(debug.click(buttons.CONFIRM_WORD, wait=True))
|
|
|
|
|
|
def click_ok(debug):
|
|
return " ".join(debug.click(buttons.OK, wait=True))
|
|
|
|
|
|
@pytest.mark.skip_t1
|
|
@pytest.mark.setup_client(uninitialized=True)
|
|
def test_recovery(client):
|
|
with client:
|
|
client.set_expected_responses(
|
|
[
|
|
messages.ButtonRequest(code=messages.ButtonRequestType.ProtectCall),
|
|
messages.Success(),
|
|
messages.Features(),
|
|
]
|
|
)
|
|
device.recover(client, pin_protection=False)
|
|
|
|
assert client.features.initialized is False
|
|
assert client.features.recovery_mode is True
|
|
|
|
# select number of words
|
|
state = client.debug.state()
|
|
text = " ".join(state.layout_lines)
|
|
assert "Select number of words" in text
|
|
text = click_ok(client.debug)
|
|
|
|
assert text == "WordSelector"
|
|
# click "20" at 2, 2
|
|
coords = buttons.grid34(2, 2)
|
|
lines = client.debug.click(coords, wait=True)
|
|
text = " ".join(lines)
|
|
|
|
expected_text = "Enter any share (20 words)"
|
|
remaining = len(MNEMONIC_SLIP39_BASIC_20_3of6)
|
|
for share in MNEMONIC_SLIP39_BASIC_20_3of6:
|
|
assert expected_text in text
|
|
text = click_ok(client.debug)
|
|
|
|
assert text == "Slip39Keyboard"
|
|
for word in share.split(" "):
|
|
text = enter_word(client.debug, word)
|
|
|
|
remaining -= 1
|
|
expected_text = "RecoveryHomescreen {} more".format(remaining)
|
|
|
|
assert "You have successfully recovered your wallet" in text
|
|
text = click_ok(client.debug)
|
|
|
|
assert text == "Homescreen"
|
|
client.init_device()
|
|
assert client.features.initialized is True
|
|
assert client.features.recovery_mode is False
|