106 lines
4.2 KiB
Python
106 lines
4.2 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 helper import signup
|
||
|
|
|
||
|
|
def has_punctuation(s):
|
||
|
|
return any(c in string.punctuation for c in s)
|
||
|
|
|
||
|
|
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 select_valid_word(driver):
|
||
|
|
elem = driver.find_element(By.ID, 'text-content')
|
||
|
|
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)
|
||
|
|
return valid_word
|
||
|
|
|
||
|
|
|
||
|
|
def select_more_words(driver):
|
||
|
|
list_of_words = [
|
||
|
|
"vulture", "vitreous", "vertebrata", "suture", "sessile", "prodigal", "presumptuous", "prehension","pedunculated",
|
||
|
|
"parturition", "ovigerous", "ova", "orifice", "obliterate", "niggard", "lineal", "glottis", "frena", "ductus",
|
||
|
|
"dorsal", "crustacean", "cornea", "cirriped", "chromatic", "canon","branchiae", "auditory", "articulata",
|
||
|
|
"alimentary", "adduce"
|
||
|
|
]
|
||
|
|
|
||
|
|
# valid_words = [word for word in list_of_words if len(word) >= 6 and not has_punctuation(
|
||
|
|
# word) and 'font>' not in word and 'br>' not in word and 'p>' not in word]
|
||
|
|
words_to_send = ', '.join(list_of_words)
|
||
|
|
driver.find_element(By.ID, 'selected-words').send_keys(words_to_send)
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_user_level(driver, URL):
|
||
|
|
try:
|
||
|
|
username, password = signup(URL, driver)
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
word = select_valid_word(driver)
|
||
|
|
|
||
|
|
# Wait for the button to be clickable
|
||
|
|
time.sleep(10)
|
||
|
|
# Add selected word to the list of words
|
||
|
|
add_words_button = driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
|
||
|
|
add_words_button.click()
|
||
|
|
# Wait for the page to update
|
||
|
|
time.sleep(4)
|
||
|
|
|
||
|
|
add_words_button = driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
|
||
|
|
add_words_button.click()
|
||
|
|
# Wait for the page to update again
|
||
|
|
time.sleep(4)
|
||
|
|
|
||
|
|
# Check if the added word appears in the word container
|
||
|
|
word_container = driver.find_element(By.CLASS_NAME, 'word-container')
|
||
|
|
words = word_container.find_elements(By.TAG_NAME, 'p')
|
||
|
|
added_word_exists = any(word.text == word for word in words)
|
||
|
|
time.sleep(4)
|
||
|
|
|
||
|
|
|
||
|
|
#add words that are not from d2
|
||
|
|
time.sleep(5)
|
||
|
|
word = select_more_words(driver)
|
||
|
|
time.sleep(5)
|
||
|
|
add_button = driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
|
||
|
|
add_button.click()
|
||
|
|
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
#add the words to the list of words
|
||
|
|
add_words_again = driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
|
||
|
|
time.sleep(4)
|
||
|
|
add_words_again.click()
|
||
|
|
# Wait for the page to update again
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
|
||
|
|
# compare user level
|
||
|
|
default_user_level = 3.0
|
||
|
|
user_current_level = driver.find_element(By.ID, "user_level")
|
||
|
|
current_level = float(user_current_level.text)
|
||
|
|
assert current_level > default_user_level, "The user level has dropped to 3 your bugfix is not working"
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
|
||
|
|
# Log out and close the browser
|
||
|
|
driver.find_element(By.XPATH, '//a[contains(@class, "btn-secondary") and text()="退出"]')
|
||
|
|
driver.execute_script("window.open('');window.close();")
|
||
|
|
WebDriverWait(driver, 2)
|
||
|
|
driver.execute_script("window.open('');")
|
||
|
|
driver.switch_to.window(driver.window_handles[-1])
|
||
|
|
|
||
|
|
finally:
|
||
|
|
driver.quit()
|