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

问题描述

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

现在我需要添加测试用Python编写的命令行应用程序(而不是只是一个Python函数)。我希望在stdin中使用某些参数和某些输入来调用它，在stdout中使用测试输出来调用它。

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

或使用什么代替unittest？

---

解决方案

您仍然可以使用标准的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)