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
|
from __future__ import annotations
from contextlib import contextmanager
from pathlib import Path
import pytest
@pytest.fixture
def _mock_registry(mocker): # noqa: C901
from virtualenv.discovery.windows.pep514 import winreg # noqa: PLC0415
loc, glob = {}, {}
mock_value_str = (Path(__file__).parent / "winreg-mock-values.py").read_text(encoding="utf-8")
exec(mock_value_str, glob, loc) # noqa: S102
enum_collect = loc["enum_collect"]
value_collect = loc["value_collect"]
key_open = loc["key_open"]
hive_open = loc["hive_open"]
def _enum_key(key, at):
key_id = key.value if isinstance(key, Key) else key
result = enum_collect[key_id][at]
if isinstance(result, OSError):
raise result
return result
mocker.patch.object(winreg, "EnumKey", side_effect=_enum_key)
def _query_value_ex(key, value_name):
key_id = key.value if isinstance(key, Key) else key
result = value_collect[key_id][value_name]
if isinstance(result, OSError):
raise result
return result
mocker.patch.object(winreg, "QueryValueEx", side_effect=_query_value_ex)
class Key:
def __init__(self, value) -> None:
self.value = value
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
return None
@contextmanager
def _open_key_ex(*args):
if len(args) == 2:
key, value = args
key_id = key.value if isinstance(key, Key) else key
result = Key(key_open[key_id][value]) # this needs to be something that can be with-ed, so let's wrap it
elif len(args) == 4:
result = hive_open[args]
else:
raise RuntimeError
value = result.value if isinstance(result, Key) else result
if isinstance(value, OSError):
raise value
yield result
mocker.patch.object(winreg, "OpenKeyEx", side_effect=_open_key_ex)
mocker.patch("os.path.exists", return_value=True)
def _mock_pyinfo(major, minor, arch, exe, threaded=False):
"""Return PythonInfo objects with essential metadata set for the given args"""
from virtualenv.discovery.py_info import PythonInfo, VersionInfo # noqa: PLC0415
info = PythonInfo()
info.base_prefix = str(Path(exe).parent)
info.executable = info.original_executable = info.system_executable = exe
info.implementation = "CPython"
info.architecture = arch
info.version_info = VersionInfo(major, minor, 0, "final", 0)
info.free_threaded = threaded
return info
@pytest.fixture
def _populate_pyinfo_cache(monkeypatch):
"""Add metadata to virtualenv.discovery.cached_py_info._CACHE for all (mocked) registry entries"""
import virtualenv.discovery.cached_py_info # noqa: PLC0415
# Data matches _mock_registry fixture
python_core_path = "C:\\Users\\user\\AppData\\Local\\Programs\\Python"
interpreters = [
("ContinuumAnalytics", 3, 10, 32, False, "C:\\Users\\user\\Miniconda3\\python.exe"),
("ContinuumAnalytics", 3, 10, 64, False, "C:\\Users\\user\\Miniconda3-64\\python.exe"),
("PythonCore", 3, 9, 64, False, f"{python_core_path}\\Python36\\python.exe"),
("PythonCore", 3, 9, 64, False, f"{python_core_path}\\Python36\\python.exe"),
("PythonCore", 3, 5, 64, False, f"{python_core_path}\\Python35\\python.exe"),
("PythonCore", 3, 9, 64, False, f"{python_core_path}\\Python36\\python.exe"),
("PythonCore", 3, 7, 32, False, f"{python_core_path}\\Python37-32\\python.exe"),
("PythonCore", 3, 12, 64, False, f"{python_core_path}\\Python312\\python.exe"),
("PythonCore", 3, 13, 64, True, f"{python_core_path}\\Python313\\python3.13t.exe"),
("PythonCore", 2, 7, 64, False, "C:\\Python27\\python.exe"),
("PythonCore", 3, 4, 64, False, "C:\\Python34\\python.exe"),
("CompanyA", 3, 6, 64, False, "Z:\\CompanyA\\Python\\3.6\\python.exe"),
]
for _, major, minor, arch, threaded, exe in interpreters:
info = _mock_pyinfo(major, minor, arch, exe, threaded)
monkeypatch.setitem(virtualenv.discovery.cached_py_info._CACHE, Path(info.executable), info) # noqa: SLF001
|
