我能想到的一种方法是使用钩子。它涉及使用pytest_generate_tests
钩子,因为这允许我们对测试进行参数化。
test_id
我按以下方式设置的脚本。
def get_models():
return [1,2]
class TestEnvModelIds:
envs = ["main", "develop", "ci"]
model = get_models()
def test_ids(self, environment, model, id):
pass
请注意,我们已将测试放在一个类中,这很重要,因为稍后我们想从我们的钩子中访问这些类属性。
真正的魔法发生在下面的函数中,我们将它放在我们conftest.py
的测试目录的根目录中。我为两者创建了玩具示例,get_models
并get_ids
说明这种方法是可行的。您的实际用例可能略有不同,因为您可能需要从您实际测试的项目中导入这些功能。
def get_ids(env, model):
data = {
"main": {
1: ["a", "b"],
2: ["c", "d"]
},
"develop": {
1: ["e", "f"],
2: ["g", "h"]
},
"ci": {
1: ["i", "j"],
2: ["k", "l"]
}
}
return data[env][model]
def pytest_generate_tests(metafunc):
if "TestEnvModelIds" == metafunc.cls.__qualname__:
envs = metafunc.cls.envs
models = metafunc.cls.model
argnames = metafunc.fixturenames
argvalues = []
for env in envs:
for model in models:
ids = get_ids(env, model)
for id_ in ids:
argvalues.append((env, model, id_))
metafunc.parametrize(argnames, argvalues, scope="class")
发生的事情pytest_generate_tests
是我们迭代环境,然后是模型,最后是 id。我们创建这些三元组的列表,然后最后用它们参数化我们的测试。
当我们以冗长的方式运行测试套件时,我们可以看到我们成功地生成了所需的每个可能的测试组合。
====================================== test session starts =======================================
platform darwin -- Python 3.9.1, pytest-6.2.2, py-1.10.0, pluggy-0.13.1 -- /Users/DmitryPolonskiy/Desktop/so/bin/python3.9
cachedir: .pytest_cache
rootdir: ***
collected 12 items
tests/test_models.py::TestEnvModelIds::test_ids[main-1-a] PASSED [ 8%]
tests/test_models.py::TestEnvModelIds::test_ids[main-1-b] PASSED [ 16%]
tests/test_models.py::TestEnvModelIds::test_ids[main-2-c] PASSED [ 25%]
tests/test_models.py::TestEnvModelIds::test_ids[main-2-d] PASSED [ 33%]
tests/test_models.py::TestEnvModelIds::test_ids[develop-1-e] PASSED [ 41%]
tests/test_models.py::TestEnvModelIds::test_ids[develop-1-f] PASSED [ 50%]
tests/test_models.py::TestEnvModelIds::test_ids[develop-2-g] PASSED [ 58%]
tests/test_models.py::TestEnvModelIds::test_ids[develop-2-h] PASSED [ 66%]
tests/test_models.py::TestEnvModelIds::test_ids[ci-1-i] PASSED [ 75%]
tests/test_models.py::TestEnvModelIds::test_ids[ci-1-j] PASSED [ 83%]
tests/test_models.py::TestEnvModelIds::test_ids[ci-2-k] PASSED [ 91%]
tests/test_models.py::TestEnvModelIds::test_ids[ci-2-l] PASSED [100%]
======================================= 12 passed in 0.04s =======================================