如何使用临时文件系统在python中模拟os.walk？

Question

我正在尝试测试一些使用os.walk的代码。 我想创建一个临时的内存文件系统，我可以填充样本（空）文件和os.walk将返回的目录。 这样可以省去模拟os.walk调用以模拟递归的复杂性。

具体来说，我要测试的代码是：

if recursive:
    log.debug("Recursively searching for files under %s" % path)

    for (dir_path, dirs, files) in os.walk(path):
        log.debug("Found %d files in %s: %s" % (len(files), path, files))
        for f in [os.path.join(dir_path, f) for f in files
                  if not re.search(exclude, f)]:
            yield f
else:
    log.debug("Non-recursively searching for files under %s" % path)

    for (dir_path, dirs, files) in os.walk(path):
        log.debug("Found %d files in %s: %s" % (len(files), path, files))
        for f in [os.path.join(dir_path, f) for f in files
                    if not re.search(exclude, f)]:
            yield f

这在python中可能吗？

Answer 1

os.walk()完全由os.listdir()构建，并在os.path.islink()和os.path.isdir()帮助下构建。 这些本质上是系统调用，因此您必须在系统级别模拟文件系统 。 除非你想编写一个FUSE插件，否则这并不容易模拟。

所有os.walk()需要返回的是一个元组列表，真的。 除非您正在测试操作dirs组件，否则它不会更简单：

with mock.patch('os.walk') as mockwalk:
    mockwalk.return_value = [
        ('/foo', ('bar',), ('baz',)),
        ('/foo/bar', (), ('spam', 'eggs')),
    ]

这将模拟以下目录结构：

/foo
 ├── baz
 └── bar 
     ├── spam
     └── eggs