如何对使用Popen的函数进行单元测试？

Question

我正在编写一个包含大量文件操作的程序。 一些操作是通过调用subprocess.Popen完成的，例如， split -l 50000 ${filename} ， gzip -d -f ${filename} ${filename}. 。

现在我想单元测试程序的功能。 但是我怎么能对这些功能进行单元测试？

有什么建议？

Answer 1

规范的方法是模拟对Popen的调用，并用一些测试数据替换结果。 看看mock库文档 。 ¹

你会做这样的事情：

with mock.patch.object(subprocess, 'Popen') as mocked_popen:
    mocked_popen.return_value.communicate.return_value = some_fake_result
    function_which_uses_popen_communicate()

现在你可以做一些检查或者你想要测试的任何东西......

^{1请注意，它在python3.3中作为unittest.mock包含在标准库中。}