软件测试 | Pytest测试框架之fixture进阶

pytest 测试框架之 fixture 进阶

简介

fixture 即测试用例执行的环境准备和消理，fixture 主要的目的是为了提供一种可靠和可重复性的手段去 运行那些最基本的测试内容。比如测试网站的功能时，每个测试用例都要登录和登出，利用 fixture 就可 以只做一次，否则每个测试用例都要做这两步。

fixture 类似于 setup/teardown 机制，但 fixture 更灵活的完成测试的初始化以及资源清理工作。可以使 用它将你需要执行 setup 的测试用例传入 fixture 函数名，不需要的测试用例可以不用传递，不传递 fixture 名的测试用例将不会执行 fixture 函数。它允许我们将复杂的测试需求归纳成更简单、更有组织的 测试函数。 @pytest.fixture 与 @pytest.mark.parametrize 结合实现参数化

如果测试数据需要在 fixture 方法中使用，同时也需要在测试用例中使用，可以在使用 parametrize 的时 候添加一个参数 indirect=True ，pytest 可以实现将参数传入到 fixture 方法中，也可以在当前的测试 用例中使用。

当 indirect 参数设置这个参数为 True 时，pytest 会把 argnames 当作函数去执行，将 argvalues 作 为参数传入到 argnames 这个函数里。创建“test_param.py”文件，代码如下：

# 方法名作为参数
test_user_data = ['Tome', 'Jerry']
@pytest.fixture(scope="module")
def login_r(request):
 # 通过request.param获取参数
 user = request.param
 print(f"\n 登录用户：{user}")
 return user
@pytest.mark.parametrize("login_r", test_user_data,indirect=True)
def test_login(login_r):
 a = login_r
 print(f"测试用例中login的返回值; {a}")
 assert a != ""

运行结果：

test_mark_paramize.py::test_login[Tome]
test_mark_paramize.py::test_login[Jerry]
============================== 2 passed in\
0.02s ===============================
Process finished with exit code 0
登录用户：Tome PASSED [ 50%]测试用例中login的返回值; Tome
登录用户：Jerry PASSED [100%]测试用例中login的返回值; Jerry

上面的结果可以看出，当 indirect=True 时，会将 login_r 作为参数，test_user_data 被当作参数 传入到 login_r 方法中，生成多条测试用例。通过 return 将结果返回，当调用 login_r 可以获取 到 login_r 这个方法的返回数据。

fixture 间相互调用

fixture 可以作为参数传递给其它的 fixture 函数。

工作中可能不同的功能，依赖的数据，环境不一样，需要定义多个功能函数。

比如：登录 fixture 函数中需要信赖连接数据库的 fixture 函数。可以将这些功能函数都定义为 fixture， 然后通过传递 fixture 函数，实现功能之间的数据交换。

# contents of test_append.py
import pytest
@pytest.fixture
def connectDB():
 return "name,school"
@pytest.fixture
def login(connectDB):
 return [connectDB]
def test_string(login):
 pass

测试用例/fixture 可以请求多个 fixture

下面是一个示例：

# contents of test_append.py
import pytest
# Arrange
@pytest.fixture
def first_entry():
 return "a"
# Arrange
@pytest.fixture
def second_entry():
 return 2
# Arrange
@pytest.fixture
def order(first_entry, second_entry):
 return [first_entry, second_entry]
# Arrange
@pytest.fixture
def expected_list():
 return ["a", 2, 3.0]
def test_string(order, expected_list):
 # Act
 order.append(3.0)
 # Assert
 assert order == expected_list

Fixture 终结器(Teardown/Cleanup)

添加终结器

但如果 yield 之前的内容发生异常，将不会执行 yield 后面的内容。

对于这种情况，还可以直接向测试的 request 对象添加 “finalizer” 终结器函数实现这个功能（与 yield fixtures 类似的结果，但能力更强）。

举个应用场景：

比如我们在读取文件的时候，需要打开文件，获取文件流对象，定义一个关闭文件的方法 close() ， 在终结器中实现关闭操作。最后即使后面的代码有异常，也会执行终结器里面的操作。

@pytest.fixture(scope='module', autouse=True)
def operate_file(request):
 print("打开文件")
 fs = open('./datas/data.yaml', mode="r")
 def close():
 print("关闭文件")
 fs.close()
 request.addfinalizer(close)
 datas = yaml.safe_load(fs)
 raise ValueError
 print(datas)