status-mobile/test/appium/views/keycard_view.py

73 lines
3.1 KiB
Python
Raw Normal View History

from views.base_element import Button, Text, EditBox, SilentButton
from views.base_view import BaseView
class KeycardView(BaseView):
def __init__(self, driver):
super().__init__(driver)
self.begin_setup_button = Button(self.driver, translation_id="begin-set-up")
self.connect_card_button = Button(self.driver, accessibility_id="connect-card")
self.disconnect_card_button = Button(self.driver, accessibility_id="disconnect-card")
self.reset_card_state_button = Button(self.driver, accessibility_id="keycard-reset-state")
self.connect_selected_card_button = Button(self.driver, accessibility_id="connect-selected-card")
self.pair_code_text = Text(self.driver, accessibility_id="pair-code")
self.pair_code_input = EditBox(self.driver, xpath="//android.widget.EditText")
self.pair_to_this_device_button = Button(self.driver, translation_id="pair-card")
self.connect_pairing_card_button = Button(self.driver, accessibility_id="connect-pairing-card")
# Keyboard
self.zero_button = SilentButton(self.driver, accessibility_id="numpad-button-0")
self.one_button = SilentButton(self.driver, accessibility_id="numpad-button-1")
self.two_button = SilentButton(self.driver, accessibility_id="numpad-button-2")
# Backup seed phrase
self.confirm_seed_phrase_edit_box = EditBox(self.driver, accessibility_id="enter-word")
def enter_default_pin(self):
self.driver.info("Enter default pin 111111")
[self.one_button.click() for _ in range(6)]
def enter_default_puk(self):
self.driver.info("Enter default pin 1111 1111 1111")
[self.one_button.click() for _ in range(12)]
def enter_another_pin(self):
self.driver.info("Enter not-default pin 222222")
for _ in range(6):
self.two_button.click()
def get_recovery_word(self, word_id):
word_element = SilentButton(self.driver, accessibility_id="word%s" % word_id)
return word_element.text
def get_required_word_number(self):
description = SilentButton(self.driver, accessibility_id="word-number")
full_text = description.text
word_number = ''.join(i for i in full_text if i.isdigit())
return word_number
def get_seed_phrase(self):
recovery_phrase = dict()
2021-11-18 15:16:48 +00:00
for i in range(0, 12):
word = self.get_recovery_word(i)
2021-11-18 15:16:48 +00:00
recovery_phrase[str(i + 1)] = word
return recovery_phrase
def backup_seed_phrase(self):
self.driver.info("Backing up seed phrase for keycard")
recovery_phrase = self.get_seed_phrase()
self.confirm_button.click()
self.yes_button.click()
for _ in range(2):
number = self.get_required_word_number()
self.confirm_seed_phrase_edit_box.set_value(recovery_phrase[number])
self.next_button.click()
return ' '.join(recovery_phrase.values())
2020-04-28 14:40:41 +00:00
def confirm_pin_and_proceed(self):
self.next_button.click()
self.begin_setup_button.click()
self.connect_card_button.click()
self.enter_default_pin()
2021-11-18 15:16:48 +00:00
self.enter_default_pin()