python - 在测试中检查 CLI 的退出代码

Question

我正在为命令行工具编写自动化测试。本质上，我想使用各种选项调用 CLI 并测试退出代码和/或输出。

我的测试如下所示:

from mymodule.cli_tool import main

def test_args(capfd):
    with pytest.raises(SystemExit) as e:
        main(args=['--junk_option'])
    # check the exit code to make sure it is non-zero
    ???

如何检查退出代码？

Answer 1

您需要检查value.code，如下所示:

assert e.value.code != 0