Python -模拟ZipFile

Question

我正在尝试使用python进行更深入的测试，现在我很难尝试为以下代码编写测试：

def get_files(zip_path: Path):
    archive = zipfile.ZipFile(os.path.join(os.path.dirname(__file__), '..', zip_path))
    python_files = []
    for x in archive.filelist:
        if x.filename.endswith(".py"):
            python_files.append(x)
    return python_files

我想出来的测试是这样的：

@mock.patch('zipfile.ZipFile')
def test_get_files(mock_zipfile):
    mock_zipfile.return_value.filelist.return_value = [zipfile.ZipInfo('py_file.py'), zipfile.ZipInfo('py_file.py'),
                                                       zipfile.ZipInfo('any_file.any')]
    nodes = get_ast_nodes(Path('/dummy/path/archive.zip'))
    assert len(nodes) == 2

但是我不能让测试通过，我也不知道哪里出了问题。

Answer 1

如果有人在查这个，我可能还会加上答案。这就是我是如何让它工作的：

@mock.patch('zipfile.ZipFile')
def test_get_files(mock_zipfile):
    mock_zipfile.return_value.filelist = [zipfile.ZipInfo('py_file.py'), zipfile.ZipInfo('py_file.py'),
                                          zipfile.ZipInfo('any_file.any')]
    nodes = get_python_files(zipfile.ZipFile("dummy"))
    assert len(nodes) == 2