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=True

1.-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 True

is_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 True

is_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=True

close_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开头的方法。