Pytest学习-如何在用例代码中调用fixtrue时传入参数

前言

在使用Pytest的时候，为了提高代码的复用性，我们一般会把一些常用操作，比如把登录方法写在 conftest.py 中，然后在不同测试用例中，调用这个登录方法。

但是在测试用例中，我们可能需要传入不同的账号密码·，然后通过调用fixtrue中的登录函数对该账户进行登录，这里就涉及到调用 fixture 时参数传递的问题了，今天我们就来学习下。

使用 pytest.mark.parametrize() 对函数传参

在Pytest中，我们一般使用其装饰器 @pytest.mark.parametrize() 来实现参数化。

下面是一个简单的示例：

import pytest


data = [
    ["user1", "password1"],
    ["wintest", "wintest"],
    ["user2", "password2"]
]


def login_user(username, password):
    print("登录用户：{}".format(username))
    print("登录密码：{}".format(password))
    return True if (username == "wintest" and password == "wintest") else False


@pytest.mark.parametrize("username, password", data)
def test_login_user(username, password):
    res = login_user(username, password)
    assert res

我们在 conftest.py 中把用例开始和结束给打印出来，以便看起来更直观。

@pytest.fixture(autouse=True)
def start_end():
    print("------------------------Start---------------------------")
    yield
    print("------------------------End---------------------------")

把以上用例运行后，得到结果如下：

------------------------Start---------------------------
登录用户：user1
登录密码：password1
F------------------------End---------------------------
------------------------Start---------------------------
登录用户：wintest
登录密码：wintest
.------------------------End---------------------------
------------------------Start---------------------------
登录用户：user2
登录密码：password2
F------------------------End---------------------------

============================ FAILURES =============================
________________ test_login_user[user1-password1] _________________

username = 'user1', password = 'password1'

    @pytest.mark.parametrize("username, password", data)
    def test_login_user(username, password):
        res = login_user(username, password)
>       assert res
E       assert False

test_04.py:20: AssertionError
________________ test_login_user[user2-password2] _________________

username = 'user2', password = 'password2'

    @pytest.mark.parametrize("username, password", data)
    def test_login_user(username, password):
        res = login_user(username, password)
>       assert res
E       assert False

test_04.py:20: AssertionError
2 failed, 1 passed in 0.13 seconds

request接收参数

我们通过查看fixture的源码，可以知道 fixture 下可以有以下5个参数，参数默认值如下：

def fixture(scope="function", params=None, autouse=False, ids=None, name=None):

在这些参数中，有一个参数 params ，它表示当前参数列表，而我们想要拿到当前的参数，可以通过 request.param 来获取。

    :arg params: an optional list of parameters which will cause multiple
                invocations of the fixture function and all of the tests
                using it.
                The current parameter is available in ``request.param``.

我们把上面函数传参的用例改一下，然后对 request.param 进行简单的示例：

import pytest


def test_login_user(login_user):
    res = login_user
    assert res

修改 conftest.py 如下：

import pytest


data = [
    ["user1", "password1"],
    ["wintest", "wintest"],
    ["user2", "password2"]
]


@pytest.fixture(scope="function", params=data, autouse=False)
def login_user(request):
    username = request.param[0]
    password = request.param[1]
    print("登录用户：{}".format(username))
    print("登录密码：{}".format(password))
    return True if (username == "wintest" and password == "wintest") else False


@pytest.fixture(scope="function", autouse=True)
def start_end():
    print("------------------------Start---------------------------")
    yield
    print("------------------------End---------------------------")

把以上用例运行后，得到结果如下：

------------------------Start---------------------------
登录用户：user1
登录密码：password1
F------------------------End---------------------------
------------------------Start---------------------------
登录用户：wintest
登录密码：wintest
.------------------------End---------------------------
------------------------Start---------------------------
登录用户：user2
登录密码：password2
F------------------------End---------------------------

============================ FAILURES =============================
__________________ test_login_user[login_user0] ___________________

login_user = False

    def test_login_user(login_user):
        res = login_user
>       assert res
E       assert False

test_04.py:6: AssertionError
__________________ test_login_user[login_user2] ___________________

login_user = False

    def test_login_user(login_user):
        res = login_user
>       assert res
E       assert False

test_04.py:6: AssertionError
2 failed, 1 passed in 0.17 seconds

调用fixtrue时结合request传参

在上面的代码中，我们是把含有用户账号密码的data数据直接放在 conftest.py 中，如果我需要在测试用例中调用 fixtrue 时才传入账号密码，那应该要怎么做呢？

可以在测试用例中通过 @pytest.mark.parametrize() ，把调用的fixtrue当作一个函数来进行参数传递，最后再使用request接收参数。

我们继续对上面的代码进行修改，先修改测试用例部分代码：

import pytest


data = [
    ["user1", "password1"],
    ["wintest", "wintest"],
    ["user2", "password2"]
]


# indirect=True的作用是为了把 login_user 当作一个fixture函数进行调用和传递
@pytest.mark.parametrize("login_user", data, indirect=True)
def test_login_user(login_user):
    res = login_user
    assert res

修改 conftest.py 如下：

import pytest


@pytest.fixture(scope="function", autouse=False)
def login_user(request):
    username = request.param[0]
    password = request.param[1]
    print("登录用户：{}".format(username))
    print("登录密码：{}".format(password))
    return True if (username == "wintest" and password == "wintest") else False


@pytest.fixture(scope="function", autouse=True)
def start_end():
    print("------------------------Start---------------------------")
    yield
    print("------------------------End---------------------------")

把以上用例运行后，得到结果如下：

------------------------Start---------------------------
登录用户：user1
登录密码：password1
F------------------------End---------------------------
------------------------Start---------------------------
登录用户：wintest
登录密码：wintest
.------------------------End---------------------------
------------------------Start---------------------------
登录用户：user2
登录密码：password2
F------------------------End---------------------------

============================ FAILURES =============================
__________________ test_login_user[login_user0] ___________________

login_user = False

    @pytest.mark.parametrize("login_user", data, indirect=True)
    def test_login_user(login_user):
        res = login_user
>       assert res
E       assert False

test_04.py:15: AssertionError
__________________ test_login_user[login_user2] ___________________

login_user = False

    @pytest.mark.parametrize("login_user", data, indirect=True)
    def test_login_user(login_user):
        res = login_user
>       assert res
E       assert False

test_04.py:15: AssertionError
2 failed, 1 passed in 0.10 seconds