pytest学习和使用4-pytest和Unittest中setup、teardown等方法详解和使用（最全）

2023-02-06 09:53:45 浏览数 (2)

1 Unittest两种前置和两种后置方法

  • 使用Unittest框架结合selenium做webUI自动化测试的时候，经常会遇到什么时候打开和关闭浏览器，这个时候就使用到了Unittest两种前置和两种后置方法；
  • 那具体这四种方法是什么呢？看下表：

方法

说明

setup()

每执行一个用例之前执行一次，比如每次运行某个用例前，打开一次浏览器

teardown()

每执行一个用例之后执行一次，比如每次运行某个用例后，关闭一次浏览器

setupClass()

每执行一个用例集之前执行一次，比如每运行一个testcase前，打开一次浏览器

teardownClass()

每执行一个用例集之后执行一次，比如每运行一个testcase后，关闭一次浏览器

  • 而setupClass()和teardownClass() 方法用配合@classmethod方法使用。

1.1 Unittest：setup、teardown方法举例

  • 创建一个脚本test_unittest_setup_teardown.py，写入以下代码：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 9:25
# 文件名称：test_unittest_setup_teardown.py
# 作用：验证unittest的setup、teardown方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import unittest


class TestOne(unittest.TestCase):
    def setUp(self) -> None:
        print("每运行一个case前，打开一次浏览器")

    def tearDown(self) -> None:
        print("每运行一个case后，关闭一次浏览器")

    def test_one(self):
        print("运行第一个用例")

    def test_two(self):
        print("运行第二个用例")


if __name__ == "__main__":
    unittest.main()
  • 运行后如下：
代码语言：python代码运行次数：0复制
Ran 2 tests in 0.003s

OK
Launching unittests with arguments python -m unittest F:/pytest_study/test_case/test_b/test_unittest_setup_teardown.py in F:pytest_studytest_casetest_b


进程已结束，退出代码 0
每运行一个case前，打开一次浏览器
运行第一个用例
每运行一个case后，关闭一次浏览器
每运行一个case前，打开一次浏览器
运行第二个用例
每运行一个case后，关闭一次浏览器

1.2 Unittest：setupClass、teardownClass方法举例

  • 创建一个脚本test_unittest_setupclass_teardownclass.py，写入以下代码：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 9:50
# 文件名称：test_unittest_setupclass_teardownclass.py
# 作用：验证unittest的setupclass、teardownclass方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import unittest


class TestTwo(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        print("每运行一个用例集前，打开浏览器，即这个类只打开一次浏览器")

    @classmethod
    def tearDownClass(cls) -> None:
        print("每运行一个用例集后，关闭浏览器，即这个类只关闭一次浏览器")

    def test_one(self):
        print("运行第一个用例")

    def test_two(self):
        print("运行第二个用例")


if __name__ == "__main__":
    unittest.main()
  • 运行后如下：
代码语言：python代码运行次数：0复制
Ran 2 tests in 0.002s


OK
每运行一个用例集前，打开浏览器，即这个类只打开一次浏览器
运行第一个用例
运行第二个用例
每运行一个用例集后，关闭浏览器，即这个类只关闭一次浏览器

进程已结束，退出代码 0
  • 注意这两个方法需要使用@classmethod修饰方法，如果不加的话会报错。

2 Pytest十种前置和后置方法

  • 和unittest类似，但是方法更多，达到了十种，详细看下表：

方法

运行级别

说明

setup_module()

模块级别

整个.py模块开始前只执行一次,如打开一次浏览器

teardown_module()

模块级别

整个.py模块结束后只执行一次,如关闭一次浏览器

setup_function()

函数级别

每个函数级别用例开始前都执行,此方法不在类中

teardown_function()

函数级别

每个函数级别用例结束后都执行,此方法不在类中

setup_class()

类级别

整个测试类开始前只执行一次,和Unittest基本一样

teardown_class()

类级别

整个测试类结束后只执行一次，和Unittest基本一样

setup_method()

方法级别

类里面每个用例执行前都会执行

teardown_method()

方法级别

类里面每个用例结束后都会执行

setup()

方法细化级别

类里面每个用例执行前都会执行

teardown()

方法细化级别

类里面每个用例结束后都会执行

2.1 Pytest：setup_module、teardown_module方法举例

  • 创建test_pytest_setup_teardown_module.py，代码如下：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 10:18
# 文件名称：test_pytest_setup_teardown_module.py
# 作用：验证pytest的setup_module和teardown_module方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import pytest


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


def teardown_module():
    print("整个.py模块结束后只执行一次")


def test_one():
    print("用例1")


def test_two():
    print("用例2")


class TestOne():
	# @staticmethod
    # def setup_module():
    #     print("整个.py模块开始前只执行一次")
    #
    # @staticmethod
    # def teardown_module():
    #     print("整个.py模块结束后只执行一次")
    def test_thr(self):
        print("用例3")

    def test_fo(self):
        print("用例4")


if __name__ == "__main__":
    pytest.main()
  • 运行结果：
代码语言：python代码运行次数：0复制
(venv) F:pytest_studytest_casetest_c>pytest -s -q
整个.py模块开始前只执行一次
用例1
.用例2
.用例3
.用例4
.整个.py模块结束后只执行一次

4 passed in 0.02s
  • 把这两个方法写入类中呢，那需要使用@staticmethod方法修饰，不然语法就不对，但是写入类中的话，这两个方法应该是不会运行的。

2.2 Pytest：setup_function、teardown_function方法举例

  • 创建test_pytest_setup_teardown_function.py，代码如下：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 10:41
# 文件名称：test_pytest_setup_teardown_function.py
# 作用：验证pytest的setup_function、teardown_function方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import pytest


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


def teardown_module():
    print("整个.py模块结束后只执行一次")


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


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


def test_one():
    print("用例1")


def test_two():
    print("用例2")


class TestOne():

    def test_thr(self):
        print("用例3")

    def test_fo(self):
        print("用例4")


if __name__ == "__main__":
    pytest.main()
  • 运行如下：
代码语言：python代码运行次数：0复制
(venv) F:pytest_studytest_casetest_c>pytest -s -q test_pytest_setup_teardown_function.py
整个.py模块开始前只执行一次
每个函数级别用例开始前都执行
用例1
.每个函数级别用例结束后都执行
每个函数级别用例开始前都执行
用例2
.每个函数级别用例结束后都执行
用例3
.用例4
.整个.py模块结束后只执行一次

4 passed in 0.42s
  • 同样把这两个方法写入类中呢，那需要使用@staticmethod方法修饰，不然语法就不对，但是写入类中的话，这两个方法应该是不会运行的。

2.3 Pytest：setup_class、teardown_class方法举例

  • 创建test_setup_teardoen_class.py代码如下：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 11:28
# 文件名称：test_pytest_setup_teardoen_class.py
# 作用：xxx
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import pytest


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


def teardown_module():
    print("整个.py模块结束后只执行一次")


def test_one():
    print("用例1")


def test_two():
    print("用例2")


class TestOne():

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

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


    def test_thr(self):
        print("用例3")

    def test_fo(self):
        print("用例4")


if __name__ == "__main__":
    pytest.main()
  • 运行结果为：
代码语言：python代码运行次数：0复制
(venv) F:pytest_studytest_casetest_c>pytest -s -q test_setup_teardoen_class.py
整个.py模块开始前只执行一次
用例1
.用例2
.整个测试类开始前只执行一次
用例3
.用例4
.整个测试类结束后只执行一次
整个.py模块结束后只执行一次

4 passed in 0.02s

2.4 Pytest：setup_method、teardown_method方法举例

  • 创建test_pytest_setup_teardown_method.py,代码如下：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 12:28
# 文件名称：test_pytest_setup_teardown_method.py
# 作用：验证pytest的setup_method、teardown_method方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import pytest


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


def teardown_module():
    print("整个.py模块结束后只执行一次")


def test_one():
    print("用例1")


def test_two():
    print("用例2")


class TestOne():

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

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

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

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

    def test_thr(self):
        print("用例3")

    def test_fo(self):
        print("用例4")


if __name__ == "__main__":
    pytest.main()
  • 运行结果为：
代码语言：python代码运行次数：0复制
(venv) F:pytest_studytest_casetest_c>pytest -s -q test_pytest_setup_teardown_method.py
整个.py模块开始前只执行一次
用例1
.用例2
.整个测试类开始前只执行一次
1类里面每个用例执行前都会执行
用例3
.1类里面每个用例结束后都会执行
1类里面每个用例执行前都会执行
用例4
.1类里面每个用例结束后都会执行
整个测试类结束后只执行一次
整个.py模块结束后只执行一次

4 passed in 0.14s

2.5 Pytest：setup、teardown方法举例

  • 创建test_pytest_setup_teardown.py，代码如下：
代码语言：python代码运行次数：0复制
# -*- coding:utf-8 -*-
# 作者：NoamaNelson
# 日期：2021/9/9 12:28
# 文件名称：test_pytest_setup_teardown.py
# 作用：验证pytest的setup、teardown方法
# 联系：VX(NoamaNelson)
# 博客：https://blog.csdn.net/NoamaNelson


import pytest


def setup_module():
    print("setup_module：整个.py模块开始前只执行一次")


def teardown_module():
    print("teardown_module：整个.py模块结束后只执行一次")


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


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

def test_one():
    print("用例1")


def test_two():
    print("用例2")


class TestOne():

    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_thr(self):
        print("用例3")

    def test_fo(self):
        print("用例4")


if __name__ == "__main__":
    pytest.main()
  • 运行结果如下：
代码语言：python代码运行次数：0复制
(venv) F:pytest_studytest_casetest_c>pytest -s -q test_pytest_setup_teardown.py
setup_module：整个.py模块开始前只执行一次
setup_function：每个函数级别用例开始前都执行
用例1
.teardown_function：每个函数级别用例结束后都执行
setup_function：每个函数级别用例开始前都执行
用例2
.teardown_function：每个函数级别用例结束后都执行
setup_class：整个测试类开始前只执行一次
setup_method：类里面每个用例执行前都会执行
setup：类里面每个用例执行前都会执行
用例3
.teardown：类里面每个用例结束后都会执行
teardown_method：类里面每个用例结束后都会执行
setup_method：类里面每个用例执行前都会执行
setup：类里面每个用例执行前都会执行
用例4
.teardown：类里面每个用例结束后都会执行
teardown_method：类里面每个用例结束后都会执行
teardown_class：整个测试类结束后只执行一次
teardown_module：整个.py模块结束后只执行一次

4 passed in 0.14s
uml python 编程算法自动化测试 python pytest

0 人点赞