from base.base_page import BasePage
|
from elements.login_elements import LoginElements
|
from selenium.common.exceptions import TimeoutException
|
|
|
class LoginPage(BasePage):
|
"""
|
登录页面
|
"""
|
# 页面元素
|
es = LoginElements
|
|
def __init__(self, driver):
|
super().__init__(driver)
|
self.driver = driver
|
# 切换到最顶层文档
|
driver.switch_to.default_content()
|
|
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 login_assert(self):
|
"""
|
判断是否登录成功
|
:return: bool
|
"""
|
try:
|
ele = self.get_ele(self.es.home_logo, timeout=10)
|
if ele.text == "TAI-OTS":
|
return True
|
except TimeoutException:
|
return False
|
