Pytest框架 — 05、setup和teardown（全部前后置）

目录

• 1、前言
• 2、各级别详解
• 3、总结

1、前言

在自动化测试过程中经常需要进行初始化和后期处理等工作，如电商加购物车测试需要先登录，测试完毕需要关闭浏览器等。Pytest提供了5种类型的setup和teardown的方法，具体如下：

• 模块级别：setup_module，teardown_module
• 函数级别：setup_function，teardown_function
• 类级别：setup_class，teardown_class
• 类方法级别：setup_method，teardown_method
• 函数或类方法级别（兼容unittest）：setup，teardown

2、各级别详解

1. 模块级别：是全局的，在模块运行前执行一次setup_module()方法，在模块运行后执行一次teardown_module()方法

def setup_module():
    print("【初始化...】")


def teardown_module():
    print("【后期处理】")


class Test_1:

    def test_1_a(self):
        print("这是类方法test_1_a")

    def test_1_b(self):
        print("这是类方法test_1_b")


class Test_2:
    def test_2_a(self):
        print("这是类方法test_2_a")

    def test_2_b(self):
        print("这是类方法test_2_b")

def test_3():
    print("这是函数test_3")

"""
执行结果
setupAndteardown/test_module.py::Test_1::test_1_a 【初始化...】
这是类方法test_1_a
PASSED
setupAndteardown/test_module.py::Test_1::test_1_b 这是类方法test_1_b
PASSED
setupAndteardown/test_module.py::Test_2::test_2_a 这是类方法test_2_a
PASSED
setupAndteardown/test_module.py::Test_2::test_2_b 这是类方法test_2_b
PASSED
setupAndteardown/test_module.py::test_3 这是函数test_3
PASSED【后期处理】
"""

2. 函数级别：只对函数用例生效，在每个函数用例前后执行，不包括类中方法

def setup_function():
    print("【初始化...】")

def teardown_function():
    print("【后续处理】")

class Test_1:

    def test_1_a(self):
        print("这是类方法test_1_a")

    def test_1_b(self):
        print("这是类方法test_1_b")

def test_2():
    print("这是函数test_2")

def test_3():
    print("这是函数test_3")


"""
执行结果
setupAndteardown/test_function.py::Test_1::test_1_a 这是类方法test_1_a
PASSED
setupAndteardown/test_function.py::Test_1::test_1_b 这是类方法test_1_b
PASSED
setupAndteardown/test_function.py::test_2 【初始化...】
这是函数test_2
PASSED【后续处理】

setupAndteardown/test_function.py::test_3 【初始化...】
这是函数test_3
PASSED【后续处理】
"""

3. 类级别：只在类中运行。在类的全部方法前执行一次setup_class()，全部方法后执行一次teardown_class()

class Test_1:
    def setup_class(self):
        print("【初始化...】")

    def teardown_class(self):
        print("【后续处理】")

    def test_1_a(self):
        print("这是类中方法test_1_a")

    def test_1_b(self):
        print("这是类中方法test_1_b")

class Test_2:
    def test_2_a(self):
        print("这是类中方法test_2_a")

    def test_2_b(self):
        print("这是类中方法test_2_b")

def test_3():
    print("这是函数test_3")

"""
执行结果
setupAndteardown/test_class.py::Test_1::test_1_a 【初始化...】
这是类中方法test_1_a
PASSED
setupAndteardown/test_class.py::Test_1::test_1_b 这是类中方法test_1_b
PASSED【后续处理】

setupAndteardown/test_class.py::Test_2::test_2_a 这是类中方法test_2_a
PASSED
setupAndteardown/test_class.py::Test_2::test_2_b 这是类中方法test_2_b
PASSED
setupAndteardown/test_class.py::test_3 这是函数test_3
PASSED
"""

4. 类方法级别：只在类中运行。类中每个方法级别的测试用例之前先执行一次setup_method()，执行后再执行一次teardown_method ()

class Test_1:
    def setup_method(self):
        print("【初始化...】")

    def teardown_method(self):
        print("【后续处理】")

    def test_1_a(self):
        print("这是类方法test_1_a")

    def test_1_b(self):
        print("这是类方法test_1_b")

    def test_1_c(self):
        print("这是类方法test_1_c")

    def test_1_d(self):
        print("这是类方法test_1_d")

"""
执行结果
setupAndteardown/test_method.py::Test_1::test_1_a 【初始化...】
这是类方法test_1_a
PASSED【后续处理】

setupAndteardown/test_method.py::Test_1::test_1_b 【初始化...】
这是类方法test_1_b
PASSED【后续处理】

setupAndteardown/test_method.py::Test_1::test_1_c 【初始化...】
这是类方法test_1_c
PASSED【后续处理】

setupAndteardown/test_method.py::Test_1::test_1_d 【初始化...】
这是类方法test_1_d
PASSED【后续处理】
"""

5. 函数或类方法级别：setup()和teardown()放类里面就会对类方法前后置，放在类外面就会对函数前后置，同unittest中效果。

# 类外前后置
def setup():
    print("【函数初始化...】")


def teardown():
    print("【函数后期处理】")


class Test_1:
    # 类中前后置
    def setup(self):
        print("【类方法初始化...】")

    def teardown(self):
        print("【类方法后期处理】")

    def test_1_a(self):
        print("这是类方法test_1_a")

    def test_1_b(self):
        print("这是类方法test_1_b")

def test_2():
    print("这是函数test_2")

def test_3():
    print("这是函数test_3")

"""
执行结果
setupAndteardown/test_setup.py::Test_1::test_1_a 【类方法初始化...】
这是类方法test_1_a
PASSED【类方法后期处理】

setupAndteardown/test_setup.py::Test_1::test_1_b 【类方法初始化...】
这是类方法test_1_b
PASSED【类方法后期处理】

setupAndteardown/test_setup.py::test_2 【函数初始化...】
这是函数test_2
PASSED【函数后期处理】

setupAndteardown/test_setup.py::test_3 【函数初始化...】
这是函数test_3
PASSED【函数后期处理】
"""

3、总结

从各级别详解中的模块级别例子中可以看到，优先级最高的是模块级别，从其他例子中可以看到类级别，类方法级别和函数级别互不影响。类级别是在类中所有方法前后置，类方法级别是类中所有测试方法前后置，函数级别是所有测试函数前后置。setup和teardown在类中等同于setup_method和teardown_method，在类外等同于setup_function和teardown_function，与unittest中的setup和teardown作用及用法均相同