From 8c9395d6a0e07ea64a7ed88a466c5befb83178e3 Mon Sep 17 00:00:00 2001
From: 阳洁 <2077506045@qq.com>
Date: 星期五, 28 七月 2023 18:42:59 +0800
Subject: [PATCH] 测评系统自动测试

---
 po/login_page.py |   74 +++++++++++++------------------------
 1 files changed, 26 insertions(+), 48 deletions(-)

diff --git a/po/login_page.py b/po/login_page.py
index 08b28b2..14cd9cc 100644
--- a/po/login_page.py
+++ b/po/login_page.py
@@ -1,65 +1,43 @@
 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:
-            return True
-        else:
+        try:
+            ele = self.get_ele(self.es.home_logo, timeout=10)
+            if ele.text == "TAI-OTS":
+                return True
+        except TimeoutException:
             return False
 
 
+
+

--
Gitblit v1.9.1