58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
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
|