pytest教程-7-用例前后置方法

上一小节，我们学习了pytest跳过测试用例的方法，本小节我们讲解一下pytest用例的前后置方法。

在unittest中就有前置setup和后置teardown来处理测试用例执行前的准备工作（浏览器驱动实例化，数据库连接等）以及执行后的处理工作（清理数据，关闭浏览器驱动，关闭数据库连接等），那么pytest同样也提供了前置后置的方法来满足这个需求。

pytest提供了以下5个前置后置方法：

setup_module、teardown_module（模块级别）:整个py文件，测试用例执行前后只执行一次
setup_class、teardown_class:（类级别）：整个测试类中，测试用例执行前后只执行一次
setup、teardown（函数级别）:每条用例执行前后都会执行一次，既可以作用于类中测试函数，也可以作用于类外测试函数
setup_method、teardown_method（方法级别）:每条用例执行前后都会执行一次，只作用于类中测试函数
setup_function、teardown_function（函数级别）:每条用例执行前后都会执行一次，只作用于类外测试函数使用

1、setup_module、teardown_module（模块级别）示例

#test_demo.py

import pytest


class Test_04:
    def test_01(self):
        print('用例01执行')

    def test_02(self):
        print('用例02执行')


def test_03():
    print('类外用例03执行')


def setup_module():
    print('\n我是setup_module前置执行\n')


def teardown_module():
    print('\n我是teardown_module后置执行')


def test_04():
    print('类外用例04执行')


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

运行结果如下：整个py文件，测试用例执行前后只执行了一次

Launching pytest with arguments D:\PycharmProjects\Source_Code\pytest_demo\test_demo.py --no-header --no-summary -q in D:\PycharmProjects\Source_Code\pytest_demo

============================= test session starts =============================
collecting ... collected 4 items

test_demo.py::Test_04::test_01 
我是setup_module前置执行

PASSED                                    [ 25%]用例01执行

test_demo.py::Test_04::test_02 PASSED                                    [ 50%]用例02执行

test_demo.py::test_03 PASSED                                             [ 75%]类外用例03执行

test_demo.py::test_04 PASSED                                             [100%]类外用例04执行

我是teardown_module后置执行


============================== 4 passed in 0.03s ==============================

2、setup_class、teardown_class:（类级别）示例

import pytest


class Test_04:
    def test_01(self):
        print('用例01执行')

    def setup_class(self):
        print('\n我是setup_class前置执行\n')

    def teardown_class(self):
        print('\n我是teardown_class后置执行')

    def test_02(self):
        print('用例02执行')


def test_03():
    print('类外用例03执行')


def test_04():
    print('类外用例04执行')


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

运行结果如下：整个测试类中，测试用例执行前后只执行了一次

============================= test session starts =============================
collecting ... collected 4 items

test_demo.py::Test_04::test_01 
我是setup_class前置执行

PASSED                                    [ 25%]用例01执行

test_demo.py::Test_04::test_02 PASSED                                    [ 50%]用例02执行

我是teardown_class后置执行

test_demo.py::test_03 PASSED                                             [ 75%]类外用例03执行

test_demo.py::test_04 PASSED                                             [100%]类外用例04执行


============================== 4 passed in 0.02s ==============================

3.1、setup、teardown（函数级别）作用于类中测试函数示例

import pytest


class Test_04:
    def setup(self):
        print('\nsetup前置执行\n')

    def teardown(self):
        print('\nteardown后置执行\n')

    def test_01(self):
        print('用例01执行')

    def test_02(self):
        print('用例02执行')


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

运行结果如下：每条用例执行前后都会执行一次，既可以作用于类中测试函数

============================= test session starts =============================
collecting ... collected 2 items

test_demo.py::Test_04::test_01 
setup前置执行

PASSED                                    [ 50%]用例01执行

teardown后置执行


test_demo.py::Test_04::test_02 
setup前置执行

PASSED                                    [100%]用例02执行

teardown后置执行



======================== 2 passed, 4 warnings in 0.03s ========================

3.2、setup、teardown（函数级别）作用于类外测试函数示例

import pytest


def setup():
    print('\n类外setup前置执行\n')


def teardown():
    print('\n类外teardown后置执行\n')


def test_03():
    print('类外用例03执行')


def test_04():
    print('类外用例04执行')


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

运行结果如下：每条用例执行前后都会执行一次，也可以作用于类外测试函数


test_demo.py::test_03 
类外setup前置执行

PASSED                                             [ 50%]类外用例03执行

类外teardown后置执行


test_demo.py::test_04 
类外setup前置执行

PASSED                                             [100%]类外用例04执行

类外teardown后置执行



============================== 2 passed in 0.03s ==============================

4、setup_method、teardown_method（方法级别）示例

import pytest


class Test_04:
    def setup_method(self):
        print('\nsetup_method前置执行\n')

    def teardown_method(self):
        print('\nteardown_method后置执行\n')

    def test_01(self):
        print('用例01执行')

    def test_02(self):
        print('用例02执行')


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

运行结果如下：每条用例执行前后都会执行一次，只作用于类中测试函数

============================= test session starts =============================
collecting ... collected 2 items

test_demo.py::Test_04::test_01 
setup_method前置执行

PASSED                                    [ 50%]用例01执行

teardown_method后置执行


test_demo.py::Test_04::test_02 
setup_method前置执行

PASSED                                    [100%]用例02执行

teardown_method后置执行



============================== 2 passed in 0.02s ==============================

5、setup_function、teardown_function（函数级别）示例

import pytest

class Test_04:
    def test_01(self):
        print('用例01执行')
    def test_02(self):
        print('用例02执行')

def setup_function():
    print('\nsetup_function前置执行\n')
def teardown_function():
    print('\nteardown_function后置执行\n')

def test_03():
    print('类外用例03执行')
def test_04():
    print('类外用例04执行')

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

运行结果如下：每条用例执行前后都会执行一次，只作用于类外测试函数使用

============================= test session starts =============================
collecting ... collected 4 items

test_demo.py::Test_04::test_01 PASSED                                    [ 25%]用例01执行

test_demo.py::Test_04::test_02 PASSED                                    [ 50%]用例02执行

test_demo.py::test_03 
setup_function前置执行

PASSED                                             [ 75%]类外用例03执行

teardown_function后置执行


test_demo.py::test_04 
setup_function前置执行

PASSED                                             [100%]类外用例04执行

teardown_function后置执行



============================== 4 passed in 0.03s ==============================

前、后置执行顺序

1、测试类中前、后置执行顺序示例

import pytest

class Test_04:
    def setup(self):
        print('setup前置执行')
    def teardown(self):
        print('teardown后置执行')
    def setup_class(self):
        print('\nsetup_class前置执行\n')
    def teardown_class(self):
        print('\nteardown_class后置执行\n')
    def setup_method(self):
        print('setup_method前置执行')
    def teardown_method(self):
        print('teardown_method后置执行')

    def test_01(self):
        print('用例01执行')
    def test_02(self):
        print('用例02执行')


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

运行结果如下：前置执行顺序：setup_class > setup_method > setup （后置执行顺序则相反）

============================= test session starts =============================
collecting ... collected 2 items

test_demo.py::Test_04::test_01 
setup_class前置执行

setup_method前置执行
PASSED                                    [ 50%]用例01执行
teardown_method后置执行

test_demo.py::Test_04::test_02 setup_method前置执行
PASSED                                    [100%]用例02执行
teardown_method后置执行

teardown_class后置执行



============================== 2 passed in 0.02s ==============================

2、测试类外前、后置执行顺序示例

import pytest


def setup():
    print('类外setup前置执行')


def teardown():
    print('类外teardown后置执行')


def setup_function():
    print('\nsetup_function前置执行\n')


def teardown_function():
    print('\nteardown_function后置执行\n')


def setup_module():
    print('\nsetup_module前置执行\n')


def teardown_module():
    print('teardown_module后置执行')


def test_03():
    print('类外用例03执行')


def test_04():
    print('类外用例04执行')


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

运行结果如下：前置执行顺序：setup_module > setup_function > setup （后置执行顺序则相反）

test_demo.py::test_03 
setup_module前置执行


setup_function前置执行

类外setup前置执行
PASSED                                             [ 50%]类外用例03执行
类外teardown后置执行

teardown_function后置执行


test_demo.py::test_04 
setup_function前置执行

类外setup前置执行
PASSED                                             [100%]类外用例04执行
类外teardown后置执行

teardown_function后置执行

teardown_module后置执行


============================== 2 passed in 0.02s ==============================

总结：

1、测试类中：setup_class、teardown_class

setup_method、teardown_method

测试类外：setup_function、teardown_function

setup_module、teardown_module

测试类中、类外均可：setup、teardown

2、执行顺序：

类中前置执行顺序：setup_class > setup_method > setup （后置执行顺序则相反）

类外前置执行顺序：setup_module > setup_function > setup （后置执行顺序则相反）

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走，希望可以帮助到大家！