class LoginPage():
def __init__(self, driver):
self.driver = driver
self.base = BasePage(self.driver)
def get_username_element(self):
"""获取username输入栏的element"""
return self.base.get_element("username")
def get_password_element(self):
"""获取password输入栏的element"""
return self.base.get_element("password")
def get_login_button_element(self):
"""获取login_button按钮的element"""
return self.base.get_element("login_button")
def get_forget_password_button_element(self):
"""获取忘记密码按钮的element"""
return self.base.get_element("forget_password_button")
def get_register_button_element(self):
"""获取注册的按钮的element"""
return self.base.get_element("register_button")
def get_tost_element(self, message):
"""获取tost的element"""
time.sleep(2)
# 获取test的定位控件
tost_element = ("xpath", "//*[contains(@text," + message + ")]")
return WebDriverWait(self.driver, 10, 0.1).until(EC.presence_of_element_located(tost_element))
class RegisterPage():
def __init__(self, driver):
self.base = BasePage(driver)
def get_email_element(self):
return self.base.get_element("user_email")
def get_email_error_element(self):
return self.base.get_element("user_email_error")
def get_name_element(self):
return self.base.get_element("user_nickname")
def get_name_error_element(self):
return self.base.get_element("user_nickname_error")
def get_password_element(self):
return self.base.get_element("user_password")
def get_password_error_element(self):
return self.base.get_element("user_password_error")
def get_code_element(self):
return self.base.get_element("code_text")
def get_code_error_element(self):
return self.base.get_element("code_text_error")
def get_button_element(self):
return self.base.get_element("register_button")
def get_user_element(self, key):
"""
定位用户信息，获取element
:param key: 配置文件的key
:return: 返回定位的元素的位置element
"""
find_element = BasePage(self.driver)
user_element = find_element.get_element(key)
return user_element
class RegisterPage():
def __init__(self, driver):
self.driver = driver
self.base = BasePage(self.driver)
def get_login_element(self):
"""获取跳转登陆页面的element"""
return self.base.get_element("login_button", node="register_element")
# if __name__ == "__main__":
# ll = RegisterPage(driver)
# ll.get_login_element()
