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

Posted 2021-07-15 上海-悠悠

前言

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

使用案例

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

import pytest


@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 0x000001C7E77CE7B8>

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

\\test_xx.py:16: TypeError
F
arg = <function b at 0x000001C7E77CE8C8>

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

\\test_xx.py:16: TypeError

关于此问题的讨论可以看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-lazy-fixture 插件解决 https://www.cnblogs.com/yoyoketang/p/14096749.html