unittest框架登录测试

from selenium 
import webdriverimport unittest
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time 
class StudentLogin(unittest.TestCase):
   def setUp(self):
         self.driver = webdriver.Chrome()      
         self.url = 'http://yjsglpt.sut.edu.cn/'
   def tearDown(self):
         self.driver.quit()
   def test_login_success(self):
      self.driver.get(self.url)
      self.driver.find_element('id','txtUserID').send_keys('xxxxxxxx')
      self.driver.find_element('id','txtUserPwd').send_keys('xxxxxxx')
      time.sleep(10)
      self.driver.find_element('xpath',"//input[@type='submit']").click()
      self.driver.implicitly_wait(5)      
      name = self.driver.find_element('xpath',"//div[@class='head_t_msg']/span").text
      self.assertEqual('李...', name)

def test_login_without_password(self): 
	self.driver.get(self.url)
	self.driver.find_element('id','txtUserID').send_keys('xxxxxxx')
	time.sleep(10)
	self.driver.find_element('xpath',"//input[@type='submit']").click()
	self.driver.find_element('id','txtUserPwd').clear()
	WebDriverWait(self.driver,10).until(EC.text_to_be_present_in_element(('xpath',"//div[contains(text(),'错误')]"),'账号或密码错误!')
	errmsg = self.driver.find_element('xpath',"//div[contains(text(),'错误')]").text
	self.assertEqual(errmsg,'账号或密码错误!')
if __name__ == '__main__':
   unittest.main()