python - 在 python 中模拟 zipfile

Question

我正在尝试使用 Python 模拟库来模拟 zipfile 模块的一些方法。

我要测试的示例源:

def zipStuff(listOfPathToFiles):
    with ZipFile(fName, 'w') as archive:
        for each in listOfPathToFiles:
             archive.write(each, strippedfName)
    return archive

上面的“存档”在正常执行时将被忽略，但在测试期间将是文件列表。

示例单元测试代码:

emptyList=[]

def mockWrite(fName):
    emptyList.append(fName)
    return

mockZip.__enter__ = Mock(return_value=emptyList)
mockZip.__exit__ = Mock(return_value=True)

现在，我想模拟 archive.write，这样它就被 mockWrite 函数取代，而不是实际的 write 调用，这样我就可以获得所有应该被压缩的文件的列表。

我试过:

mockZip.write = Mock(side_effect=mockWrite)

但这并没有被调用。调试显示函数正在调用 mockZip.enter().write。如果我尝试:

mockZip.__enter__().write = Mock(side_effect=mockWrite)

Python 发出“列表”没有属性写入的错误(这是正确的)。我是 Mock 和 Python 的新手，非常感谢任何指点。有什么建议吗？

Answer 1

与其让 mockZip.__enter__ 返回一个空列表，不如让它返回一个如下所示的对象:

class MockZipFile:
    def __init__(self):
        self.files = []
    def __iter__(self):
        return iter(self.files)
    def write(self, fname):
        self.files.append(fname)

您可以根据需要添加方法和实现以满足您的模拟需求。