Python pytest进阶之xunit fixture详解
本文深入解析pytest框架中的xunit fixture机制，详细对比function、method、class及module四个级别的setup与teardown执行逻辑。通过完整代码示例与输出结果，阐明各层级初始化和销毁操作的触发时机及注意事项，帮助开发者掌握pytest测试环境管理的最佳实践。
前言
前言
本文将继续探讨pytest框架中与unittest框架相似的fixture用法。熟悉unittest的开发者应当了解，在初始化测试环境和清理工作资源时，unittest通常采用setUp和tearDown方法。而在pytest框架中，同样提供了功能类似的方法机制，接下来我们将对这些方法进行具体说明。
首先引用官方文档的一段说明作为背景参考。
1. 各个级别的setup和teardown方法均可被多次复用。
2. 若对应的初始化函数执行失败或被跳过，则不会执行相应的teardown方法。
3. 在pytest 4.2版本之前，xunit fixture并不遵循标准fixture的作用域规则，因此可以在一个session级别且参数auto=True的fixture之前执行setup_method方法。
然而，截至目前，所有的xunit fixture均已遵循fixture的执行规则。
function级别
function级别
实例
def setup_function(function):
print('\n--------------------')
print('函数执行前所做的操作')
print('\n--------------------')
def teardown_function(function):
print('\n--------------------')
print('函数执行后所做的操作')
print('\n--------------------')
def test_function_1():
print('\n测试函数1')
def test_function_2():
print('\n测试函数2')
if __name__ == '__main__':
import pytest
pytest.main(['-sq', 'functionLevel.py'])输出结果
functionLevel.py
--------------------
函数执行前所做的操作
--------------------
测试函数1
--------------------
函数执行后所做的操作
--------------------
--------------------
函数执行前所做的操作
--------------------
测试函数2
--------------------
函数执行后所做的操作
--------------------
[100%]
========================== 2 passed in 0.03 seconds ===========================说明
根据上述输出结果可以总结出：setup_function会在每个测试函数执行前运行初始化操作；而teardown_function则会在每个测试函数执行完毕后执行销毁工作。
method级别
method级别
实例
class TestMethod(object):
def setup_method(self, method):
print('\n--------------------')
print('方法执行前所做的操作')
print('\n--------------------')
def teardown_method(self, method):
print('\n--------------------')
print('方法执行后所做的操作')
print('\n--------------------')
def test_method_1(self):
print('\n测试方法1')
def test_method_2(self):
print('\n测试方法2')
if __name__ == '__main__':
import pytest
pytest.main(['-sq', 'methodLevel.py'])输出结果
methodLevel.py
--------------------
方法执行前所做的操作
--------------------
测试方法1
--------------------
方法执行后所做的操作
--------------------
--------------------
方法执行前所做的操作
--------------------
测试方法2
--------------------
方法执行后所做的操作
--------------------
[100%]
========================== 2 passed in 0.03 seconds ===========================说明
通过输出结果可以得出：setup_method会在每个测试方法执行前运行初始化操作；teardown_method则会在每个测试方法执行完毕后执行销毁工作。此外，方法级别的fixture专门作用于测试类中的方法。
class级别
class级别
实例
class TestClass(object):
@classmethod
def setup_class(cls):
print('\nsetup_class() for {}'.format(cls.__name__))
@classmethod
def teardown_class(cls):
print('\nteardown_class() for {}'.format(cls.__name__))
def test_1(self):
print('self.test_1()')
def test_2(self):
print('self.test_2()')
if __name__ == '__main__':
import pytest
pytest.main(['-sq', 'classLevel.py'])输出结果
classLevel.py
setup_class() for TestClass
.self.test_1()
.self.test_2()
teardown_class() for TestClass
[100%]
========================== 2 passed in 0.06 seconds ===========================说明
根据输出结果可以总结：setup_class会在测试类执行前仅运行一次初始化操作；teardown_class则会在测试类执行完毕后仅运行一次销毁工作。需要注意的是，class级别的fixture必须使用@classmethod装饰器进行修饰。
module级别
module级别
实例
def setup_module(module):
print('\nsetup_module() for {}'.format(module.__name__))
def teardown_module(module):
print('\nteardown_module() for {}'.format(module.__name__))
def test_1():
print('test_1()')
def test_2():
print('test_2()')
class TestClass(object):
def test_3(self):
print('self.test_3()')
def test_4(self):
print('self.test_4()')
if __name__ == '__main__':
import pytest
pytest.main(['-sq', 'moduleLevel.py'])输出结果
moduleLevel.py
setup_module() for moduleLevel
.test_1()
.test_2()
.self.test_3()
.self.test_4()
teardown_module() for moduleLevel
[100%]
========================== 4 passed in 0.04 seconds ===========================说明
通过输出结果可以总结：setup_module会在整个测试文件（即模块）中的测试类或测试函数、测试方法执行前，仅运行一次初始化操作；teardown_module则会在整个测试文件中的测试类或测试函数、方法执行完毕后，仅运行一次销毁工作。
以上即为xunit fixture的四个级别。在实际工作中如何灵活运用，仍需通过大量练习来深入理解，从而达到得心应手的效果。
附上官方文档供参考，虽然内容为英文，但描述非常详尽。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
