1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
import builtins
import glob
import os
from pathlib import Path
import pytest
from cloudinit import atomic_helper, util
from tests.unittests.helpers import retarget_many_wrapper
FS_FUNCS = {
os.path: [
("isfile", 1),
("exists", 1),
("islink", 1),
("isdir", 1),
("lexists", 1),
("relpath", 1),
],
os: [
("listdir", 1),
("mkdir", 1),
("lstat", 1),
("symlink", 2),
("stat", 1),
("scandir", 1),
],
util: [
("write_file", 1),
("append_file", 1),
("load_file", 1),
("ensure_dir", 1),
("chmod", 1),
("delete_dir_contents", 1),
("del_file", 1),
("sym_link", -1),
("copy", -1),
],
glob: [
("glob", 1),
],
builtins: [
("open", 1),
],
atomic_helper: [
("write_file", 1),
],
}
@pytest.fixture
def fake_filesystem(mocker, tmpdir):
"""Mocks fs functions to operate under `tmpdir`"""
for (mod, funcs) in FS_FUNCS.items():
for f, nargs in funcs:
func = getattr(mod, f)
trap_func = retarget_many_wrapper(str(tmpdir), nargs, func)
mocker.patch.object(mod, f, trap_func)
PYTEST_VERSION_TUPLE = tuple(map(int, pytest.__version__.split(".")))
if PYTEST_VERSION_TUPLE < (3, 9, 0):
@pytest.fixture
def tmp_path(tmpdir):
return Path(tmpdir)
|
