python之路-untitest单元测试框架组件使用详细介绍

unittest-xmind思维导图

UnitTest介绍

• UnitTest是python自带的自动化测试框架
• UnitTest主要包含的内容
  • TestCase(测试用例)
  • TestSuite(测试套件,把多个TestCase集成到一个测试TestSuite)
  • TestRunner(执行测试用例)
  • TestLoader(自动从代码中加载多个测试用例TestCase)
  • Fixture(UnitTest特性)

TestCase

• 第一步:导入unittest模块
• 第二步:实现一个类,这个类必须继承自unittest.TestCase类
• 第三步:类中每个方法代表一个测试用例,方法名必须以test开头

import unittest

def my_sum(a, b):
    return a + b

class my_test(unittest.TestCase):
    def test_001(self):
        print(my_sum(5, 6))

    def test_002(self):
        print(my_sum(0, 3))

• 如果鼠标右键不出现unittest运行选项的处理方法

TestSuite

• 把多个测试用例整合成一个测试套件
• 使用方法
  • import导入unittest
  • import导入其他的包含测试用例的py文件
    • py文件的命名规则与变量名相同
  • 实例化unittest.TestSuite类的对象
  • 调用对象的addTest方法
    • addTest(py文件名.类名(“方法名”))

import unittest
import testcase_01

suite = unittest.TestSuite()
suite.addTest(testcase_01.my_test("test_001"))
suite.addTest(testcase_01.my_test("test_002"))
# 只是把测试用例添加到了测试套件中,并不是执行测试用例

• 用unittest.makeSuite一次导入一个类中的所有测试方法

import unittest
import testcase_01

suite = unittest.TestSuite()
# suite.addTest(testcase_01.my_test("test_001"))
# suite.addTest(testcase_01.my_test("test_002"))
# 只是把测试用例添加到了测试套件中,并不是执行测试用例
suite.addTest(unittest.makeSuite(testcase_01.my_test))

TextTestRunner

• 作用,执行在suite中的测试用例
• 使用方法
  • 先实例化TextTestRunner的对象
  • 调用对象的run方法
    • 只要把suite做为参数,放入到run方法里面

import unittest
import testcase_01

suite = unittest.TestSuite()
# suite.addTest(testcase_01.my_test("test_001"))
# suite.addTest(testcase_01.my_test("test_002"))
# 只是把测试用例添加到了测试套件中,并不是执行测试用例
suite.addTest(unittest.makeSuite(testcase_01.my_test))

runner = unittest.TextTestRunner() # 实例化TextTestRunner的对象
runner.run(suite)  # 调用对象的run方法

TestLoader

• 可以从指定目录查找指定py文件中的所有测试用例,自动加载到TestSuite中

import unittest
# 用TestLoader对象的discover方法来自动查找py,自动加载py文件中的方法
# 第一个参数是从哪里找py文件,"."从当前目录开始查找py文件
# 第二个参数是指定py文件的文件名,可以用通配符
suite = unittest.TestLoader().discover(".", "my*.py")
runner = unittest.TextTestRunner()
runner.run(suite)

TestSuite和TestLoader的使用区别

• 当只是要执行py文件中多个测试用例中的几个,而不是全部执行那么适合用TestSuite的addTest加载指定的测试用例
• 当要执行所有的py文件中的所有的测试用例,那么适合使用TestLoader

小结

• 所有的TestCase最终都是用TextTestRunner来执行的
• TextTestRunner执行的是TestSuite
• 一个TestSuite中可以有多个TestCase

Fixture

• 可以在测试用例执行执行之前自动调用指定的函数,在测试用例执行之后自动调用指定的函数
• 控制级别
  • 方法级
    • 每个方法执行前和执行后都自动调用函数
  • 类级
    • 不管类中有多少方法,一个类执行前后都自动调用函数
  • 模块级
    • 不管一个模块(一个模块就是一个py文件)中有多少类,模块执行前后自动调用函数

方法级

• 在TestCase,也就是测试用例所在的class中定义方法
• def setUp(self) 当测试用例执行前,自动被调用
• def tearDown(self) 当测试用例执行后,自动被调用
• 如果一个TestCase中有多个测试用例,那么setUp和tearDown就会被自动调用多次

mytest.py内容修改如下:

import unittest

def my_sum(a, b):
    return a + b

class my_test(unittest.TestCase):
    def setUp(self):
        print("setup被自动调用了")
    def tearDown(self):
        print("teardown被自动调用了")

    def test_001(self):
        print(my_sum(5, 6))

    def test_002(self):
        print(my_sum(0, 3))

类级

• 不管类中有多少方法,一个类开始的时候自动调用函数,结束的之后自动调用函数
• 类级的fixture一定要是类方法
• @classmethod def setUpClass(cls)类开始时自动调用的方法
• @clasmethod def tearDownClass(cls)类结束的时候自动调用的方法

mytest.py修改如下:

import unittest

def my_sum(a, b):
    return a + b

class my_test(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("setupclass自动调用了")
    @classmethod
    def tearDownClass(cls):
        print("teardownclass自动调用了")
    def setUp(self):
        print(