Selenium 2自动化测试实战33（带unittest的脚本分析）
带unittest的脚本分析
#test.py
#coding:utf-8
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest,time,re
class BaiduTest(unittest.TestCase):
def setUp(self):
self.driver=webdriver.Chrome()
self.driver.implicitly_wait(30)
self.base_url="http://www.baidu.com"
self.verificationErrors=[]
self.accept_next_alert=True
def test_baidu(self):
driver=self.driver
driver.get(self.base_url +"/")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys("selenium ide")
driver.find_element_by_id("su").click()
def is_element_present(self,how,what):
try:
self.driver.find_element(by=how,vaule=what)
except NoSuchElementException,e:
return False
return True
def is_alert_present(self):
try:
self.driver.switch_to_alert()
except NoAlertPresentException,e:
return False
return True
def close_alert_and_get_its_text(self):
try:
alert=self.driver.switch_to_alert()
alert_text=alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally:
self.accept_next_alert=True
def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors)
if __name__=="__main__":
unittest.main()代码分析如下：
Import unittest #首先引入unittest框架
Class BaiduTest(unittest.TestCase): BaiduTest类继承unitt框架的TestCase类成为标砖的测试类。
def setUp(self):
self.driver=webdriver.Chrome()
self.driver.implicitly_wait(30)
self.base_url="http://www.baidu.com"
self.verificationErrors=[]
self.accept_next_alert=True1.-setUp用于设置初始化工作，在执行每一个测试用例前先被执行，它与tearDown方法相呼应，后者在每一个测试用例执行后被执行。这里的初始化工作定义了浏览器启动和基础URL地址。
2.implicitly_wait()设置页面上元素的隐性等待时间为30秒。
3.定义空的verificationErrors数组，脚本运行时的错误信息将被记录到这个数组中。
4.定义accept_next_alert变量，表示是否继续接受下一个警告，初始状态为True。
def test_baidu(self):
driver=self.driver
driver.get(self.base_url +"/")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys("selenium ide")
driver.find_element_by_id("su").click()这里就是测试脚本
def is_element_present(self,how,what):
try:
self.driver.find_element(by=how,vaule=what)
except NoSuchElementException,e:
return False
return Trueis_element_present方法用于查找页面元素是否存在，通过find_element()来接收元素的定位方法(how)和定位值（what），如果定位到元素则返回True，否则抛出异常并返回False。Try…except….为python语言的异常处理。
def is_alert_present(self):
try:
self.driver.switch_to_alert()
except NoAlertPresentException,e:
return False
return Trueis_alert_present()方法用于判断当前页面是否存在警告框，利用webDriver提供的switch_to_alert()方法来捕捉页面上警告框。如果捕捉到警告框则返回True，否则将抛出NoAlertPresentException类型异常，并返回False。
经过验证，不管页面是否出现警告框，返回结果都是True，所有可以调整该方法为driver.switch_to_alert().text，用于获取当前页面上的警告提示信息，可以获取到就返回True，获取不到则返回False。
def close_alert_and_get_its_text(self):
try:
alert=self.driver.switch_to_alert()
alert_text=alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally:
self.accept_next_alert=Trueclose_alert_and_get_its_text()关闭警告并获得警告信息，首先通过switch_to_alert()获得警告，通过text获得警告框信息。接着通过if语句判断accept_next_alert的状态，在setUp()中已经初始状态为True，如果为True，则通过accept（）接受警告，否则dissmiss()忽略此警告。
def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors)tearDown()方法在每个测试方法执行后调用，这个方法用于测试用例执行后的清理工作，如退出浏览器、关闭驱动、恢复用例执行状态等。
在setUp()方法中定义的verificationErrors为空数组，这里通过assertEqual()比较其是否为空，如果为空则说明用例执行的过程中没有出现异常，否则将抛出AssertionError异常。
if __name__=="__main__":
unittest.main()通过unittest.main()方法来运行当前文件中的测试方法，其默认匹配并运行以test开头的方法。