前言
在使用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
网友评论