Python

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

Python - 模拟 ZipFile

Python - Mocking a ZipFile

testing

mocking