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

from views.base_element import BaseButton, BaseEditBox
from views.base_view import BaseView


class StatusTestDAppButton(BaseButton):

    def __init__(self, driver):
        super(StatusTestDAppButton, self).__init__(driver)
        self.locator = self.Locator.text_selector('Status Test DApp')

    def navigate(self):
        from views.web_views.status_test_dapp import StatusTestDAppView
        return StatusTestDAppView(self.driver)


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


class PublicKeyEditBox(BaseEditBox):
    def __init__(self, driver):
        super(PublicKeyEditBox, self).__init__(driver)
        self.locator = self.Locator.accessibility_id('enter-contact-code-input')


class ScanContactCodeButton(BaseEditBox):
    def __init__(self, driver):
        super(ScanContactCodeButton, self).__init__(driver)
        self.locator = self.Locator.accessibility_id('scan-contact-code-button')


class ConfirmPublicKeyButton(BaseButton):
    def __init__(self, driver):
        super(ConfirmPublicKeyButton, self).__init__(driver)
        self.locator = \
            self.Locator.xpath_selector('(//android.view.ViewGroup[@content-desc="icon"])[2]')


class ContactsView(BaseView):

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

        self.plus_button = PlusButton(self.driver)
        self.public_key_edit_box = PublicKeyEditBox(self.driver)
        self.scan_contact_code_button = ScanContactCodeButton(self.driver)
        self.confirm_public_key_button = ConfirmPublicKeyButton(self.driver)

        self.status_test_dapp_button = StatusTestDAppButton(self.driver)