如何从 pytest 固定装置正确返回列表以用于参数化？

Question

我正在尝试构建一个小的 pytest 测试，以确保 redis 中存在所有预期的键。我有一个预期键列表，我将其存储为 YML 文件。测试本身将查询 redis 以确保列表中的每个预期键都存在。

最初，我将此设置为test_keys.py文件中的一个巨大列表。这是这样设置的：

expected_keys = ['key1','key2','key3']
@pytest.mark.parametrize('expected_key', expected_keys)
def test_expected_key(expected_key):
    ...

这有效。由于我想为 redis 环境的其他一些检查复制这种类型的测试，因此我不想将包含几百个键的多个列表放入这些文件中。

我想我可以将它们提取到 YML 文件中并通过装置加载键。

我的夹具看起来像这样：

@pytest.fixture
def expected_keys_fixture():
    with open('expected_keys.yml'), 'r') as f:
        return yaml.safe_load(f)['keys']

YML 看起来像这样：

keys:
  - key1
  - key2
  - key3

我的测试装饰器更改为：

@pytest.mark.parametrize("expected_keys", [
    (pytest.lazy_fixture('expected_keys_fixture'))
])
def test_expected_key(expected_key):
    ...

我正在pytest-lazy-fixture为此使用该软件包。

我在这里expected_keys遇到的问题是现在等于预期键的整个列表。当静态列表在我的测试文件中时，它不再是每个单独的密钥。

我试图按照 Oleh Rybalchenko 的回答建议去做

@pytest.mark.parametrize("expected_keys", pytest.lazy_fixture('expected_keys_fixture')
)
def test_expected_key(expected_key):
    ...

但是，这失败了TypeError: 'LazyFixture' object is not iterable。

我知道我只有一个 argname 并且我应该有一个列表，但是文档中的示例将参数传递给fixture()装饰器。我不是。我的列表是由 YML 文件的结果生成的。

---

问题：如何调整我的装置，以便它一次为parametrize装饰者正确返回一个项目？

Answer 1

在同样的情况下，我发现的唯一方法是拒绝夹具并直接调用该函数。

def expected_keys_fixture():
    with open('expected_keys.yml', 'r') as f:
        return yaml.safe_load(f)['keys']


@pytest.mark.parametrize("expected_key",
                         expected_keys_fixture())
def test_expected_key(expected_key):
    assert expected_key in ['key1', 'key2', 'key3']