pytest 文件中的函数仅适用于硬编码值

Question

我有下面的 test_dss.py 文件用于 pytest:

import dataikuapi

import pytest
 

def setup_list():

    client = dataikuapi.DSSClient("{DSS_URL}", "{APY_KEY}")

    client._session.verify = False

    project = client.get_project("{DSS_PROJECT}")

    # Check that there is at least one scenario TEST_XXXXX & that all test scenarios pass

    scenarios = project.list_scenarios()

    scenarios_filter = [obj for obj in scenarios if obj["name"].startswith("TEST")]

    return scenarios_filter

 
def test_check_scenario_exist():

    assert len(setup_list()) > 0, "You need at least one test scenario (name starts with 'TEST_')"

 
@pytest.mark.parametrize("scenario", setup_list())

def test_scenario_run(scenario, params):

    client = dataikuapi.DSSClient(params['host'], params['api'])

    client._session.verify = False

    project = client.get_project(params['project'])

    scenario_id = scenario["id"]

    print("Executing scenario ", scenario["name"])

    scenario_result = project.get_scenario(scenario_id).run_and_wait()

    assert scenario_result.get_details()["scenarioRun"]["result"]["outcome"] == "SUCCESS", "test " + scenario[

    "name"] + " failed"

我的问题是 setup_list 函数，它只能获取 {DSS_URL}、{APY_KEY}、{PROJECT} 的硬编码值。我无法使用 PARAMS 或 test_scenario_run 中的其他方法知道如何将 PARAMS 也传递给此函数吗？

Answer 1

mark.parametrize 标记中的参数在加载时读取，其中有关配置参数的信息尚不可用。因此，您必须在运行时间对测试进行参数化，您可以访问配置。

这可以在 pytest_generate_tests 中完成（它可以存在于您的测试模块中）：

@pytest.hookimpl
def pytest_generate_tests(metafunc):
    if "scenario" in metafunc.fixturenames:
        host = metafunc.config.getoption('--host')
        api = metafuc.config.getoption('--api')
        project = metafuc.config.getoption('--project')
        metafunc.parametrize("scenario", setup_list(host, api, project))

这意味着您的 setup_list 函数采用这些参数：

def setup_list(host, api, project):
    client = dataikuapi.DSSClient(host, api)
    client._session.verify = False
    project = client.get_project(project)
    ...

你的测试看起来像这样（没有 parametrize 标记，因为参数化现在在 pytest_generate_tests 中完成）：

def test_scenario_run(scenario, params):
    scenario_id = scenario["id"]
    ...

参数化现在在运行时完成，因此其行为与您在测试中放置 parametrize 标记相同。

而另一个测试 setup_list 的测试现在也必须使用 params fixture 来获取所需的参数：

def test_check_scenario_exist(params):
    assert len(setup_list(params["host"], params["api"], params["project"])) > 0,
 "You need at least ..."

pytest 文件中的函数仅适用于硬编码值

Function in pytest file works only with hard coded values

pytest

assert

fixtures

hardcoded

parametrized-testing