Pytest：如何在运行时更改 ~/.bashrc 令牌？

Question

我想在我的 bashrc 中动态更改令牌以断言预期的结果。

例如：在我的 ~/.bashrc 我有我的令牌集

export GITHUB_ACCESS_TOKEN=ghp_NNNNNNNNNNNN

在测试期间我想设置令牌

export GITHUB_ACCESS_TOKEN=TEST

然后断言通过运行单击命令来检查我是否无法访问我的存储库：

result = runner.invoke(clicker_cli, ["git", "clone", "<url_here>"])

它没有按预期工作。我仍然可以使用我的原始令牌访问我的仓库。

上下文： 我正在使用https://click.palletsprojects.com/en/8.0.x/ https://docs.pytest.org/en/6.2.x/monkeypatch.html

Answer 1

您可以设置将在运行时使用的可选env 字典 (Mapping)。查看invokedocumentation 了解详情。所以下面的代码就可以解决问题。

result = runner.invoke(
   clicker_cli, ["git", "clone", "<url_here>"], 
   env={"GITHUB_ACCESS_TOKEN": "TEST"}
)

完整示例

这是一个完整的工作示例。

import click
import os
from click.testing import CliRunner


@click.command()
@click.argument("msg")
def echo_token(msg):
    click.echo(f"{msg} {os.environ.get('GITHUB_ACCESS_TOKEN')}")


def test_echo_token(token="MY_TOKEN"):
    runner = CliRunner()
    result = runner.invoke(echo_token, ["Token is"],
             env={"GITHUB_ACCESS_TOKEN": token})
    assert result.exit_code == 0
    assert token in result.output
    print(result.output)

运行我们可以看到环境变量设置正确。

pytest -s click_test.py
# click_test.py Token is MY_TOKEN