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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
from __future__ import annotations
import re
import shutil
import tempfile
from pathlib import Path
import pytest
from auditwheel.architecture import Architecture
from auditwheel.error import NonPlatformWheelError
from auditwheel.libc import Libc
from auditwheel.wheeltools import (
InWheelCtx,
WheelToolsError,
add_platforms,
get_wheel_architecture,
get_wheel_libc,
)
HERE = Path(__file__).parent.resolve()
@pytest.mark.parametrize(
("filename", "expected"),
[(f"foo-1.0-py3-none-linux_{arch}.whl", arch) for arch in Architecture]
+ [("foo-1.0-py3-none-linux_x86_64.manylinux1_x86_64.whl", Architecture.x86_64)],
)
def test_get_wheel_architecture(filename: str, expected: Architecture) -> None:
arch = get_wheel_architecture(filename)
assert arch == expected.baseline
def test_get_wheel_architecture_unknown() -> None:
with pytest.raises(WheelToolsError, match=re.escape("unknown architecture")):
get_wheel_architecture("foo-1.0-py3-none-linux_mipsel.whl")
def test_get_wheel_architecture_pure() -> None:
with pytest.raises(NonPlatformWheelError):
get_wheel_architecture("foo-1.0-py3-none-any.whl")
@pytest.mark.parametrize(
"filename",
[
"foo-1.0-py3-none-linux_x86_64.linux_aarch64.whl",
"foo-1.0-py3-none-linux_x86_64.linux_mipsel.whl",
"foo-1.0-py3-none-linux_x86_64.any.whl",
],
)
def test_get_wheel_architecture_multiple(filename: str) -> None:
match = re.escape("multiple architectures are not supported")
with pytest.raises(WheelToolsError, match=match):
get_wheel_architecture(filename)
@pytest.mark.parametrize(
("filename", "expected"),
[
("foo-1.0-py3-none-manylinux1_x86_64.whl", Libc.GLIBC),
("foo-1.0-py3-none-manylinux1_x86_64.manylinux2010_x86_64.whl", Libc.GLIBC),
("foo-1.0-py3-none-musllinux_1_1_x86_64.whl", Libc.MUSL),
],
)
def test_get_wheel_libc(filename: str, expected: Libc) -> None:
libc = get_wheel_libc(filename)
assert libc == expected
@pytest.mark.parametrize(
"filename",
["foo-1.0-py3-none-any.whl", "foo-1.0-py3-none-something.whl"],
)
def test_get_wheel_libc_unknown(filename: str) -> None:
with pytest.raises(WheelToolsError, match=re.escape("unknown libc used")):
get_wheel_libc(filename)
@pytest.mark.parametrize(
"filename",
["foo-1.0-py3-none-manylinux1_x86_64.musllinux_1_1_x86_64.whl"],
)
def test_get_wheel_libc_multiple(filename: str) -> None:
match = re.escape("multiple libc are not supported")
with pytest.raises(WheelToolsError, match=match):
get_wheel_libc(filename)
def test_inwheel_tmpdir(tmp_path, monkeypatch):
wheel_path = (
HERE / "../integration/arch-wheels/glibc/testsimple-0.0.1-cp313-cp313-linux_x86_64.whl"
)
tmp_path = tmp_path.resolve(strict=True)
tmpdir = tmp_path / "tmpdir"
tmpdir.mkdir()
tmpdir_symlink = tmp_path / "symlink"
tmpdir_symlink.symlink_to(str(tmpdir), target_is_directory=True)
monkeypatch.setattr(tempfile, "gettempdir", lambda: str(tmpdir_symlink))
with InWheelCtx(wheel_path, tmp_path / wheel_path.name) as context:
Path(context._tmpdir.name).relative_to(tmpdir_symlink)
context.name.relative_to(tmpdir)
def test_inwheel_no_manager(tmp_path):
wheel_path = (
HERE / "../integration/arch-wheels/glibc/testsimple-0.0.1-cp313-cp313-linux_x86_64.whl"
)
context = InWheelCtx(wheel_path, tmp_path / wheel_path.name)
with pytest.raises(
ValueError,
match=re.escape("This function should be called from context manager"),
):
next(context.iter_files())
with pytest.raises(
ValueError,
match=re.escape("This function should be called from wheel_ctx context manager"),
):
add_platforms(context, [], [])
def test_inwheel_no_distinfo():
wheel_path = (
HERE / "../integration/arch-wheels/glibc/testsimple-0.0.1-cp313-cp313-linux_x86_64.whl"
)
with InWheelCtx(wheel_path, None) as context:
dist_info = list(context.path.glob("*.dist-info"))
assert len(dist_info) == 1
shutil.rmtree(dist_info[0])
with pytest.raises(
WheelToolsError,
match=re.escape("Should be exactly one `*.dist_info` directory"),
):
next(context.iter_files())
|
