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

114 lines
4.1 KiB
Python

from views.base_element import BaseButton, BaseText, BaseElement, BaseEditBox
from views.base_view import BaseView
class BeginSetupButton(BaseButton):
def __init__(self, driver):
super(BeginSetupButton, self).__init__(driver)
self.locator = self.Locator.text_selector("Begin setup")
class OnePinKeyboardButton(BaseButton):
def __init__(self, driver):
super(OnePinKeyboardButton, self).__init__(driver)
self.locator = self.Locator.text_selector("1")
class TwoPinKeyboardButton(BaseButton):
def __init__(self, driver):
super(TwoPinKeyboardButton, self).__init__(driver)
self.locator = self.Locator.text_selector("2")
class ConnectCardButton(BaseButton):
def __init__(self, driver):
super(ConnectCardButton, self).__init__(driver)
self.locator = self.Locator.accessibility_id("connect-card")
class ConnectSelectedCardButton(BaseButton):
def __init__(self, driver):
super(ConnectSelectedCardButton, self).__init__(driver)
self.locator = self.Locator.accessibility_id("connect-selected-card")
class DisconnectCardButton(BaseButton):
def __init__(self, driver):
super(DisconnectCardButton, self).__init__(driver)
self.locator = self.Locator.accessibility_id("disconnect-card")
class ResetCardButton(BaseButton):
def __init__(self, driver):
super(ResetCardButton, self).__init__(driver)
self.locator = self.Locator.accessibility_id("keycard-reset-state")
class RecoveryWordText(BaseText):
def __init__(self, driver, word_id):
super(RecoveryWordText, self).__init__(driver)
self.word_id = word_id
self.locator = self.Locator.accessibility_id("word%s" % word_id)
class WordNumberText(BaseText):
def __init__(self, driver):
super(WordNumberText, self).__init__(driver)
self.locator = self.Locator.accessibility_id("word-number")
class ConfirmSeedPhraseInput(BaseEditBox):
def __init__(self, driver):
super(ConfirmSeedPhraseInput, self).__init__(driver)
self.locator = self.Locator.accessibility_id("enter-word")
class KeycardView(BaseView):
def __init__(self, driver):
super(KeycardView, self).__init__(driver)
self.begin_setup_button = BeginSetupButton(self.driver)
self.connect_card_button = ConnectCardButton(self.driver)
self.disconnect_card_button = DisconnectCardButton(self.driver)
self.reset_card_state_button = ResetCardButton(self.driver)
self.connect_selected_card_button = ConnectSelectedCardButton(self.driver)
#keyboard
self.one_button = OnePinKeyboardButton(self.driver)
self.two_button = TwoPinKeyboardButton(self.driver)
#backup seed phrase
self.confirm_seed_phrase_edit_box = ConfirmSeedPhraseInput(self.driver)
def enter_default_pin(self):
for _ in range(3):
self.one_button.click()
self.two_button.click()
def get_recovery_word(self, word_id):
word_element = RecoveryWordText(self.driver, word_id)
return word_element.text
def get_required_word_number(self):
description = WordNumberText(self.driver)
full_text = description.text
word_number = ''.join(i for i in full_text if i.isdigit())
return word_number
def backup_seed_phrase(self):
recovery_phrase = dict()
for i in range(0,12):
word = self.get_recovery_word(i)
recovery_phrase[str(i+1)] = word
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()
def confirm_pin_and_proceed(self):
self.next_button.click()
self.begin_setup_button.click()
self.connect_card_button.click()
self.enter_default_pin()
self.enter_default_pin()
self.next_button.scroll_to_element()
self.next_button.wait_for_visibility_of_element(20)
self.next_button.click()
self.yes_button.click()