02-Python Unittest 自动化单元测试框架Demo

1. unittest框架概述

• unittest单元测试框架不仅可以适用于单元测试，还可以适用WEB自动化测试用例的开发与执行，
• 该测试框架可组织执行测试用例，并且提供了丰富的断言方法，判断测试用例是否通过，最终生成测试结果。
• unittest是一个单元测试框架，是Python编程的单元测试框架。有时候，也做叫做“PyUnit”,是Junit的Python语言版本。这里了解下,Junit是Java语言的单元测试框架，Java还有一个很好用的单元测试框架叫TestNG,本系列只学习Python，所以只需要unittest是Python里的一个单元测试框架就可以了。
• 当我们写的用例越来越多时，我们就需要考虑用例编写的规范与组织，以便于后期的维护，而unittest正是这样一款工具。

2. unittest框架原理

unittest支持测试自动化，共享测试用例中的初始化和关闭退出代码，在unittest中最小单元是test，也就是一个测试用例。要了解unittest单元测试框架，先来了解以下几个重要的概念。

• 测试固件（test fixture）
  一个测试固件包括两部分，执行测试代码之前的准备部分和测试结束之后的清扫代码。这两部分一般用函数setUp()和tearDown()表示。这里举例以下，例如要测试百度搜索selenium这个场景，我们的测试固件可以这样写，setUp()里写打开浏览器，浏览器最大化，和打开百度首页等脚本代码；在tearDown（）里写结束搜索后，退出并关闭浏览器的代码。
• 测试用例（test case）
  unittest中管理的最小单元是测试用例，一个测试用例，包括测试固件，和具体测试业务的函数或者方法。一个测试用例中，测试固件可以不写，但是至少有一个已test开头的函数。unittest会自动化识别test开头的函数是测试代码，如果你写的函数不是test开头，unittest是不会执行这个函数里面的脚本的，这个千万要记住，所有的测试函数都要test开头，记住是小写的哦。
• 测试套件 （test suite）
  很简单，就是很多测试用例的集合，叫测试套件，一个测试套件可以随意管理多个测试用例。如果测试用例比作单个学生，测试套件就是好像是班级的概念。
• 测试执行器 （test runner）
  test runner是一个用来执行加载测试用例，并执行用例，且提供测试输出的一个组建。test runner可以加载test case或者test suite进行执行测试任务。

案例

• python 测试框架(本文只涉及 PyUnit)
  https://wiki.python.org/moin/PythonTestingToolsTaxonomy

• 环境准备
  首先确定已经安装有Python，之后通过安装PyUnit，Python版本比较新的已经集成有PyUnit(PyUnit 提供了一个图形测试界面UnittestGUI.py)
  参考： https://wiki.python.org/moin/PyUnit

• 代码实例
  使用的IDE为 PyCharm,DEMO结构如图

  
  
  image.png

• 1.简单地一个实例

# Test002_Fail.py


# -*- coding:utf-8 -*-

import  unittest

class Test002_Fail(unittest.TestCase):

    #测试用例前执行
    def setUp(self):
        print 'Case Before'
        pass

    #测试用例后执行
    def tearDown(self):
        print 'Case After'
        pass

    #测试用例1
    def test_Case1(self):
        a = 3
        b = 2
        self.assertEqual(a+b,4,'Result Fail')

    #测试用例2
    def test_Case2(self):
        a = 2
        b = 3
        self.assertEqual(a*b,7,'Result Fail')

if __name__ == '__main__':
    unittest.main()

2.当case比较多的时候，可以使用Suite来管理多个case,suite 可以看做是一个case 的容器，可以装很多的case

# SuiteDemo.py


# -*- coding:utf-8 -*-


#一个类，添加测试集


import  unittest

class SuiteDemo(unittest.TestCase):


    #测试用例前执行
    def setUp(self):
        print 'Case Before'
        pass

    #测试用例后执行
    def tearDown(self):
        print 'Case After'
        pass


    def test_Case1(self):
        a = 3
        b = 2
        self.assertEqual(a+b,5,'Result Fail')
        print 'Case1'


    def test_Case2(self):
        a = 2
        b = 3
        self.assertEqual(a*b,6,'Result Fail')
        print 'Case2'



#定义一个测试集合，方便添加Case

def suite():
    suiteTest = unittest.TestSuite()
    suiteTest.addTest(SuiteDemo("test_Case1"))
    suiteTest.addTest(SuiteDemo("test_Case2"))
    return suiteTest



#默认运行时通过 Suite 运行

if __name__ == '__main__':
    unittest.main(defaultTest='suite')

3.Suite 和 Case可以分开写在不同的Python文件中，这样方便区分case模块和Suite管理模块，在测试测试的时候只需要在case模块里面添加case,之后添加到Suite模块即可了
case模块:

# TestCaseDemo


# -*- coding:utf-8 -*-


import unittest


class TestCaseDemo(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_Case1(self):
        print 'test_Case1'


    def test_Case2(self):
        print 'test_Case2'


    def test_Case3(self):
        print 'test_Case3'


    def test_Case4(self):
        print 'test_Case4'

接下来为三个Suite模块，都是先导入Case模块，之后添加Suite,再在Suite里面添加Case，最后执行的时候需要使用TestRunner,这个TestRunner 相当于Suite 的执行器
简单地TestSuite:

# TestCase_Demo1.py


# -*- coding:utf-8 -*-


#集合写入一个方法，main调用并启动



import unittest
from TestCase_Demo import TestCaseDemo


#添加一个测试集合，并添加Case

def suite():
    suiteTest = unittest.TestSuite()
    suiteTest.addTest(TestCaseDemo('test_Case1'))
    return suiteTest



#指定并启动测试集合,运行集合方法

if __name__ == '__main__':

    runner = unittest.TextTestRunner()
    runner.run(suite())

也可以不定义Suite方法，直接在main方法中进行Case的添加，之后执行

# TestCase_Demo2.py


# -*- coding:utf-8 -*-


#集合可以写在main


import unittest
from TestCase_Demo import TestCaseDemo


#指定并启动测试集合

if __name__ == '__main__':

    #添加测试集合Case,并启动
    suiteTest = unittest.TestSuite()
    suiteTest.addTest(TestCaseDemo('test_Case1'))
    suiteTest.addTest(TestCaseDemo('test_Case2'))

    #直接启动集合
    runner = unittest.TextTestRunner()
    runner.run(suiteTest)

可以把不同的Case 添加道不同的Suite里面，同样，可以给Suite分组，用于区分不同的Case,并且，Suite 之间还可以包含

# TestCase_Demo3.py


# -*- coding:utf-8 -*-


#包含集合，多集合



import unittest
from TestCase_Demo import TestCaseDemo



#添加不同的合集

def Suite1():
    suiteTest = unittest.TestSuite()
    suiteTest.addTest(TestCaseDemo("test_Case1"))
    suiteTest.addTest(TestCaseDemo("test_Case2"))
    print 'Suite1 运行'
    return suiteTest


def Suite2():
    suiteTest = unittest.TestSuite()
    suiteTest.addTest(TestCaseDemo("test_Case3"))
    suiteTest.addTest(TestCaseDemo("test_Case4"))
    print 'Suite2 运行'
    return suiteTest



#包含所有的Suite

def AllSuite():
    allTest = unittest.TestSuite((Suite1(),Suite2()))
    return allTest



#运行的时候，可以根据不同的要求，运行不同的Suite,或者全部运行，这样就方便管理每次运行的case

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(AllSuite())

小结

PyUnit 是Python的一个单元测试框架，使用它可以方便的管理自动化运行的case.
在Python 中，每一个方法 只要方法名使用test_ 开头的都被作为是一个用例，都可以看做成一个case,这里介绍了使用Suite管理Case，并且分组运行Case,还有其他很多的功能，请参考：
https://docs.python.org/2.7/library/unittest.html