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

import logging

from selenium.common.exceptions import NoSuchElementException, TimeoutException

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


class RequestPasswordIcon(BaseButton):

    def __init__(self, driver):
        super(RequestPasswordIcon, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[@content-desc='request-password']")

    def click(self):
        self.wait_for_element(10)
        self.find_element().click()
        logging.info('Tap on %s' % self.name)
        return self.navigate()


class RecoverButton(BaseButton):

    def __init__(self, driver):
        super(RecoverButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[@text='Recover']")

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


class ChatRequestInput(BaseEditBox):

    def __init__(self, driver):
        super(ChatRequestInput, self).__init__(driver)
        self.locator = \
            self.Locator.xpath_selector("//android.widget.EditText[@content-desc!='chat-message-input']")


class ConsoleView(BaseView):

    def __init__(self, driver):
        super(ConsoleView, self).__init__(driver)

        self.request_password_icon = RequestPasswordIcon(self.driver)
        self.recover_button = RecoverButton(self.driver)
        self.chat_request_input = ChatRequestInput(self.driver)

        self.accept_agreements()

    def accept_agreements(self):
        for i in self.ok_button_apk, self.continue_button_apk:
            try:
                i.click()
            except (NoSuchElementException, TimeoutException):
                pass