status-react/test/appium/views/sign_in_view.py

from tests import get_current_time
from views.base_element import BaseButton, BaseEditBox
from views.base_view import BaseView
import time

class AccountButton(BaseButton):

    def __init__(self, driver):
        super(AccountButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[contains(@text,'0x')]")


class PasswordInput(BaseEditBox):

    def __init__(self, driver):
        super(PasswordInput, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='Password']"
                                                   "/following-sibling::android.view.ViewGroup/android.widget.EditText")


class SignInButton(BaseButton):

    def __init__(self, driver):
        super(SignInButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='Sign in' or @text='SIGN IN']")

    def navigate(self):
        from views.home_view import HomeView
        return HomeView(self.driver)


class RecoverAccessButton(BaseButton):

    def __init__(self, driver):
        super(RecoverAccessButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='Recover access']")

    def navigate(self):
        from views.recover_access_view import RecoverAccessView
        return RecoverAccessView(self.driver)


class CreateAccountButton(BaseButton):
    def __init__(self, driver):
        super(CreateAccountButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector(
            "//android.widget.TextView[@text='CREATE ACCOUNT' or @text='CREATE NEW ACCOUNT']")


class IHaveAccountButton(RecoverAccessButton):
    def __init__(self, driver):
        super(IHaveAccountButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='I ALREADY HAVE AN ACCOUNT']")


class AddExistingAccountButton(RecoverAccessButton):
    def __init__(self, driver):
        super(AddExistingAccountButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='ADD EXISTING ACCOUNT']")


class ConfirmPasswordInput(BaseEditBox):
    def __init__(self, driver):
        super(ConfirmPasswordInput, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='Confirm']")


class NameInput(BaseEditBox):
    def __init__(self, driver):
        super(NameInput, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.EditText")


class DonNotShare(BaseButton):

    def __init__(self, driver):
        super(DonNotShare, self).__init__(driver)
        self.locator = self.Locator.xpath_selector('//*[@text="NO, I DON%sT WANT TO SHARE" '
                                                   'or @text="Do not share"]' % "'")


class SignInView(BaseView):

    def __init__(self, driver):
        super(SignInView, self).__init__(driver)
        self.driver = driver

        self.account_button = AccountButton(self.driver)
        self.password_input = PasswordInput(self.driver)
        self.sign_in_button = SignInButton(self.driver)
        self.recover_access_button = RecoverAccessButton(self.driver)

        # new design
        self.create_account_button = CreateAccountButton(self.driver)
        self.i_have_account_button = IHaveAccountButton(self.driver)
        self.add_existing_account_button = AddExistingAccountButton(self.driver)
        self.confirm_password_input = ConfirmPasswordInput(self.driver)
        self.name_input = NameInput(self.driver)
        self.do_not_share = DonNotShare(self.driver)

    def create_user(self, password: str = 'qwerty1234'):
        self.create_account_button.click()
        self.password_input.set_value(password)
        self.next_button.click()
        self.confirm_password_input.set_value(password)
        self.next_button.click()

        # bypass StaleElementReferenceException
        time.sleep(5)
        self.name_input.wait_for_element(10)
        self.name_input.click()
        username = 'user_%s' % get_current_time()
        self.send_as_keyevent(username)

        self.next_button.click()
        self.do_not_share.wait_for_visibility_of_element(10)
        self.do_not_share.click_until_presence_of_element(self.home_button)
        return username

    def recover_access(self, passphrase, password):
        recover_access_view = self.i_have_account_button.click()
        recover_access_view.passphrase_input.click()
        recover_access_view.send_as_keyevent(passphrase)
        recover_access_view.password_input.click()
        recover_access_view.send_as_keyevent(password)
        recover_access_view.sign_in_button.click()
        self.do_not_share.wait_for_element(10)
        self.do_not_share.click_until_presence_of_element(self.home_button)

    def sign_in(self, password):
        self.password_input.set_value(password)
        self.sign_in_button.click()

    def click_account_by_position(self, position: int):
        self.account_button.find_elements()[position].click()