pytest文档67-在 pytest.mark.parametrize 中使用 fixture

前言

测试用例参数化的时候,使用 pytest.mark.parametrize 参数化传测试数据,如果我们想引用前面 不同fixture 返回的数据当测试用例的入参,目前没好的解决办法。
可以用fixture 参数化 prams 来间接解决这个问题

使用案例

我们需要在测试用例里面参数化,参数化的数据来源于前面不同fixture的返回值,示例

import pytest
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/


@pytest.fixture
def a():
    return 'a'


@pytest.fixture
def b():
    return 'b'


@pytest.mark.parametrize('arg', [a, b])
def test_foo(arg):
    assert len(arg) == 1

这时候运行会报错

..\test_xx.py F
arg = <function a at 0x000002502E933D08>

    @pytest.mark.parametrize('arg', [a, b])
    def test_foo(arg):
>       assert len(arg) == 1
E       TypeError: object of type 'function' has no len()

D:\test_xx.py:13: TypeError
F
arg = <function b at 0x000002502E933E18>

    @pytest.mark.parametrize('arg', [a, b])
    def test_foo(arg):
>       assert len(arg) == 1
E       TypeError: object of type 'function' has no len()

关于此问题的讨论可以看github 上的issue Using fixtures in pytest.mark.parametrize #349

使用 fixture 参数化

可以使用 fixture 的参数化来解决上面的问题,通过 request.getfixturevalue("fixture name") 方法来获取fixture的返回值
有些文档看到的是 request.getfuncargvalue 那是早期的版本,目前新版本改名换成了 request.getfixturevalue
getfixturevalue 的作用是获取 fixture 的返回值

import pytest
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/


@pytest.fixture
def a():
    return 'a'


@pytest.fixture
def b():
    return 'b'


@pytest.fixture(params=['a', 'b'])
def arg(request):
    return request.getfixturevalue(request.param)


def test_foo(arg):
    assert len(arg) == 1

这样运行就不会有问题了

实例场景

web自动化的时候,想在 chrome 和 firefox 浏览器上测试同一功能的测试用例

import pytest
from selenium import webdriver
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/


@pytest.fixture
def chrome():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

@pytest.fixture
def firefox():
    driver = webdriver.Firefox()
    yield driver
    driver.quit()


@pytest.fixture(params=['chrome', 'firefox'])
def driver(request):
    '''启动浏览器参数化'''
    return request.getfixturevalue(request.param)


def test_foo(driver):
    '''测试用例'''
    driver.get("https://www.cnblogs.com/yoyoketang/")

这样就可以分别打开 chrome 和 fixfox 执行测试用例了
关于此问题的更多讨论可以看github 上的issue Using fixtures in pytest.mark.parametrize #349

上一篇:Pytest测试框架(三):pytest fixture 用法


下一篇:直播