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()