如何在 Python 中测试命令行应用程序？答案

【问题标题】：How to test command line applications in Python?如何在 Python 中测试命令行应用程序？
【发布时间】：2018-08-07 23:47:09
【问题描述】：

我知道 Python unittest。我有一些使用它来测试 Python 子程序的经验。

现在我需要添加测试我用 Python 编写的命令行应用程序（不只是一个 Python 函数）。我想在标准输入中使用某些参数和某些输入来调用它，并在标准输出中测试输出。

如何将测试命令行工具与其他unittest 测试用例集成？

或者用什么来代替unittest？

【问题讨论】：

听起来像是验证或集成测试，可以用任何语言实现，python unittest 应该可以。

标签： python unit-testing command-line-interface python-3.6 python-unittest

【解决方案1】：

您仍然可以使用标准的unittest 格式并将整个应用程序作为标准功能进行测试。制作一个包装器，使脚本入口点成为一个简单的包装器，例如：

if __name__ == "__main__":
    sys.exit(main(sys.argv))

只要不滥用全局变量保持状态，只需要测试main()函数即可。

如果您想测试您无法控制的脚本，您仍然可以通过编写如下测试来使用unittest 和subprocess：

def test_process(self):
    result = subprocess.run(['your_script', 'your_args', ...], capture_output=True)
    self.assertIn('expected out', result.stdout)

def test_process_failure(self):
    result = subprocess.run(['your_script', 'your_args', ...], capture_output=True)
    self.assertEqual(result.returncode, 1)

【讨论】：

我想过这个。但是还有其他方法吗？（我希望它可以与各种 Python 脚本一起使用，而不仅仅是按照您在回答中给出的指导方针完成的脚本。）只是为了确保。
@porton 添加了一个从外部测试脚本的示例