grdddj
11 months ago
committed by
Jiří Musil
parent
c97c515718
commit
c99fd824b3
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,249 @@
|
||||
from trezorlib import messages
|
||||
from trezorlib.debuglink import TrezorClientDebugLink as Client
|
||||
|
||||
from .common import BRGeneratorType
|
||||
|
||||
B = messages.ButtonRequestType
|
||||
|
||||
|
||||
class PinFlow:
|
||||
def __init__(self, client: Client):
|
||||
self.client = client
|
||||
self.debug = self.client.debug
|
||||
|
||||
def setup_new_pin(
|
||||
self, pin: str, second_different_pin: str | None = None
|
||||
) -> BRGeneratorType:
|
||||
yield # Enter PIN
|
||||
assert "PinKeyboard" in self.debug.wait_layout().all_components()
|
||||
self.debug.input(pin)
|
||||
if self.debug.model == "R":
|
||||
yield # Reenter PIN
|
||||
assert "re-enter PIN" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
yield # Enter PIN again
|
||||
assert "PinKeyboard" in self.debug.wait_layout().all_components()
|
||||
if second_different_pin is not None:
|
||||
self.debug.input(second_different_pin)
|
||||
else:
|
||||
self.debug.input(pin)
|
||||
|
||||
|
||||
class BackupFlow:
|
||||
def __init__(self, client: Client):
|
||||
self.client = client
|
||||
self.debug = self.client.debug
|
||||
|
||||
def confirm_new_wallet(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "By continuing you agree" in self.debug.wait_layout().text_content()
|
||||
if self.debug.model == "R":
|
||||
self.debug.press_right()
|
||||
self.debug.press_yes()
|
||||
|
||||
|
||||
class RecoveryFlow:
|
||||
def __init__(self, client: Client):
|
||||
self.client = client
|
||||
self.debug = self.client.debug
|
||||
|
||||
def confirm_recovery(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "By continuing you agree" in self.debug.wait_layout().text_content()
|
||||
if self.debug.model == "R":
|
||||
self.debug.press_right()
|
||||
self.debug.press_yes()
|
||||
|
||||
def confirm_dry_run(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "check the recovery seed" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def setup_slip39_recovery(self, num_words: int) -> BRGeneratorType:
|
||||
if self.debug.model == "R":
|
||||
yield from self.tr_recovery_homescreen()
|
||||
yield from self.input_number_of_words(num_words)
|
||||
yield from self.enter_any_share()
|
||||
|
||||
def setup_bip39_recovery(self, num_words: int) -> BRGeneratorType:
|
||||
if self.debug.model == "R":
|
||||
yield from self.tr_recovery_homescreen()
|
||||
yield from self.input_number_of_words(num_words)
|
||||
yield from self.enter_your_backup()
|
||||
|
||||
def tr_recovery_homescreen(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "number of words" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def enter_your_backup(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "Enter your backup" in self.debug.wait_layout().text_content()
|
||||
if (
|
||||
self.debug.model == "R"
|
||||
and "BACKUP CHECK" not in self.debug.wait_layout().title()
|
||||
):
|
||||
# Normal recovery has extra info (not dry run)
|
||||
self.debug.press_right(wait=True)
|
||||
self.debug.press_right(wait=True)
|
||||
self.debug.press_yes()
|
||||
|
||||
def enter_any_share(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "Enter any share" in self.debug.wait_layout().text_content()
|
||||
if (
|
||||
self.debug.model == "R"
|
||||
and "BACKUP CHECK" not in self.debug.wait_layout().title()
|
||||
):
|
||||
# Normal recovery has extra info (not dry run)
|
||||
self.debug.press_right(wait=True)
|
||||
self.debug.press_right(wait=True)
|
||||
self.debug.press_yes()
|
||||
|
||||
def abort_recovery(self, confirm: bool) -> BRGeneratorType:
|
||||
yield
|
||||
if self.debug.model == "R":
|
||||
assert "number of words" in self.debug.wait_layout().text_content()
|
||||
else:
|
||||
assert "Enter any share" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_no()
|
||||
|
||||
yield
|
||||
assert "abort the recovery" in self.debug.wait_layout().text_content()
|
||||
if self.debug.model == "R":
|
||||
self.debug.press_right()
|
||||
if confirm:
|
||||
self.debug.press_yes()
|
||||
else:
|
||||
self.debug.press_no()
|
||||
|
||||
def input_number_of_words(self, num_words: int) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.MnemonicWordCount
|
||||
if self.debug.model == "R":
|
||||
assert "NUMBER OF WORDS" in self.debug.wait_layout().title()
|
||||
else:
|
||||
assert "number of words" in self.debug.wait_layout().text_content()
|
||||
self.debug.input(str(num_words))
|
||||
|
||||
def warning_invalid_recovery_seed(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "Invalid recovery seed" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_invalid_recovery_share(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "Invalid recovery share" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_group_threshold_reached(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "Group threshold reached" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_share_already_entered(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "Share already entered" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_share_from_another_shamir(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert (
|
||||
"You have entered a share from another Shamir Backup"
|
||||
in self.debug.wait_layout().text_content()
|
||||
)
|
||||
self.debug.press_yes()
|
||||
|
||||
def success_share_group_entered(self) -> BRGeneratorType:
|
||||
yield
|
||||
assert "You have entered" in self.debug.wait_layout().text_content()
|
||||
assert "Group" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def success_wallet_recovered(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Success
|
||||
assert (
|
||||
"Wallet recovered successfully" in self.debug.wait_layout().text_content()
|
||||
)
|
||||
self.debug.press_yes()
|
||||
|
||||
def success_bip39_dry_run_valid(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Success
|
||||
assert "recovery seed is valid" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def success_slip39_dryrun_valid(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Success
|
||||
assert "recovery shares are valid" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_slip39_dryrun_mismatch(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "do not match" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def warning_bip39_dryrun_mismatch(self) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.Warning
|
||||
assert "does not match" in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def success_more_shares_needed(
|
||||
self, count_needed: int | None = None
|
||||
) -> BRGeneratorType:
|
||||
yield
|
||||
assert (
|
||||
"1 more share needed" in self.debug.wait_layout().text_content().lower()
|
||||
or "more shares needed" in self.debug.wait_layout().text_content().lower()
|
||||
)
|
||||
if count_needed is not None:
|
||||
assert str(count_needed) in self.debug.wait_layout().text_content()
|
||||
self.debug.press_yes()
|
||||
|
||||
def input_mnemonic(self, mnemonic: list[str]) -> BRGeneratorType:
|
||||
br = yield
|
||||
assert br.code == B.MnemonicInput
|
||||
assert "MnemonicKeyboard" in self.debug.wait_layout().all_components()
|
||||
for index, word in enumerate(mnemonic):
|
||||
if self.debug.model == "R":
|
||||
assert f"WORD {index + 1}" in self.debug.wait_layout().title()
|
||||
else:
|
||||
assert (
|
||||
f"Type word {index + 1}" in self.debug.wait_layout().text_content()
|
||||
)
|
||||
self.debug.input(word)
|
||||
|
||||
def input_all_slip39_shares(
|
||||
self,
|
||||
shares: list[str],
|
||||
has_groups: bool = False,
|
||||
click_info: bool = False,
|
||||
) -> BRGeneratorType:
|
||||
for index, share in enumerate(shares):
|
||||
mnemonic = share.split(" ")
|
||||
yield from self.input_mnemonic(mnemonic)
|
||||
|
||||
if index < len(shares) - 1:
|
||||
if has_groups:
|
||||
yield from self.success_share_group_entered()
|
||||
if self.debug.model == "T" and click_info:
|
||||
yield from self.tt_click_info()
|
||||
yield from self.success_more_shares_needed()
|
||||
|
||||
def tt_click_info(
|
||||
self,
|
||||
) -> BRGeneratorType:
|
||||
# Moving through the INFO button
|
||||
self.debug.press_info()
|
||||
yield
|
||||
self.debug.swipe_up()
|
||||
self.debug.press_yes()
|
||||
Loading…
Reference in new issue