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

Question

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

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

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

或者用什么代替 unittest？

Answer 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 中测试命令行应用程序？

How to test command line applications in Python?

python

unit-testing

command-line-interface

python-unittest

python-3.6