Pytest学习笔记（3）-Setup和Teardown

文章目录

setup和teardown
- 前言
- module（模块）级别
- function（函数）级别
- class（类）级别
- method（方法）级别
- 混合使用
- 总结

setup和teardown

前言

在做自动化的时候，常常有这样的诉求：

用例执行前，需要重新启动一次浏览器
用例执行结束时，需要清除测试数据

在unittest中，我们可以使用 setUp() 和 tearDown() 两个方法来实现以上需求，其中 setUp() 方法用于初始化测试固件；而 tearDown() 方法用于销毁测试固件。程序会在运行每个测试用例（以 test_ 开头的方法）之前自动执行 setUp() 方法来初始化测试固件，井在每个测试用例（以 test_ 开头的方法）运行完成之后自动执行 tearDown() 方法来销毁测试固件。

那么如何实现只启动一次浏览器，所有测试用例执行结束后再清除数据这样的需求呢？

unittest提供了setUpClass()和tearDownClass()两个方法，配合@classmethod装饰器使用即可

作为比unittest更强大的框架，pytest自然也有类似的方法

pytest的setup/teardown方法包括：

模块级别(setup_module/teardown_module)
函数级别(setup_function/teardown_function)
类级别(setup_class/ teardown_class)
方法级别(setup_method/teardown_methond或者setup/teardown)

module（模块）级别

模块中的第一个用例开始前执行一次setup_module方法，模块中的最后一个测试用例结束后执行一次teardown_module方法

import pytestdef setup_module():print("执行setup_module")def teardown_module():print("执行teardown_module")class TestDemo(object):def test_case1(self):print("执行测试用例1")assert 1 + 1 == 2def test_case2(self):print("执行测试用例2")assert 1 + 3 == 4def test_case3(self):print("执行测试用例3")assert 1 + 5 == 6

运行结果如下：

function（函数）级别

在每个测试函数前运行一次setup_function方法，在每个测试函数结束后运行一次teardown_function方法，只对函数用例生效，不在类中

import pytestdef setup_function():print("执行setup_function")def teardown_function():print("执行teardown_function")def test_case1():print("执行测试用例1")assert 1 + 1 == 2def test_case2():print("执行测试用例2")assert 1 + 3 == 4def test_case3():print("执行测试用例3")assert 1 + 5 == 6

运行结果如下：

class（类）级别

setup_class/teardown_class 对类有效，位于类中，在执行测试类之前和之后各调用一次

import pytestclass TestDemo(object):def setup_class(self):print("执行setup_class")def teardown_class(self):print("执行teardown_class")def test_case1(self):print("执行测试用例1")assert 1 + 1 == 2def test_case2(self):print("执行测试用例2")assert 1 + 3 == 4def test_case3(self):print("执行测试用例3")assert 1 + 5 == 6

运行结果如下：

method（方法）级别

setup_method/teardown_method和setup/teardown，在测试类中每个测试方法前后调用一次。这两个方法效果是一样的

import pytestclass TestDemo(object):def setup_method(self):print("执行setup_method")def teardown_method(self):print("执行teardown_method")def test_case1(self):print("执行测试用例1")assert 1 + 1 == 2def test_case2(self):print("执行测试用例2")assert 1 + 3 == 4def test_case3(self):print("执行测试用3")assert 1 + 5 == 6

运行结果如下：

混合使用

import pytestdef setup_module():print("模块开始时，执行setup_module")def teardown_module():print("模块结束时，执行teardown_module")def setup_function():print("函数用例开始时，执行setup_function")def teardown_function():print("函数用例结束时，执行teardown_function")def test_a():print("执行测试函数a")def test_b():print("执行测试函数b")class TestDemo(object):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 test_case1(self):print("执行测试用例1")assert 1 + 1 == 2def test_case2(self):print("执行测试用例2")assert 1 + 3 == 4def test_case3(self):print("执行测试用例3")assert 1 + 5 == 6

运行结果如下：

总结

模块级（setup_module/teardown_module）开始于模块始末，全局的
函数级（setup_function/teardown_function）只对函数用例生效（不在类中）
类级（setup_class/teardown_class）只在类中前后运行一次(在类中)
方法级（setup_method/teardown_method或setup/teardown）开始于方法始末（在类中）