Question

当试图模仿Popen时，如果在单元测试代码和主模块代码中导入子进程匹配，我只能使它成功。

给出以下模块listdir.py：

from subprocess import Popen, PIPE

def listdir(dir):
    cmd = ['ls', dir]
    pc = Popen(cmd, stdout=PIPE, stderr=PIPE)
    out, err = pc.communicate()
    if pc.returncode != 0:
        raise Exception
    return out

并跟随单元测试代码test_listdir.py

import subprocess
import listdir
import mock

@mock.patch.object(subprocess, 'Popen', autospec=True)
def test_listdir(mock_popen):
    mock_popen.return_value.returncode = 0
    mock_popen.return_value.communicate.return_value = ("output", "Error")
    listdir.listdir("/fake_dir")

出于某种原因，Popen没有被嘲笑，因为两个python模块之间的导入风格不同，并且运行测试总是引发异常。

如果我更改listdir.py以导入所有子进程，例如

import subprocess

def listdir(dir):
    cmd = ['ls', dir]
    pc = subprocess.Popen(cmd, stdout=subprocess.PIPE, 
                          stderr=subprocess.PIPE)
    out, err = pc.communicate()
    if pc.returncode != 0:
        raise ListingErrorException
    return out

然后＆＃34;输出＆＃34;在测试中返回。

任何人都想了解为什么，我的偏好是有的从子进程导入Popen，在两个模块中使用Pipe，但我无法将其转换为模拟。

Answer 1

您需要在listdir中修补Popen的副本，而不是刚导入的那个。因此，请尝试@mock.patch.object(subprocess, 'Popen', autospec=True)

，而不是@mock.patch.object(listdir, 'Popen', autospec=True)

有关详细信息，请参阅此文档：http://www.voidspace.org.uk/python/mock/patch.html#where-to-patch

mocking subprocess.Popen依赖于导入样式

1 个答案: