使用 subprocess.Popen 的单元测试 Python 代码

我有一个 Python 项目，在其中读取外部文件，处理它们，并将结果写入新文件。输入文件可以直接读取，也可以使用以下命令从 git 存储库中提取git show。要调用的函数git show并返回标准输出如下所示：

def git_show(fname, rev):
    '''Runs git show and returns stdout'''
    process = subprocess.Popen(['git', 'show', '{}:{}'.format(rev, fname)],
                               stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = process.communicate()
    ret_code = process.wait()
    if ret_code:
        raise Exception(stderr)
    return stdout

我有单元测试来测试程序的整个处理部分，即除了读取和写入文件之外的所有内容。但是，我偶然发现（并修复了）有关返回字符串的编码的问题git_show()，取决于 Python 版本，很可能还取决于操作系统和要读取的实际文件。

我想设置一个单元测试git_show()这样我就可以确保整个应用程序从输入到输出都能正常工作。然而，据我所知，如果没有实际的 git 存储库来测试，这是不可能的。整个包是用 git 进行版本管理的，我希望如果我在 git 存储库中有一个 git 存储库，那么它本身可能会导致问题，并且我脑海中的一个声音告诉我这可能不是最好的解决方案。

如何最好地实现从以下位置获取输入的单元测试代码git show（一般来说，命令行 /Popen.communicate())?

所以我这样做的方法是pytest https://pytest.org

Example: (做作的)

from subprocess import Popen, PIPE


def test():
    p = Popen(["echo", "Hello World!"], stdout=PIPE)
    stdout, _ = p.communicate()

    assert stdout == b"Hello World!\n"

Output:

$ py.test -x -s test_subprocess.py 
======================================= test session starts ========================================
platform linux2 -- Python 2.7.9 -- py-1.4.28 -- pytest-2.7.1
rootdir: /home/prologic/work/circuits, inifile: 
plugins: cov
collected 1 items 

test_subprocess.py .

===================================== 1 passed in 0.01 seconds =====================================

或者使用标准库unittest https://docs.python.org/2/library/unittest.html:

Example:

#!/usr/bin/env python


from unittest import main, TestCase


from subprocess import Popen, PIPE


class TestProcess(TestCase):

    def test(self):
        p = Popen(["echo", "Hello World!"], stdout=PIPE)
        stdout, _ = p.communicate()

        self.assertEquals(stdout, b"Hello World!\n")


if __name__ == "__main__":
    main()

Output:

$ python test_subprocess.py 
.
----------------------------------------------------------------------
Ran 1 test in 0.001s

OK