# 利用 pytest_generate_tests 实现命令行参数动态注入
在自动化测试中，硬编码的测试参数往往难以适应频繁变化的数据源或 CI 环境。pytest 提供了强大的钩子函数 `pytest_generate_tests`，允许我们在 `conftest.py` 中定义自定义的参数化方案，将命令行传入的参数列表动态绑定到 fixture 上。
## 核心原理：从硬编码到动态获取
传统的参数化通常使用 `@pytest.mark.parametrize` 装饰器，但这要求直接在代码中列出所有测试用例。当需要频繁修改数据源（如读取配置文件或根据环境变化）时，这种静态定义显得僵化。
`pytest_generate_tests` 钩子函数在 pytest 收集测试阶段被调用。通过传入的 `metafunc` 对象，我们可以访问当前测试上下文，特别是通过 `metafunc.config.getoption()` 获取命令行选项的值，并调用 `metafunc.parametrize()` 动态生成测试用例。
## 实现步骤：定义命令行选项与钩子函数
首先，需要在 `conftest.py` 中注册一个新的命令行选项。这可以通过 `pytest_addoption` 钩子完成，该钩子允许添加自定义的 pytest 参数。
```python
# conftest.py
def pytest_addoption(parser):
parser.addoption(
"--stringinput",
action="append",
default=[],
help="list of stringinputs to pass to test functions",
)
```
在此示例中，`--stringinput` 选项被定义为可多次追加（`action="append"`），默认值为空列表。这使得用户可以通过多次传入该参数来提供多个测试数据。
接下来，定义 `pytest_generate_tests` 钩子函数。该函数检查当前测试是否依赖名为 `stringinput` 的 fixture。如果是，则从命令行获取选项值并执行参数化。
```python
def pytest_generate_tests(metafunc):
if "stringinput" in metafunc.fixturenames:
# 获取命令行传入的参数列表，默认为空列表
param_values = metafunc.config.getoption("stringinput")
# 动态绑定参数到 fixture
metafunc.parametrize("stringinput", param_values)
```
## 运行测试：动态数据驱动验证
当运行测试时，如果未指定命令行参数，`param_values` 为空列表。此时 pytest 会跳过该测试（取决于配置中的 `empty_parameter_set_mark` 选项），避免无意义的执行。
若通过命令行传入多个字符串值，如 `--stringinput="hello" --stringinput="world"`，pytest 将收集并运行两次测试，每次使用不同的输入数据。反之，若传入非法字符（如 `!`），测试将正常执行并可能失败，从而验证了动态注入的有效性。
## 优势与注意事项
使用 `pytest_generate_tests` 的主要优势在于解耦了测试逻辑与测试数据。测试代码无需感知数据来源，只需依赖 fixture 名称即可。这使得测试脚本更加通用和可维护。
需要注意的是，`pytest_generate_tests` 钩子可以在 `conftest.py` 中定义，也可以直接在测试模块或类中实现（其他钩子则必须位于 `conftest.py` 或插件中）。此外，若多次调用 `metafunc.parametrize()`，参数名称不得重复，否则会导致错误。
通过这种方式，你可以轻松构建适应 CI/CD 流水线的灵活测试框架，无需修改测试代码即可调整测试数据源。