114 lines
4.1 KiB
Python
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() |