1 files changed, 24 insertions, 23 deletions
diff --git a/app/test/test_signup.py b/app/test/test_signup.py
index 867ca53..1a5dad9 100644
--- a/app/test/test_signup.py
+++ b/app/test/test_signup.py
@@ -13,31 +13,32 @@ HOME_PAGE = 'http://121.4.94.30:91/'
 
 
 def test_signup():
-    driver.get(HOME_PAGE)
-    driver.save_screenshot('test_signup_pic0.png')
+    try:
+        driver.get(HOME_PAGE)
+        driver.save_screenshot('test_signup_pic0.png')
+        
+        assert 'English Pal -' in driver.page_source
     
-    assert 'English Pal -' in driver.page_source
-
-    elem = driver.find_element_by_link_text('成为会员')
-    elem.click()
-
-    uname = ''.join ( [random.choice (string.ascii_letters) for x in range (8)] )
-    elem = driver.find_element_by_name('username')
-    elem.send_keys(uname)
-
-    elem = driver.find_element_by_name('password')
-    elem.send_keys('iamc00l!')
-
-    driver.save_screenshot('test_signup_pic1.png')
+        elem = driver.find_element_by_link_text('成为会员')
+        elem.click()
     
-    elem = driver.find_element_by_xpath('//form[1]/p[3]/input[1]') # 找到登录按钮
-    elem.click()
-
-    driver.save_screenshot('test_signup_pic2.png')
+        uname = ''.join ( [random.choice (string.ascii_letters) for x in range (8)] )
+        elem = driver.find_element_by_name('username')
+        elem.send_keys(uname)
     
-    assert '恭喜，你已成功注册' in driver.page_source
-    assert uname in driver.page_source
-
-    driver.quit()
+        elem = driver.find_element_by_name('password')
+        elem.send_keys('iamc00l!')
+    
+        driver.save_screenshot('test_signup_pic1.png')
+        
+        elem = driver.find_element_by_xpath('//form[1]/p[3]/input[1]') # 找到登录按钮
+        elem.click()
+    
+        driver.save_screenshot('test_signup_pic2.png')
+        
+        assert '恭喜，你已成功注册' in driver.page_source
+        assert uname in driver.page_source
+    finally:
+        driver.quit()