Pytest Fixture 参数化

简介

pytest 还支持参数化 fixture，即可以为一个 fixture 提供多个不同的值，让多个测试用例使用不同的测试数据。

这样可以有效减少重复代码，尤其适合测试多组输入输出的情况。

基本语法

params：传入一个列表，包含 Fixture 的不同参数。
request.param：可以在 Fixture 函数中通过 request.param 获取当前参数。

import pytest

@pytest.fixture(params=[param1, param2, param3, ...])
def fixture_name(request):
    return request.param

单参数

import pytest

@pytest.fixture(params=[1, 2, 3])
def number(request):
    return request.param

def test_number(number):
    assert number > 0

运行结果：

多参数

import pytest

@pytest.fixture(params=[
   ["selenium", 123],
   ["appium", 123456]
])
def login(request):
    print(f"用户名：{request.param}")
    yield request.param

def test_demo(login):
    print(f"demo case: 数据为: {login}")

运行结果：

fixture 与 parametrize 结合实现参数化

通过 parametrize 中设置 indirect=True 可以将参数传递到指定的 Fixture 中

import pytest

@pytest.fixture
def database(request):
    # database fixture 接收 request.param，返回数据库连接字符串。
    db_name = request.param
    return f"连接数据库: {db_name}"

# @pytest.mark.parametrize 使用 indirect=True 将参数传递给 database fixture
@pytest.mark.parametrize(
   "database", 
   ["test_db1", "test_db2"],
   indirect=True
)
def test_database_connection(database):
    print(f"database = {database}")
    assert "连接数据库" in database

运行结果：

./assets/PytestFixture参数化.mp4

然后的话我们再来看一下这个 fixture 和 params 的结合。那比如说现在我先来定义一个测试函数叫 test database connection。然后在这个方法里面可能我们要去这个打印一下我们的这个数据库的一个信息，那数据库的信息的话就是待会我们要传入的那个 fixture。然后的话，嗯，我们比如说我们要去断言一下某一个数据，在我们拿到的这样的一个数据里。好，那这个数据在哪里呢？我们就需要通过一个 fixture 来给它进行一个指定啊。那比如说我们来定义一个就叫做 database 这样的一个fixture 方法，那然后的话我们在这边就可以通过 @pytest点fixture() 把它定义为成是一个 fixture 函数。然后同样要把我们的这个 request 给它传进来。那么这样的话，其实我们就比如说我们的这个数据库的名称，那我们就可以用 request点param 来指定，对吧？然后用 return 把我们的这个，比如说我们指定为连接数据库，对吧？然后的话把我们的这个 db name 给它返回去啊。好，那这样的话我们就定义好了一个 fixture 的这样的一个方法。那然后的话我们在这个里面就可以直接的把我们这个 database 给它传入进来。那么传入进来之后，那我们在这边的话就打印一下 database 的这个值。这样就可以去看一下我们的这个连接数据库的这个内容，在没在我们拿到的这个数据里。那具体的数据我们从哪拿呢？就可以用我们的这个 @pytest点mark点parametrize，那这个数据的话其实就是我们要传到 fixture 中的，对吧？我们先把 parametrize 中需要的参数传入进来，比如说数据库名称叫做 test db1，那么传完两个数据之后，最主要的是把这个 indirect 把它设置为 true，那这样的话我们就可以去结合我们的这个 fixture，然后具体的数据用 params 来给它传递。下面我们来执行一下看看。哎，执行完了之后，大家可以看到我们连接数据库这块儿，是不是哎？是可以看到对应的这个数据的，对吧？

./assets/PytestFixture参数化.mp4

总结

Pytest Fixture 参数化的使用。