用python实现接口测试(九、Unittest测试框架)-CSDN博客

一、单元测试框架

不论对于开发还是测试，都离不开单元测试框架，对于开发而言使用单元测试框架，可以编写测试代码来验证验证自己编写的功能是否正确，对于测试而言，使用单元测试框架，可以编写自动化的测试用例，在Python中单元测试框架是Pyunit，即unittest,unittest我一直认为是一个很优秀的单元测试框架。

官方详细的地址是：https://docs.python.org/2/library/unittest.html

单元测试支持测试自动化、共享的安装程序和关闭代码测试、聚合成集合，测试和报告框架从测试的独立性。单元测试模块提供可以很容易地支持这些素质的一组测试的类。

二、unittest框架间各个模块的联系

模块间的联系

在一个完整的单元测试用例中，是包含了测试固件(setUp()和tearDown())，在测试执行的阶段，我们更加愿意使用测试套件(TestSuite())来组织每个测试用例来执行(TestRunner)并得到测试结果（TestReport），什么是测试固件，在unittest中，setUp()与tearDown()被成为测试固件，某些人称为钩子（仅仅只一个称呼而已），它的主要目标初始化测试用例，执行测试用例后，对测试用例执行的结果做后期的处理。

三、测试用例

以下是二个测试用例，不管执行那个测试用例，都会执行setUp()和tearDown()，也就是说，在一个测试类中，如果有N个测试用例，在执行该测试类中的测试用例的时候，会执行N次setUp()和tearDown()，见代码：

#coding:utf-8

import unittest

def div(a,b):

return a/b

class TestDiv(unittest.TestCase):

def setUp(self):

print u'开始...'

def tearDown(self):

print u'结束...'

def test_001(self):

self.assertEqual(div(1,1),1)

def test_002(self):

self.assertRaises(ZeroDivisionError,div,1,0)

if __name__=='__main__':

unittest.main(verbosity=2)

运行结果：

运行结果

可以看到，执行了二个测试用例，也执行了2次setUp()和tearDown()方法，如果这样你感觉不明显，可以结合selenium的测试框架来看更加直观，见代码：

#coding:utf-8

import unittest

from seleniumimport webdriver

def div(a,b):

return a/b

class TestDiv(unittest.TestCase):

def setUp(self):

self.driver=webdriver.Chrome()

self.driver.get('http://www.baidu.com')

def tearDown(self):

self.driver.quit()

def test_001(self):

self.assertEqual(self.driver.title,u'百度一下，你就知道')

def test_002(self):

self.assertRaises(self.driver.current_url,'http://www.baidu.com/')

if __name__=='__main__':

unittest.main(verbosity=2)

执代码后，会看到打开浏览器二次，当然关闭浏览器也是二次。

四、代码优化

在一个测试类中，测试用例是非常多的，这样添加或者删除实在是浪费时间，我们可以把源码修改下，来实现执行一个测试，某些用例不执行的可以忽略，使用的方法是makeSuite()

#coding:utf-8

import unittest

from seleniumimport webdriver

def div(a,b):

return a/b

class TestDiv(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.driver=webdriver.Chrome()

cls.driver.get('http://www.baidu.com')

@classmethod

def tearDownClass(cls):

cls.driver.quit()

def test_001(self):

self.assertEqual(self.driver.title,u'百度一下，你就知道')

@unittest.skip(u'忽略该测试用例,谢谢!')

def test_002(self):

self.assertEqual(self.driver.current_url,'https://www.baidu.com/')

if __name__=='__main__':

suite=unittest.TestSuite(unittest.makeSuite(TestDiv))

unittest.TextTestRunner(verbosity=2).run(suite)

运行结果：

makeSuite()运行结果

五、添加HTML测试报告

在python中，提供了HTMLTestRunner.py来生成测试报告，把该文件下载后，直接放到C:Python27的Lib的目录下，就可以导入该模块使用了，见代码：

#coding:utf-8

import unittest

from seleniumimport webdriver

import HTMLTestRunner

def div(a,b):

return a/b

class TestDiv(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.driver=webdriver.Chrome()

cls.driver.get('http://www.baidu.com')

@classmethod

def tearDownClass(cls):

cls.driver.quit()

def test_001(self):

self.assertEqual(self.driver.title,u'百度一下，你就知道')

@unittest.skip(u'忽略该测试用例,谢谢!')

def test_002(self):

self.assertEqual(self.driver.current_url,'https://www.baidu.com/')

if __name__=='__main__':

suite=unittest.TestSuite(unittest.makeSuite(TestDiv))

#unittest.TextTestRunner(verbosity=2).run(suite)

runner = HTMLTestRunner.HTMLTestRunner(

stream=file('testReport.html', 'wb'),

title=u'TestReport',

description=u'测试报告详细信息'

)

runner.run(suite)

运行代码后，会在当前工作目录下生成testReport.html的测试报告，见截图

测试报告截图

注：

1.在打开Chrome浏览器中会提示如图所示：

提示被控制

解决办法如下：

1.在浏览器配置里加个参数，忽略掉这个警告提示语

option = webdriver.ChromeOptions()

option.add_argument('disable-infobars')

2.代码如下

#coding:utf-8

import unittest

from seleniumimport webdriver

import HTMLTestRunner

def div(a,b):

return a/b

# 加启动配置

option = webdriver.ChromeOptions()

option.add_argument('disable-infobars')

class TestDiv(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.driver=webdriver.Chrome(chrome_options=option)

cls.driver.get('http://www.baidu.com')