python + unittest 做单元测试之学习笔记

Posted nancy05

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python + unittest 做单元测试之学习笔记相关的知识,希望对你有一定的参考价值。

单元测试在保证开发效率、可维护性和软件质量等方面有很重要的地位,所谓的单元测试,就是对一个类,一个模块或者一个函数进行正确性检测的一种测试方式。

这里主要是就应用 python + unitest 做单元测试做的一些学习笔记。

 

Unittest 提供了test cases, test suites, test fixtures,test runner:

  • test case :通过继承TestCase类,我们可以创建一个test,或者一组tests
  • test suites : 测试套件, 使用时要将test case 装载到测试套件里
  • test fixtures : setup + test case + teardown,每个test case执行的时候都会运行setUp->test_case->tearDown ,因此setUp 和 tearDown 可以用来做环境的初始化。例如,写一个有关数据操作的类,setUp放连接数据库,tearDown 放关闭数据库连接。

一、unittest.main() 的使用

 

所有的test case 要以test开头,例如 def test_pass_case(self)

具体的单元测试例子如下:

import unittest
def calc(x,y):
    return x + y

class TestCalc(unittest.TestCase):
    def test_pass_case(self):
        \'\'\'这是通过的测试用例\'\'\'
        print(\'这个是通过用例\')
        res = calc(1,2)
        self.assertEqual(3,res)
        self.assertNotEqual(2,3)
    def setUp(self):#每个用例执行前都会执行它
        print(\'我是setUp\')
    def tearDown(self):#每个用例运行之后都会执行它
        print(\'我是tearDown\')

    @classmethod
    def setUpClass(cls):#所有的用例执行之前执行一次,setUp()前
        print(\'我是setupClass\')#所有的用例都运行完之后执行一次,setDown 后
    @classmethod
    def tearDownClass(cls):#所有的用例执行之后执行一次,setDown()后
        print(\'我是teardownClass\')
    def test_a(self):
        \'\'\'这是普通的测试用例\'\'\'
        print(\'a\')
    def test_fail_case(self):
        \'\'\'这是失败的测试用例\'\'\'
        print(\'这个是失败的测试用例\')
        res = calc(9,8)
        self.assertEqual(98,res)
    def test_haha(self):
        \'\'\'这是哈哈\'\'\'
        self.assertEqual(1,2)

if __name__ == \'__main__\':
    unittest.main()#它会帮你运行当前这个python里面所有的测试用例

 

由此可以看到,执行顺序是:静态方法 setUpClass()  -> setUp -> test case -> SetDown ->... -> SetUp -> testcase -> testDown -> 静态方法 setDownClass()。

所以可以用setUpClass、tearDownClass做所有用例的前置初始化和结束。

unittest.main()  #它会帮你运行当前这个python里面所有的测试用例

 

二、 应用 HTMLTestRunner 生成测试报告

if __name__ == \'__main__\':
    suite = unittest.TestSuite()#定义一个测试套件
    suite.addTest(TestCalc(\'test_pass_case\'))#测试套件中添加测试用例
    suite.addTest(TestCalc(\'test_a\'))
    f = open(\'report.html\',\'wb\')#打开一个保存结果的html文件
    runner = HTMLTestRunner.HTMLTestRunner(stream=f,title=\'测试报告\',description=\'测试情况\')#生成执行用例的对象
    runner.run(suite)#运行

可以将某几个test case加入test suite ,这样生成的测试报告里只有这两个test case 的结果。

具体如下:

可以将类里所有的test case加入test suite,如下

suite.addTests(unittest.makeSuite(TestCalc))#把类里面所有的测试用例加入测试套件

得到的测试报告结果为:

三、针对整个项目的test case,生成测试报告

例如有以下的结构:

 

test_buy 里关于购买的cases, test_login是关于login 的cases, test_reg是关于注册的cases, run_all_cases执行所有的cases

注意,在test_buy, test_login,test_reg里的case 需要以test 开头,这样才会被执行到,还有这三个文件不一定以test开头,这里只是为了方便才这么写的,但是必须得有相同的规则,比如这里的文件名都以test_ 开头。

run_all_cases.py代码如下所示:

import unittest,HTMLTestRunner
suite = unittest.TestSuite()
all_cases = unittest.defaultTestLoader.discover(\'.\',\'test_*.py\')#第一个参数:从哪个目录,第二个参数:定义规则
# for case in all_cases:
#     suite.addTests(case)
[suite.addTests(case) for case in all_cases]#列表生成式,和上面的注释写法一样
f = open(\'new_report.html\',\'wb\')#打开一个保存结果的html文件
runner = HTMLTestRunner.HTMLTestRunner(stream=f,title=\'测试报告\',description=\'测试情况\')#生成执行用例的对象
runner.run(suite)#运行

 

四、集成jenkins,生成jenkins能识别的测试报告

这里就要用到  xmlrunner 模块,jenkins能识别xml格式。

runner = xmlrunner.XMLTestRunner(\'.\')#当前目录

 

 以上就是对Unittest的简单总结,后续会持续更新。

以上是关于python + unittest 做单元测试之学习笔记的主要内容,如果未能解决你的问题,请参考以下文章

python_unittest_单元测试

python单元测试-unittest

Python单元测试--unittest

Python单元测试unittest

Python单元测试unittest

Python单元测试unittest测试框架