forked from mrlan/EnglishPal
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
|
from selenium.webdriver.common.by import By
|
||
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
||
|
|
from selenium.webdriver.support import expected_conditions as EC
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from helper import signup
|
||
|
|
|
||
|
|
def login(driver, home, uname, password):
|
||
|
|
driver.get(home)
|
||
|
|
WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.LINK_TEXT, '登录'))).click()
|
||
|
|
driver.find_element(By.ID, 'username').send_keys(uname)
|
||
|
|
driver.find_element(By.ID, 'password').send_keys(password)
|
||
|
|
driver.find_element(By.XPATH, '//button[text()="登录"]').click()
|
||
|
|
WebDriverWait(driver, 10).until(EC.title_is(f"EnglishPal Study Room for {uname}"))
|
||
|
|
|
||
|
|
def logout(driver):
|
||
|
|
WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.LINK_TEXT, '退出'))).click()
|
||
|
|
|
||
|
|
# 标记文章
|
||
|
|
def collect_article(driver):
|
||
|
|
driver.find_element(By.XPATH, '//button[text()="标记文章"]').click()
|
||
|
|
|
||
|
|
def test_collect_article(driver, URL):
|
||
|
|
try:
|
||
|
|
username, password = signup(URL, driver)
|
||
|
|
title = driver.find_element(By.ID, 'article_title').text
|
||
|
|
article = driver.find_element(By.ID, 'article').text
|
||
|
|
|
||
|
|
collect_article(driver)
|
||
|
|
collected_title = driver.execute_script('return localStorage.getItem("articleTitle");')
|
||
|
|
assert title == collected_title, "Unable to add the article to your collection."
|
||
|
|
|
||
|
|
# 退出登录
|
||
|
|
logout(driver)
|
||
|
|
|
||
|
|
# 再次登录并检查收藏状态
|
||
|
|
login(driver, URL, username, password)
|
||
|
|
rechecked_title = driver.execute_script('return localStorage.getItem("articleTitle");')
|
||
|
|
assert title == rechecked_title, "Collected article not found after re-login."
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
# 输出异常信息
|
||
|
|
logging.error(e)
|
||
|
|
# 关闭浏览器
|
||
|
|
driver.quit()
|
||
|
|
finally:
|
||
|
|
driver.quit()
|