python - 使用单击选项 python 测试 cli 命令

Question

嗨，我正在尝试测试这个功能。在 cli.py 中我有

import myfunction
import click
@click.command(name="run")
@click.option("--param", required=True, type=int)
@click.option("--plot", required=False, default=False)
def run(param, plot):
    myfunction(param, plot) 

在我的 test_cli.py 中

从 click.testing 导入 CliRunner 从 cli 导入运行

def test_cli():
    kwargs = {"int": 5, "plot": False}
    runner = CliRunner()
    result = runner.invoke(run, args=kwargs)
    assert resutlts.output == ""

我收到以下错误:缺少选项 --param

Answer 1

CliRunner.invoke 采用命令行参数列表，而不是函数参数。

具体来说，您需要这样调用它:

runner.invoke(run, args=["--param", "5"])
runner.invoke(run, "--param 5")

对于多个参数，您可以使用任一模式:

runner.invoke(run, args=["--param", "5", "6"])
runner.invoke(run, args="--param 5 6")

引用文献

https://click.palletsprojects.com/en/8.0.x/testing/