前言
python单元测试框架（The Python Unit Testing Framework）简称PyUnit，是JUnit的python版本，自python2.1版本后，PyUnit已经成为了Python的标准库。下面我们就如何把unitest应用到python selenium2自动化测试中进行分享。
测试用例
单元测试是由一系列的测试用例(Test Cases)构成。测试用例是被设置用来检测独立场景的集合。在PyUnit中，unittest模块中的TestCase类代表测试用例。
TestCase类的实例是可以完全运行所有的方法和可选的初始化(setUp)及清理（tearDown）方法的对象。
TestCase实例的测试代码必须是包含一个或多个测试方法，简单说，它可以单独运行或与其他任意数量的用例共同运行
简单示例
# 将以下代码保存到first_webdriver.py中
#-*- coding:utf-8 -*-
__author__ = u'苦叶子'
from selenium import webdriver
import unittest
import HTMLTestRunner
import sys
from time import sleep
reload(sys)
sys.setdefaultencoding("utf-8")
class BaiduTest(unittest.TestCase):
"""百度首页搜索测试用例"""
def setUp(self):
self.driver = webdriver.Ie()
self.driver.implicitly_wait(30)
self.base_url = u"http://www.baidu.com"
def test_baidu_search(self):
driver = self.driver
print u"开始[case_0001]百度搜索"
driver.get(self.base_url)
# 验证标题
self.assertEqual(driver.title, u"百度一下，你就知道")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys(u"开源优测")
driver.find_element_by_id("su").click()
sleep(3)
# 验证搜索结果标题
self.assertEqual(driver.title, u"开源优测_百度搜索")
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
testunit = unittest.TestSuite()
testunit.addTest(BaiduTest('test_baidu_search'))
# 定义报告输出路径
htmlPath = u"testReport.html"
fp = file(htmlPath, "wb")
runner = HTMLTestRunner.HTMLTestRunner(stream=fp,
title=u"百度测试",
description=u"测试用例结果")
runner.run(testunit)
fp.close()
关键代码说明
- 在上例代码中，我们的测试用例BaiduTest继承了unittest.TestCase。
- 在初始化方法setUp中，初始化了webdriver示例和隐性等待设置，并初始化了百度首页url。该方法自动执行。
- 在清理方法中tearDown中，退出了webdriver。该方法自动执行。
- test_baidu_search方法是我们主要的测试方法，在该方法中进行了一下动作。
u1.png
主入口说
- 先定义一个测试套件集，然后将用例集添加至套件中。
- 我们使用了unittest第三方测试报告模块HTMLTestRunner（该模块不是标准库，需要自己去下载，请参考前面几篇的文章）,用以自动生产HTML格式的测试报告。
如何运行上述代码
# 将上述代码保存至first_unit_test.py中
# 在命令行中输入以下命令
python first_unit_test.py
代码组织说明
u2.png
总结
今天我们就unittest怎么和selenium测试进行结合做了初步的分享，请大家深入理解该示例，并实际动手练习代码。
