39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from views.base_view import BaseViewObject
|
|
from views.base_element import *
|
|
|
|
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.login import LoginView
|
|
return LoginView(self.driver)
|
|
|
|
|
|
class HomeView(BaseViewObject):
|
|
|
|
def __init__(self, driver):
|
|
super(HomeView, self).__init__(driver)
|
|
for i in self.ok_button_apk, self.continue_button_apk:
|
|
try:
|
|
i.click()
|
|
except (NoSuchElementException, TimeoutException):
|
|
pass
|
|
self.request_password_icon = RequestPasswordIcon(self.driver)
|
|
self.recover_button = RecoverButton(self.driver)
|