| | |
| | | from base.base_page import BasePage |
| | | from elements.login_elements import LoginElements as Login |
| | | from elements.login_elements import LoginElements |
| | | from selenium.common.exceptions import TimeoutException |
| | | |
| | | |
| | | class LoginPage(BasePage): |
| | | """ |
| | | 登录页面 |
| | | """ |
| | | # 页面元素 |
| | | es = LoginElements |
| | | |
| | | def username_input(self, username): |
| | | """ |
| | | 填写用户名 |
| | | :param username: 用户名 |
| | | :return: |
| | | """ |
| | | self.send_text(username, Login.username_input) |
| | | def __init__(self, driver): |
| | | super().__init__(driver) |
| | | self.driver = driver |
| | | # 切换到最顶层文档 |
| | | driver.switch_to.default_content() |
| | | |
| | | def password_input(self, password): |
| | | """ |
| | | 填写密码 |
| | | :param password: 密码 |
| | | :return: |
| | | """ |
| | | self.send_text(password, Login.password_input) |
| | | def login(self, username, password, code): |
| | | # 输入用户名 |
| | | self.fill(self.es.username_input, username) |
| | | # 输入密码 |
| | | self.fill(self.es.password_input, password) |
| | | # 输入验证码 |
| | | self.fill(self.es.validate_code_input, code) |
| | | # 点击登录按钮 |
| | | self.click(self.es.login_btn) |
| | | |
| | | def validate_code_input(self, validate_code): |
| | | """ |
| | | 填写验证码 |
| | | :param validate_code: 验证码 |
| | | :return: |
| | | """ |
| | | self.send_text(validate_code, Login.validate_code_input) |
| | | |
| | | def login_btn(self): |
| | | """ |
| | | 点击登录按钮 |
| | | :return: |
| | | """ |
| | | self.click(Login.login_btn) |
| | | |
| | | def login(self, username, password, validate_code): |
| | | """ |
| | | 登录 |
| | | :param username: 用户名 |
| | | :param password: 密码 |
| | | :param validate_code: 验证码 |
| | | :return: |
| | | """ |
| | | self.username_input(username) |
| | | self.password_input(password) |
| | | self.validate_code_input(validate_code) |
| | | self.login_btn() |
| | | |
| | | def is_login_success(self): |
| | | def login_assert(self): |
| | | """ |
| | | 判断是否登录成功 |
| | | :return: bool |
| | | """ |
| | | current_url = self.driver.current_url |
| | | if "index" in self.driver.current_url: |
| | | try: |
| | | ele = self.get_ele(self.es.home_logo, timeout=10) |
| | | if ele.text == "TAI-OTS": |
| | | return True |
| | | else: |
| | | except TimeoutException: |
| | | return False |
| | | |
| | | |
| | | |
| | | |