【测试框架篇】单元测试框架pytest（5）：setup和teardown的详细使用

一、前言

用过unittest的童鞋们应用都知道，有两个前置方法，两个后置方法如下：

• setup()
• setupClass()
• teardown()
• teardownClass()

Pytest也提供了类似于setup、teardown的方法，并且还超过四个，一共有十种分级别的方法：

• 模块级别：setup_module、teardown_module
• 函数级别：setup_function、teardown_function，不在类中的方法
• 类级别：setup_class、teardown_class
• 方法级别：setup_method、teardown_method
• 方法细化级别：setup、teardown

二、代码

用过unittest的童鞋，对这个前置、后置方法应该不陌生了，我们直接来看代码和运行结果

#!/usr/bin/env python
# -*- coding: utf-8 -*-


import pytest


def setup_module():
    print("=====整个.py模块开始前只执行一次:打开浏览器=====")


def teardown_module():
    print("=====整个.py模块结束后只执行一次:关闭浏览器=====")


def setup_function():
    print("===每个函数级别用例开始前都执行setup_function===")


def teardown_function():
    print("===每个函数级别用例结束后都执行teardown_function====")


def test_one():
    print("one")


def test_two():
    print("two")


class TestCase():
    def setup_class(self):
        print("====整个测试类开始前只执行一次setup_class====")

    def teardown_class(self):
        print("====整个测试类结束后只执行一次teardown_class====")

    def setup_method(self):
        print("==类里面每个用例执行前都会执行setup_method==")

    def teardown_method(self):
        print("==类里面每个用例结束后都会执行teardown_method==")

    def setup(self):
        print("=类里面每个用例执行前都会执行setup=")

    def teardown(self):
        print("=类里面每个用例结束后都会执行teardown=")

    def test_three(self):
        print("three")

    def test_four(self):
        print("four")


if __name__ == '__main__':
    pytest.main(["-q", "-s", "-ra", "setup_teardown.py"])

执行结果

注意，从执行结果我们可以看到：

• 模块级别：setup_module、teardown_module，只执行一次，在整个.py模块开始前和结束后都需要；
• 函数级别：setup_function、teardown_function，不在类中的方法，主要取决于有几个函数级别用例，有几个就执行几次，在每个函数级别用例开始前和结束后都需要；
• 类级别：setup_class、teardown_class，整个测试类开始前和结束后执行一次；
• 方法级别：setup_method、teardown_method，取决于类里面有多少测试用例，类里面每个测试用例开始前和结束后都需要执行一次；
• 方法细化级别：setup、teardown，跟前面方法级别使用类似，不过细化级别需要先执行；