上传文件至 app/test

app/test/test_bug546_lixiaofeng.py

@@ -1,35 +1,15 @@
-import pytest
-from selenium import webdriver
 from selenium.webdriver.common.action_chains import ActionChains
-import time
-from webdriver_manager.chrome import ChromeDriverManager
+from helper import signup
 
-@pytest.fixture
-def driver():
-    # 初始化浏览器驱动
-    driver = webdriver.Chrome(ChromeDriverManager().install())
+
+def test_highlight(driver, URL):
+    try:
+        # 打开网页
+        driver.get(URL)
         driver.maximize_window()
 
-    yield driver
-    # 测试结束后关闭浏览器
-    driver.quit()
-
-def test_highlight(driver):
-    # 打开网页
-    driver.get("http://127.0.0.1:5000/")
-
-    # login
-    elem = driver.find_element_by_link_text('登录')
-    elem.click()
-    uname = 'lxf'
-    password = 'kklo567'
-    elem = driver.find_element_by_id('username')
-    elem.send_keys(uname)
-    elem = driver.find_element_by_id('password')
-    elem.send_keys(password)
-
-    elem = driver.find_element_by_class_name('btn')  # 找到登录按钮
-    elem.click()
+        # 注册
+        signup(URL, driver)
 
         # 取消勾选“划词入库按钮”
         highlight_checkbox = driver.find_element_by_id("test1")
@@ -51,4 +31,9 @@ def test_highlight(driver):
         actions.release()
         # 执行操作链
         actions.perform()
-    time.sleep(10)
+        # time.sleep(10)
+
+        assert driver.find_element_by_tag_name("mark") is not None
+    finally:
+        # 测试结束后关闭浏览器
+        driver.quit()