Question

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

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

如何将测试命令行工具与其他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中测试命令行应用程序？

1 个答案: