前言
为什么需要单元测试？
如果没有单元测试，我们会遇到这种情况：已有的健康运行的代码在经过改动之后，我们无法得知改动之后是否引入了Bug。如果有单元测试的话，只要单元测试全部通过，我们就可以保证没有Bug被引入。因此，单元测试是保证软件工程质量的一个很重要的方面。
Python中的单元测试
Python最强大的地方在于，开发效率高，并且有丰富的Package，避免重复造轮子。那么Python中的Unittest模块有很丰富的功能提供给我们调用：完整的测试框架，丰富的拓展，比如我们可以设置测试之前的一些初始化工作，比如链接数据库等，规划测试集中有哪些测试用例需要跳过，以及跳过的原因。
Unittest中几个类(Class)的基本概念
TestCase 是我们要写的具体的测试用例
TestSuite 多个测试用例集合在一起，中文翻译过来叫测试套件，其实就是测试集。
TestLoader是用来加载TestCase到TestSuite中的（更通俗一点，就是用来把符合我们定义的条件的测试用例组合起来，成为一个测试集），一般会以参数的形式传进去一些条件，比如收集某个目录下所有的test case组成新的测试集。
TestRunner是来执行测试用例的,测试的结果会保存到TestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息
一个简单的测试例子
>>> class MyTest(unittest.TestCase):
#Run before whole testcase set execution, decorator classmethod is essential
@classmethod
def setUpClass(self):
print("UnitTest Begin...")
#Run after whole testcase set execution, decorator classmethod is essential
@classmethod
def tearDownClass(self):
print("UnitTest End...")
#Run before each test case execution
def setUp(self):
print("Begin...")
#Run after each test case execution
def tearDown(self):
print("End...")
def test_1(self):
self.assertEqual(1,1)
def test_2(self):
self.assertEqual(1,2)
>>> if __name__ == '__main__':unittest.main()
UnitTest Begin...
Begin...
End...
.Begin...
End...
FUnitTest End...
======================================================================
FAIL: test_2 (__main__.MyTest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "<pyshell#41>", line 15, in test_2
AssertionError: 1 != 2
----------------------------------------------------------------------
Ran 2 tests in 0.097s
FAILED (failures=1)
在这个例子中，有几个函数要注意：
setUp()和tearDown()：每个test case执行之前和执行之后要运行的操作：我们可以在这里定义测试的准备工作，比如链接数据库，web登录等等。
用装饰器classmethod装饰的setUpClass()和tearDownClass(): 跑类中定义的所有test cases之前和之后，需要执行的操作。
test_1()和test_2()，具体的测试用例，一定要以test为开头，因为unittest框架中，定义为，如果TestCase类中以test为开头的函数，将会作为具体的testcase收录进要执行的测试集里。
self.assertEqual()，是TestCase类中的断言函数，用来做判断的，用以判断该条测试用例是否通过。通过名字我们可以看出，这个函数的意思是判断两个值是否相等，如果相等，则用例通过，如果不等，则用例不通过。类似的，断言函数还有很多：有一个msg参数，如果指定msg参数的值，则将该信息作为失败的错误信息返回
三种常见测试写法
第一种： 搜索该模块下所有以test开头的测试用例方法，并自动执行它们
#执行测试用例方案一如下：
#unittest.main()方法会搜索该模块下所有以test开头的测试用例方法，并自动执行它们。
import unittest
#定义测试类，父类为unittest.TestCase。
#可继承unittest.TestCase的方法，如setUp和tearDown方法，不过此方法可以在子类重写，覆盖父类方法。
#可继承unittest.TestCase的各种断言方法。
class Test(unittest.TestCase):
def setUp(self):
self.number=raw_input('Enter a number:')
self.number=int(self.number)
#定义测试用例，以“test_”开头命名的方法
#可使用unittest.TestCase类下面的各种断言方法用于对测试结果的判断
def test_case1(self):
print self.number
self.assertEqual(self.number,10,msg='Your input is not 10')
def test_case2(self):
print self.number
self.assertEqual(self.number,20,msg='Your input is not 20')
@unittest.skip('暂时跳过用例3的测试')
def test_case3(self):
print self.number
self.assertEqual(self.number,30,msg='Your input is not 30')
def tearDown(self):
print 'Test over'
#如果直接运行该文件(__name__值为__main__),则执行以下语句，常用于测试脚本是否能够正常运行
if __name__=='__main__':
#执行顺序是命名顺序：先执行test_case1，再执行test_case2
unittest.main()第二种： 构造测试集，实例化test suite(即TestRunner类), 运行test suite中所有的测试用例。
'''
执行测试用例方案二如下：
先构造测试集
实例化测试套件
'''
suite=unittest.TestSuite()
#将测试用例加载到测试套件中。
#执行顺序是安装加载顺序：先执行test_case2，再执行test_case1
suite.addTest(Test('test_case2'))
suite.addTest(Test('test_case1'))
#执行测试用例
#实例化TextTestRunner类
runner=unittest.TextTestRunner()
#使用run()方法运行测试套件（即运行测试套件中的所有用例）
runner.run(suite)
第三种：通过收集指定目录下的目标测试用例，构造测试集再执行
#构造测试集（简化了方案二中先要创建测试套件然后再依次加载测试用例）
#执行顺序同方案一：执行顺序是命名顺序：先执行test_case1，再执行test_case2
test_dir = './'
discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')
#执行测试用例
#实例化TextTestRunner类
runner=unittest.TextTestRunner()
#使用run()方法运行测试套件（即运行测试套件中的所有用例）
runner.run(discover)
如何生成HTML和XML格式的测试报告
上面我们得到的测试结果是文本格式的，可读性不好，并且也无法直接用来作后续的测试结果数据处理，比如测试结果的分类统计等等。
那么有两种方式可供我们选择：HTML和XML
HTML格式的报告，就是网页格式，可读性会比较好。XML格式的报告，则比较方便作后续的数据处理。
需要注意的是，我们需要安装额外的package，即HtmlTestRunner和xmlrunner。
在配置好Pip的前提下，可以通过以下命令安装：
pip install html-testrunner
pip instll xmlrunner如果没有配置好pip或者用pip安装失败，则需要用以下方式安装(xmlrunner同理)：
#!/usr/bin/python3
# -*- coding: utf-8 -*-
import unittest
import HtmlTestRunner
class TestStringMethods(unittest.TestCase):
def test_upper(self):
self.assertEqual('foo'.upper(),'FOO')
def test_isupper(self):
self.assertFalse('Foo'.isupper())
def test_split(self):
s = 'hello world'
self.assertEqual(s.split(),['hello','world'])
with self.assertRaises(TypeError):
s.split(2)
if __name__ == '__main__':
suite = unittest.TestSuite()
suite.addTest(TestStringMethods('test_upper'))
suite.addTest(TestStringMethods('test_isupper'))
suite.addTest(TestStringMethods('test_split'))
runner = HtmlTestRunner.HTMLTestRunner(output='MyUnitTest')
runner.run(suite)最终我们会得到一个可读性比较好的网页报告。
XML报告：
有时我们需要得到格式化数据的测试报告，此时XML格式就要比HTML格式好的多。
因为XML格式的test result容易被读取和数据处理。
示例代码如下：
#!/usr/bin/python3
# -*- coding: utf-8 -*-
import unittest
import xmlrunner
class TestStringMethods(unittest.TestCase):
def test_upper(self):
self.assertEqual('foo'.upper(),'FOO')
def test_isupper(self):
self.assertFalse('Foo'.isupper())
def test_split(self):
s = 'hello world'
self.assertEqual(s.split(),['hello','world'])
with self.assertRaises(TypeError):
s.split(2)
if __name__ == '__main__':
suite = unittest.TestSuite()
suite.addTest(TestStringMethods('test_upper'))
suite.addTest(TestStringMethods('test_isupper'))
suite.addTest(TestStringMethods('test_split'))
#fp = open('result.html','w')
runner = xmlrunner.XMLTestRunner(output='MyUnitTest')
#runner = HtmlTestRunner.HTMLTestRunner(stream=fp,output='MyUnitTest')
runner.run(suite)
得到的结果是这样的：
<?xml version="1.0"?>
-<testsuite time="0.000" tests="3" name="TestStringMethods-20181115000346" failures="0" errors="0">
<testcase time="0.000" name="test_upper" classname="TestStringMethods"/>
<testcase time="0.000" name="test_isupper" classname="TestStringMethods"/>
<testcase time="0.000" name="test_split" classname="TestStringMethods"/>
-<system-out>
<![CDATA[]]>
</system-out>
-<system-err>
<![CDATA[]]>
</system-err>
</testsuite>
几个利用unittest做测试的实际例子
百度搜索测试用例
from selenium import webdriver
import unittest, time
class BaiduTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(30) #隐性等待时间为30秒
self.base_url = "https://www.baidu.com"
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("unittest")
driver.find_element_by_id("su").click()
time.sleep(3)
title=driver.title
self.assertEqual(title, u"unittest_百度搜索")
def tearDown(self):
self.driver.quit()
if __name__ == "__main__":
unittest.main()有道翻译测试用例
from selenium import webdriver
import unittest, time
class YoudaoTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(30) #隐性等待时间为30秒
self.base_url = "http://www.youdao.com"
def test_youdao(self):
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("translateContent").clear()
driver.find_element_by_id("translateContent").send_keys(u"你好")
driver.find_element_by_id("translateContent").submit()
time.sleep(3)
page_source=driver.page_source
self.assertIn( "hello",page_source)
def tearDown(self):
self.driver.quit()
if __name__ == "__main__":
unittest.main()web测试用例：通过测试套件TestSuite来组装多个测试用例。
import unittest
from test_case import test_baidu
from test_case import test_youdao
#构造测试集
suite = unittest.TestSuite()
suite.addTest(test_baidu.BaiduTest('test_baidu'))
suite.addTest(test_youdao.YoudaoTest('test_youdao'))
if __name__=='__main__':
#执行测试
runner = unittest.TextTestRunner()
runner.run(suite)
参考链接：
2. unittest — Unit testing framework https://docs.python.org/3/library/unittest.html
来源：oschina
链接：https://my.oschina.net/u/4352268/blog/3277122