1 files changed, 59 insertions, 0 deletions
diff --git a/app/test/test_add_word.py b/app/test/test_add_word.py
new file mode 100644
index 0000000..5cf6b6c
--- /dev/null
+++ b/app/test/test_add_word.py
@@ -0,0 +1,59 @@
+# Run the docker image using the following command:
+# docker run -d -p 4444:4444 selenium/standalone-chrome
+from selenium import webdriver
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+
+import random, time
+
+driver = webdriver.Remote('http://localhost:4444/wd/hub', DesiredCapabilities.CHROME)
+
+HOME_PAGE = 'http://121.4.94.30:91/'
+
+
+
+def test_add_word():
+    driver.get(HOME_PAGE)
+    assert 'English Pal -' in driver.page_source
+
+    # login
+    elem = driver.find_element_by_link_text('登录')
+    elem.click()
+
+    uname = 'lanhui'
+    password = 'l0ve1t'
+    elem = driver.find_element_by_name('username')
+    elem.send_keys(uname)
+
+    elem = driver.find_element_by_name('password')
+    elem.send_keys(password)
+    
+    elem = driver.find_element_by_xpath('//form[1]/p[3]/input[1]') # 找到登录按钮
+    elem.click()
+
+    assert 'EnglishPal Study Room for ' + uname in  driver.title
+
+    # get essay content
+    driver.save_screenshot('./app/test/test_add_word_pic0.png')    
+    elem = driver.find_element_by_id('text-content')
+    essay_content = elem.text
+
+    # click Next
+    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:
+        word = random.choice(essay_content.split())        
+    elem.send_keys(word)
+    driver.save_screenshot('./app/test/test_add_word_pic1.png')        
+    elem = driver.find_element_by_xpath('//form[1]//input[1]') # 找到get所有词频按钮
+    elem.click()
+
+    elems = driver.find_element_by_xpath("//input[@type='checkbox']")
+    for elem in elems:
+        if elem.get_attribute('name') == 'marked':
+            elem.click()
+
+    driver.save_screenshot('./app/test/test_add_word_pic2.png')
+    elem = driver.find_element_by_xpath('//form[1]/input[1]') # 找到加入我的生词簿按钮
+    elem.click()
+    
+    driver.quit()