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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
|
from __future__ import annotations
import os
import sys
from datetime import datetime, timezone
from pathlib import Path
from subprocess import CalledProcessError
from typing import TYPE_CHECKING
import pytest
from virtualenv.app_data import AppDataDiskFolder
from virtualenv.seed.wheels.acquire import download_wheel, get_wheel, pip_wheel_env_run
from virtualenv.seed.wheels.embed import BUNDLE_FOLDER, get_embed_wheel
from virtualenv.seed.wheels.periodic_update import dump_datetime
from virtualenv.seed.wheels.util import Wheel, discover_wheels
if TYPE_CHECKING:
from collections.abc import Callable
from unittest.mock import MagicMock
from pytest_mock import MockerFixture
@pytest.fixture(autouse=True)
def _fake_release_date(mocker):
mocker.patch("virtualenv.seed.wheels.periodic_update.release_date_for_wheel_path", return_value=None)
def test_pip_wheel_env_run_could_not_find(session_app_data, mocker):
mocker.patch("virtualenv.seed.wheels.acquire.from_bundle", return_value=None)
with pytest.raises(RuntimeError, match="could not find the embedded pip"):
pip_wheel_env_run([], session_app_data, os.environ)
def test_download_wheel_bad_output(mocker, for_py_version, session_app_data):
"""if the download contains no match for what wheel was downloaded, pick one that matches from target"""
distribution = "setuptools"
p_open = mocker.MagicMock()
mocker.patch("virtualenv.seed.wheels.acquire.Popen", return_value=p_open)
p_open.communicate.return_value = "", ""
p_open.returncode = 0
embed = get_embed_wheel(distribution, for_py_version)
as_path = mocker.MagicMock()
available = discover_wheels(BUNDLE_FOLDER, "setuptools", None, for_py_version)
as_path.iterdir.return_value = [i.path for i in available]
result = download_wheel(
distribution,
f"=={embed.version}",
for_py_version,
[],
session_app_data,
as_path,
os.environ,
)
assert result.path == embed.path
def test_download_fails(mocker, for_py_version, session_app_data):
p_open = mocker.MagicMock()
mocker.patch("virtualenv.seed.wheels.acquire.Popen", return_value=p_open)
p_open.communicate.return_value = "out", "err"
p_open.returncode = 1
as_path = mocker.MagicMock()
with pytest.raises(CalledProcessError) as context:
download_wheel("pip", "==1", for_py_version, [], session_app_data, as_path, os.environ)
exc = context.value
assert exc.output == "out"
assert exc.stderr == "err"
assert exc.returncode == 1
assert [
sys.executable,
"-m",
"pip",
"download",
"--progress-bar",
"off",
"--disable-pip-version-check",
"--only-binary=:all:",
"--no-deps",
"--python-version",
for_py_version,
"-d",
str(as_path),
"pip==1",
] == exc.cmd
def test_download_wheel_python_io_encoding(mocker, for_py_version, session_app_data):
mock_popen = mocker.patch("virtualenv.seed.wheels.acquire.Popen")
mock_popen.return_value.communicate.return_value = "Saved a-b-c.whl", ""
mock_popen.return_value.returncode = 0
mocker.patch("pathlib.Path.absolute", return_value=Path("a-b-c.whl"))
download_wheel("pip", "==1", for_py_version, [], session_app_data, "folder", os.environ.copy())
env = mock_popen.call_args[1]["env"]
assert env["PYTHONIOENCODING"] == "utf-8"
@pytest.fixture
def downloaded_wheel(mocker):
wheel = Wheel.from_path(Path("setuptools-0.0.0-py2.py3-none-any.whl"))
return wheel, mocker.patch("virtualenv.seed.wheels.acquire.download_wheel", return_value=wheel)
@pytest.mark.parametrize("version", ["bundle", "0.0.0"])
def test_get_wheel_download_called(mocker, for_py_version, session_app_data, downloaded_wheel, version):
distribution = "setuptools"
write = mocker.patch("virtualenv.app_data.via_disk_folder.JSONStoreDisk.write")
wheel = get_wheel(distribution, version, for_py_version, [], True, session_app_data, False, os.environ)
assert wheel is not None
assert wheel.name == downloaded_wheel[0].name
assert downloaded_wheel[1].call_count == 1
assert write.call_count == 1
@pytest.mark.parametrize("version", ["embed", "pinned"])
def test_get_wheel_download_not_called(mocker, for_py_version, session_app_data, downloaded_wheel, version):
distribution = "setuptools"
expected = get_embed_wheel(distribution, for_py_version)
if version == "pinned":
version = expected.version
write = mocker.patch("virtualenv.app_data.via_disk_folder.JSONStoreDisk.write")
wheel = get_wheel(distribution, version, for_py_version, [], True, session_app_data, False, os.environ)
assert wheel is not None
assert wheel.name == expected.name
assert downloaded_wheel[1].call_count == 0
assert write.call_count == 0
def test_get_wheel_download_cached(
tmp_path: Path,
mocker: MockerFixture,
for_py_version: str,
downloaded_wheel: tuple[Wheel, MagicMock],
time_freeze: Callable[[datetime], None],
) -> None:
time_freeze(datetime.now(tz=timezone.utc))
from virtualenv.app_data.via_disk_folder import JSONStoreDisk # noqa: PLC0415
app_data = AppDataDiskFolder(folder=str(tmp_path))
expected = downloaded_wheel[0]
write = mocker.spy(JSONStoreDisk, "write")
# 1st call, not cached, download is called
wheel = get_wheel(expected.distribution, expected.version, for_py_version, [], True, app_data, False, os.environ)
assert wheel is not None
assert wheel.name == expected.name
assert downloaded_wheel[1].call_count == 1
assert write.call_count == 1
# 2nd call, cached, download is not called
wheel = get_wheel(expected.distribution, expected.version, for_py_version, [], True, app_data, False, os.environ)
assert wheel is not None
assert wheel.name == expected.name
assert downloaded_wheel[1].call_count == 1
assert write.call_count == 1
wrote_json = write.call_args[0][1]
assert wrote_json == {
"completed": None,
"periodic": None,
"started": None,
"versions": [
{
"filename": expected.name,
"release_date": None,
"found_date": dump_datetime(datetime.now(tz=timezone.utc)),
"source": "download",
},
],
}
|
