forked from mrlan/EnglishPal
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
import random
|
||
|
|
import string
|
||
|
|
import time
|
||
|
|
from selenium.webdriver.common.by import By
|
||
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
||
|
|
from selenium.webdriver.support import expected_conditions as EC
|
||
|
|
from selenium.webdriver.common.action_chains import ActionChains
|
||
|
|
|
||
|
|
from helper import signup
|
||
|
|
|
||
|
|
def has_punctuation(s):
|
||
|
|
return any(c in string.punctuation for c in s)
|
||
|
|
|
||
|
|
def select_one(driver):
|
||
|
|
elem = driver.find_element(By.ID, 'article')
|
||
|
|
essay_content = elem.text
|
||
|
|
valid_word = random.choice([word for word in essay_content.split() if len(word) >= 6 and not has_punctuation(
|
||
|
|
word) and 'font>' not in word and 'br>' not in word and 'p>' not in word])
|
||
|
|
driver.find_element(By.ID, 'selected-words').send_keys(valid_word)
|
||
|
|
driver.find_element(By.ID, 'article').click()
|
||
|
|
return valid_word
|
||
|
|
|
||
|
|
def select_two(driver):
|
||
|
|
word = driver.find_element(By.CLASS_NAME, 'highlighted')
|
||
|
|
|
||
|
|
# 创建ActionChains对象
|
||
|
|
actions = ActionChains(driver)
|
||
|
|
actions.move_to_element(word)
|
||
|
|
|
||
|
|
# 模拟鼠标按下并拖动以选择文本
|
||
|
|
actions.double_click()
|
||
|
|
actions.perform()
|
||
|
|
|
||
|
|
|
||
|
|
def test_selected_second_word(driver, URL):
|
||
|
|
try:
|
||
|
|
signup(URL, driver)
|
||
|
|
selected_words = select_one(driver);
|
||
|
|
assert selected_words.strip() != "", "选中的单词被放置框中"
|
||
|
|
select_two(driver)
|
||
|
|
selected_second_words = driver.find_element(By.ID, 'selected-words').get_attribute('value')
|
||
|
|
assert selected_second_words.strip() == "", "选中的单词被删除"
|
||
|
|
finally:
|
||
|
|
driver.quit()
|