from views.base_element import BaseButton, BaseText, 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 PairCodeText(BaseText): def __init__(self, driver): super(PairCodeText, self).__init__(driver) self.locator = self.Locator.accessibility_id("pair-code") class PairCodeInput(BaseEditBox): def __init__(self, driver): super(PairCodeInput, self).__init__(driver) self.locator = self.Locator.xpath_selector("//android.widget.EditText") class PairToThisDeviceButton(BaseButton): def __init__(self, driver): super(PairToThisDeviceButton, self).__init__(driver) self.locator = self.Locator.text_selector("Pair to this device") 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) self.pair_code_text = PairCodeText(self.driver) self.pair_code_input = PairCodeInput(self.driver) self.pair_to_this_device_button = PairToThisDeviceButton(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 enter_another_pin(self): for _ in range(6): 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 get_seed_phrase(self): recovery_phrase = dict() for i in range(0,12): word = self.get_recovery_word(i) recovery_phrase[str(i+1)] = word return recovery_phrase def backup_seed_phrase(self): 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() 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()