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 contextlib
import importlib
import sys
from editables.redirector import RedirectingFinder as F
@contextlib.contextmanager
def save_import_state():
orig_modules = set(sys.modules.keys())
orig_path = list(sys.path)
orig_meta_path = list(sys.meta_path)
orig_path_hooks = list(sys.path_hooks)
orig_path_importer_cache = sys.path_importer_cache
try:
yield
finally:
remove = [key for key in sys.modules if key not in orig_modules]
for key in remove:
del sys.modules[key]
sys.path[:] = orig_path
sys.meta_path[:] = orig_meta_path
sys.path_hooks[:] = orig_path_hooks
sys.path_importer_cache.clear()
sys.path_importer_cache.update(orig_path_importer_cache)
# HACK
F._redirections = {}
def build(target, structure):
target.mkdir(exist_ok=True, parents=True)
for name, content in structure.items():
path = target / name
if isinstance(content, str):
path.write_text(content, encoding="utf-8")
else:
build(path, content)
def test_double_install():
with save_import_state():
old_len = len(sys.meta_path)
F.install()
F.install()
assert len(sys.meta_path) == old_len + 1
def test_toplevel_only():
assert F.find_spec("foo.bar") is None
def test_no_path():
assert F.find_spec("foo", path=[]) is None
def test_no_map_returns_none():
assert F.find_spec("foo") is None
def test_redirects(tmp_path):
project = tmp_path / "project"
project_files = {
"mod.py": "val = 42",
"pkg": {
"__init__.py": "val = 42",
"sub.py": "val = 42",
},
}
build(project, project_files)
with save_import_state():
F.install()
F.map_module("mod", project / "mod.py")
F.map_module("pkg", project / "pkg/__init__.py")
import mod
assert mod.val == 42
import pkg
assert pkg.val == 42
import pkg.sub
assert pkg.sub.val == 42
def test_cache_invalidation():
F.install()
# assert that the finder matches importlib's expectations
# see https://github.com/pfmoore/editables/issues/31
importlib.invalidate_caches()
|
