使用subprocess.Popen

时间:2015-05-29 11:12:54

标签: python git unit-testing

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

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资源库,可能会导致问题,而且我脑子里的一个声音告诉我,这可能不是最好的解决方案。< / p>

如何才能最好地实现从git show(通常是命令行/ Popen.communicate())获取输入的单元测试代码?

2 个答案:

答案 0 :(得分:3)

也许你想要(其中一种)不同类型的测试

单元测试

在代码中测试代码的一小部分。

  1. 模拟subprocess.Popen
  2. stdout, stderr
  3. 中返回静态值
  4. 检查处理是否正确
  5. 示例代码非常小,您只能测试stdout是否真正返回,并且在非零wait()时会引发异常。

    介于两者之间

    测试向量,即给定的设置输入,应该产生设置输出

    1. 模拟git,而是使用以特定方式编码的cat vector1.txt
    2. 测试结果
    3. 整合测试

      测试代码如何连接到外部实体,在本例中为git。此类测试可以防止您意外更改内部系统的期望。就是它&#34;冻结&#34; API。

      1. 使用小型git存储库创建tarball
      2. 可选择将git二进制文件打包到相同的tarball
      3. 解压缩tarball
      4. 运行git命令
      5. 将输出与预期
      6. 进行比较

答案 1 :(得分:1)

所以我这样做是为了pytest

示例:人为

from subprocess import Popen, PIPE


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

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

<强>输出:

$ 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

示例:

#!/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()

<强>输出:

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

OK