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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
import os
import pytest
from installer.__main__ import _get_scheme_dict as get_scheme_dict
from installer.__main__ import _main as main
def test_get_scheme_dict():
d = get_scheme_dict(distribution_name="foo")
assert set(d.keys()) >= {"purelib", "platlib", "headers", "scripts", "data"}
def test_get_scheme_dict_prefix():
d = get_scheme_dict(distribution_name="foo", prefix="/foo")
for key in ("purelib", "platlib", "headers", "scripts", "data"):
assert d[key].startswith(f"{os.sep}foo"), (
f"{key} does not start with /foo: {d[key]}"
)
def test_main(fancy_wheel, tmp_path):
destdir = tmp_path / "dest"
main([str(fancy_wheel), "-d", str(destdir)], "python -m installer")
installed_py_files = destdir.rglob("*.py")
assert {f.stem for f in installed_py_files} == {"__init__", "__main__", "data"}
installed_pyc_files = destdir.rglob("*.pyc")
assert {f.name.split(".")[0] for f in installed_pyc_files} == {
"__init__",
"__main__",
}
def test_main_multiple_wheels(fancy_wheel, another_fancy_wheel, tmp_path):
destdir = tmp_path / "dest"
main(
[str(fancy_wheel), str(another_fancy_wheel), "-d", str(destdir)],
"python -m installer",
)
for wheel_name in ("fancy", "another_fancy"):
installed_py_files = destdir.rglob(f"**/{wheel_name}/**/*.py")
assert {f.stem for f in installed_py_files} == {"__init__", "__main__", "data"}
installed_pyc_files = destdir.rglob(f"**/{wheel_name}/**/*.pyc")
assert {f.name.split(".")[0] for f in installed_pyc_files} == {
"__init__",
"__main__",
}
def test_main_prefix(fancy_wheel, tmp_path):
destdir = tmp_path / "dest"
main([str(fancy_wheel), "-d", str(destdir), "-p", "/foo"], "python -m installer")
installed_py_files = list(destdir.rglob("*.py"))
for f in installed_py_files:
assert str(f.parent).startswith(str(destdir / "foo")), (
f"path does not respect destdir+prefix: {f}"
)
assert {f.stem for f in installed_py_files} == {"__init__", "__main__", "data"}
installed_pyc_files = destdir.rglob("*.pyc")
assert {f.name.split(".")[0] for f in installed_pyc_files} == {
"__init__",
"__main__",
}
def test_main_no_pyc(fancy_wheel, tmp_path):
destdir = tmp_path / "dest"
main(
[str(fancy_wheel), "-d", str(destdir), "--no-compile-bytecode"],
"python -m installer",
)
installed_py_files = destdir.rglob("*.py")
assert {f.stem for f in installed_py_files} == {"__init__", "__main__", "data"}
installed_pyc_files = destdir.rglob("*.pyc")
assert set(installed_pyc_files) == set()
@pytest.mark.parametrize(
"validation_part",
["all", "entries", "none"],
)
def test_main_validate_record_all_pass(fancy_wheel, tmp_path, validation_part):
destdir = tmp_path / "dest"
main(
[str(fancy_wheel), "-d", str(destdir), "--validate-record", validation_part],
"python -m installer",
)
installed_py_files = destdir.rglob("*.py")
assert {f.stem for f in installed_py_files} == {"__init__", "__main__", "data"}
installed_pyc_files = destdir.rglob("*.pyc")
assert {f.name.split(".")[0] for f in installed_pyc_files} == {
"__init__",
"__main__",
}
|
