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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
import builtins
import glob
import os
from pathlib import Path
from unittest import mock
import pytest
from cloudinit import atomic_helper, util
from tests.hypothesis import HAS_HYPOTHESIS
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.fixture(autouse=True)
def disable_dns_lookup(request):
if "allow_dns_lookup" in request.keywords:
yield
return
def side_effect(args, *other_args, **kwargs):
raise AssertionError("Unexpectedly used util.is_resolvable")
with mock.patch(
"cloudinit.util.is_resolvable", side_effect=side_effect, autospec=True
):
yield
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)
if HAS_HYPOTHESIS:
from hypothesis import settings # pylint: disable=import-error
settings.register_profile("ci", max_examples=1000)
settings.load_profile(os.getenv("HYPOTHESIS_PROFILE", "default"))
|
