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
|
from __future__ import annotations
from pathlib import Path
from subprocess import CalledProcessError
from unittest.mock import call, patch
import pytest
from auditwheel.patcher import Patchelf
@patch("auditwheel.patcher.which")
def test_patchelf_unavailable(which):
which.return_value = False
with pytest.raises(ValueError, match="Cannot find required utility"):
Patchelf()
@patch("auditwheel.patcher.which")
@patch("auditwheel.patcher.check_output")
def test_patchelf_check_output_fail(check_output, which):
which.return_value = True
check_output.side_effect = CalledProcessError(1, "patchelf --version")
with pytest.raises(ValueError, match="Could not call"):
Patchelf()
@patch("auditwheel.patcher.which")
@patch("auditwheel.patcher.check_output")
@pytest.mark.parametrize("version", ["0.14", "0.14.1", "0.15"])
def test_patchelf_version_check(check_output, which, version):
which.return_value = True
check_output.return_value.decode.return_value = f"patchelf {version}"
Patchelf()
@patch("auditwheel.patcher.which")
@patch("auditwheel.patcher.check_output")
@pytest.mark.parametrize("version", ["0.13.99", "0.13", "0.9", "0.1"])
def test_patchelf_version_check_fail(check_output, which, version):
which.return_value = True
check_output.return_value.decode.return_value = f"patchelf {version}"
with pytest.raises(ValueError, match=f"patchelf {version} found"):
Patchelf()
@patch("auditwheel.patcher._verify_patchelf")
@patch("auditwheel.patcher.check_output")
@patch("auditwheel.patcher.check_call")
class TestPatchElf:
""" "Validate that patchelf is invoked with the correct arguments."""
def test_replace_needed_one(self, check_call, _0, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
soname_old = "TEST_OLD"
soname_new = "TEST_NEW"
patcher.replace_needed(filename, (soname_old, soname_new))
check_call.assert_called_once_with(
["patchelf", "--replace-needed", soname_old, soname_new, filename],
)
def test_replace_needed_multple(self, check_call, _0, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
replacements = [
("TEST_OLD1", "TEST_NEW1"),
("TEST_OLD2", "TEST_NEW2"),
]
patcher.replace_needed(filename, *replacements)
check_call.assert_called_once_with(
[
"patchelf",
"--replace-needed",
*replacements[0],
"--replace-needed",
*replacements[1],
filename,
],
)
def test_set_soname(self, check_call, _0, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
soname_new = "TEST_NEW"
patcher.set_soname(filename, soname_new)
check_call.assert_called_once_with(
["patchelf", "--set-soname", soname_new, filename],
)
def test_set_rpath(self, check_call, _0, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
patcher.set_rpath(filename, "$ORIGIN/.lib")
check_call_expected_args = [
call(["patchelf", "--remove-rpath", filename]),
call(
["patchelf", "--force-rpath", "--set-rpath", "$ORIGIN/.lib", filename],
),
]
assert check_call.call_args_list == check_call_expected_args
def test_get_rpath(self, _0, check_output, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
check_output.return_value = b"existing_rpath"
result = patcher.get_rpath(filename)
check_output_expected_args = [call(["patchelf", "--print-rpath", filename])]
assert result == check_output.return_value.decode()
assert check_output.call_args_list == check_output_expected_args
def test_remove_needed(self, check_call, _0, _1): # noqa: PT019
patcher = Patchelf()
filename = Path("test.so")
soname_1 = "TEST_REM_1"
soname_2 = "TEST_REM_2"
patcher.remove_needed(filename, soname_1, soname_2)
check_call.assert_called_once_with(
[
"patchelf",
"--remove-needed",
soname_1,
"--remove-needed",
soname_2,
filename,
],
)
|
