pytest进阶使用【fixture(一)fixture与setup/teardown区别】
fixture翻译为装置。
我觉得名字是很贴合功能的,可以自由给函数装置上自己想要的功能。
当在说pytest比unitest灵活时,fixture肯定是其中的一个理由。
测试数据的准备和执行以后的数据清理。
很多人可能会第一时间想到的是setup/teardown,但是fixture也能实现同样的效果,并且在某些场景下能做到setup做不到的事情。
比如setup虽然说是支持函数级,但是你是没办法指定某个用例执行的时候才去执行setup或者teardown。
只能说,要么都要、要么都不要。
但是fixture可以。
比如有三个用例,但是我想在用例1执行后轮到用例2的时候再去获取测试数据,你会发现setup/teardown使用。
要么用函数级,每个用例执行前都获取一遍数据。
要么用类级,在用例1执行前就去获取数据。
fixture则是装饰到用例2上即可。
具体代码区别如下。
unitest 函数级
class TestApi(unittest.TestCase):
def setUp(self):
print("开始执行")
def test_01(self):
print(1)
def test_02(self):
print(2)
def test_03(self):
print(3)
执行结果
开始执行
1
结束执行
开始执行
2
结束执行
开始执行
3
结束执行
unitest 类级
class TestApi(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("开始执行")
@classmethod
def tearDownClass(cls):
print("结束执行")
def test_01(self):
print(1)
def test_02(self):
print(2)
def test_03(self):
print(3)
执行结果
开始执行
1
2
3
结束执行
pytest fixture
class TestApi():
@pytest.fixture(scope='function')
def setup_function(self):
print("开始执行")
yield
print("结束执行")
def test_01(self):
print(1)
def test_02(self,setup_function):
print(2)
def test_03(self):
print(3)
执行结果
1
.开始执行
2
.结束执行
3
可以看到,开始执行和结束执行,是只单独在用例2前后执行。
至于为什么一个函数能在分两次执行,原因就是yield了,如果是函数级的,那么在用例执行完成后,就会去执行yield后面的代码,这就完成了setup/teardown的作用。
那这样看,fixture确实要比setup来的好用多。
但是,fixture最终是要装饰到用例上才会生效,也就是说如果你十个用例都要执行一次,那每个用例都得装饰上。。。
又或者你并不是想给某个用例执行,而是全模块、整个类。
灵活是灵活了,但是也有自身的局限性。
所以pytest也并没有舍去setup/teardown,两者都能共用。
下篇文章谈谈fixture的函数级、类级、模块级又和setup/teardown有什么不一样。