forked from mrlan/EnglishPal
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
|
import random
|
|||
|
|
import string
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
from selenium import webdriver
|
|||
|
|
from selenium.webdriver.support.wait import WebDriverWait
|
|||
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|||
|
|
|
|||
|
|
HOME_PAGE = "http://127.0.0.1:5000/"
|
|||
|
|
|
|||
|
|
word = []
|
|||
|
|
uname = 'lanhui'
|
|||
|
|
password = 'l0ve1t'
|
|||
|
|
def has_punctuation(s): #用于检查单词是否包含标点符号
|
|||
|
|
return [c for c in s if c in string.punctuation] != []
|
|||
|
|
|
|||
|
|
def login(driver):
|
|||
|
|
driver.get(HOME_PAGE)
|
|||
|
|
assert 'English Pal - Learn English smartly!' in driver.page_source
|
|||
|
|
# 登录
|
|||
|
|
login_elem = driver.find_element_by_link_text('登录')
|
|||
|
|
login_elem.click()
|
|||
|
|
|
|||
|
|
username_input = driver.find_element_by_id('username')
|
|||
|
|
username_input.send_keys(uname)
|
|||
|
|
|
|||
|
|
password_input = driver.find_element_by_id('password')
|
|||
|
|
password_input.send_keys(password)
|
|||
|
|
|
|||
|
|
login_btn = driver.find_element_by_xpath('//button[text()="登录"]') # 找到登录按钮
|
|||
|
|
login_btn.click()
|
|||
|
|
|
|||
|
|
# 确保页面加载完
|
|||
|
|
try:
|
|||
|
|
WebDriverWait(driver, 10).until(
|
|||
|
|
EC.title_is("EnglishPal Study Room for " + uname) # 替换为实际的页面标题
|
|||
|
|
)
|
|||
|
|
except Exception as e:
|
|||
|
|
print("页面加载失败:", e)
|
|||
|
|
driver.quit()
|
|||
|
|
exit()
|
|||
|
|
assert 'EnglishPal Study Room for ' + uname in driver.title
|
|||
|
|
|
|||
|
|
def test_save_selected_word():
|
|||
|
|
global word
|
|||
|
|
# 调用本地chromedriver
|
|||
|
|
driver = webdriver.Chrome(executable_path="C:\Program Files\Google\Chrome\Application\chromedriver.exe")
|
|||
|
|
try:
|
|||
|
|
login(driver)
|
|||
|
|
#点击单词添加到已选单词列表
|
|||
|
|
# 获取文章内容
|
|||
|
|
elem = driver.find_element_by_id('text-content')
|
|||
|
|
essay_content = elem.text
|
|||
|
|
|
|||
|
|
#随机选择单词 长度不小于6个字符，并且不包含标点符号
|
|||
|
|
elem = driver.find_element_by_id('selected-words')
|
|||
|
|
word = random.choice(essay_content.split())
|
|||
|
|
while 'font>' in word or 'br>' in word or 'p>' in word or len(word) < 6 or has_punctuation(word):
|
|||
|
|
word = random.choice(essay_content.split())
|
|||
|
|
print(type(word)) #str
|
|||
|
|
elem.send_keys(word) #单词输入到输入框
|
|||
|
|
|
|||
|
|
# 检查单词是否已成功添加到已选列表
|
|||
|
|
# 再次获取<textarea>中的文本内容，以确认单词已被输入
|
|||
|
|
updated_textarea_content = elem.get_attribute('value')
|
|||
|
|
# 检查<textarea>中的内容是否包含输入的单词
|
|||
|
|
assert word == updated_textarea_content
|
|||
|
|
# 检查是否保存到 localStorage
|
|||
|
|
stored_words = driver.execute_script('return localStorage.getItem("selectedWords");')
|
|||
|
|
assert word == stored_words
|
|||
|
|
#检查退出登录、关闭web页面后是否保存已选单词
|
|||
|
|
# 退出登录
|
|||
|
|
elem = driver.find_element_by_link_text('退出')
|
|||
|
|
elem.click()
|
|||
|
|
|
|||
|
|
# 关闭当前标签页
|
|||
|
|
driver.execute_script("window.open('');window.close();")
|
|||
|
|
|
|||
|
|
# 等待一会儿，让浏览器有足够的时间关闭标签页
|
|||
|
|
time.sleep(2)
|
|||
|
|
|
|||
|
|
# 重新打开一个新的标签页
|
|||
|
|
driver.execute_script("window.open('');")
|
|||
|
|
driver.switch_to.window(driver.window_handles[-1]) # 切换到新打开的标签页
|
|||
|
|
|
|||
|
|
login(driver)
|
|||
|
|
elem = driver.find_element_by_id('selected-words')
|
|||
|
|
textarea_content = elem.get_attribute('value')
|
|||
|
|
print(f'word:{word}')
|
|||
|
|
print(f'selected:{textarea_content}')
|
|||
|
|
assert word == textarea_content
|
|||
|
|
except Exception as e:
|
|||
|
|
print("An error occurred:", e)
|
|||
|
|
finally:
|
|||
|
|
# 关闭浏览器
|
|||
|
|
driver.quit()
|
|||
|
|
|
|||
|
|
test_save_selected_word()
|