python - 如何在不运行 CLIrunner.invoke() 的情况下将输入传递给 click.confirm

Question

我有一个用例，其中一个函数需要用户确认才能继续。这基本上是:

def abc()
    response = click.confirm("Do you want to continue?")

根据此响应，它要么中止，要么继续。

问题是要测试这个功能 abc .

@click.option('--yes', prompt=True)
def test_abc():
    with mock.patch.object(click.confirm, input="n"):
        click.echo("Aborted")

Answer 1

老问题，但我有同样的问题，我决定 mock click.confirm这让我可以模拟某人选择 [y/N]让我们说你的 abc()看起来像这样:

def abc():
    if click.confirm("do you want to continue?", abort=True):
        return "Confirmed"

    return "Aborted"

然后你的测试模拟 click.confirm至 y/n

@mock.patch("click.confirm")
def test_abc_aborted(mock_click):
    mock_click.return_value = False
    assert abc() == "Aborted"


@mock.patch("click.confirm")
def test_abc_confirmed(mock_click):
    mock_click.return_value = "y"
    assert abc() == "Confirmed"