前面一篇文章,我写了如何在一个测试类中实现环境初始化,那么,如果我每执行一个测试方法(用例)就想要初始化或清除一遍,该怎么操作呢
比如,我定义一个添加商品列表的测试方法 ,定义一个修改商品的测试方法
class Test_shop:#这是一个测试类
def test_addshop(self):
print("执行商品添加测试用例")
def test_updateshop(self):
print("执行修改商品测试用例")
每一个用例之前,我都需要初始化测试环境,应该怎么做呢,这个时候要用到 pytest中的setup 方法
import pytest
class Test_shop:#这是一个测试类
def setup(self):
print("环境初始化完成")
def test_addshop(self):
print("执行商品添加测试用例")
def test_updateshop(self):
print("执行修改商品用例")
if __name__ == '__main__':
pytest.main(['test_01.py','-s'])
输出结果如下:
test_01.py 环境初始化完成
执行商品添加测试用例
.环境初始化完成
执行修改商品用例
.
============================== 2 passed in 0.03s ==============================
加上setup方法之后,每个测试用例执行之前,都会执行一边setup进行环境初始化,并且setup方法的位置,同setup_class 一样,无论放在哪都不影响执行顺序,小伙伴们可以自己去试一下
有环境初始化了,那么就肯定也有环境清除,环境清除用到的是 teardown方法,它的规则和setup一样,也是每执行完一个用例后,就会进行环境清除操作,同样的, teardown方法的位置也不会影响其执行顺序
import pytest
class Test_shop:#这是一个测试类
def test_addshop(self):
print("执行商品添加测试用例")
def test_updateshop(self):
print("执行修改商品用例")
def teardown(self):
print("环境清除完成")
if __name__ == '__main__':
pytest.main(['test_01.py','-s'])
输出结果如下:
test_01.py 执行商品添加测试用例
.环境清除完成
执行修改商品用例
.环境清除完成
============================== 2 passed in 0.04s ==============================
完整的初始化和清除代码如下:
import pytest
class Test_shop:#这是一个测试类
def setup(self):
print("环境初始化完成")
def test_addshop(self):
print("执行商品添加测试用例")
def test_updateshop(self):
print("执行修改商品用例")
def teardown(self):
print("环境清除完成")
if __name__ == '__main__':
pytest.main(['test_01.py','-s'])
输出结果:
test_01.py 环境初始化完成
执行商品添加测试用例
.环境清除完成
环境初始化完成
执行修改商品用例
.环境清除完成
============================== 2 passed in 0.03s ==============================