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
|
from __future__ import annotations
import sys
from pathlib import Path
def test_virtualenv_py_race_condition_find_spec(tmp_path):
"""Test that _Finder.find_spec handles NameError gracefully when _DISTUTILS_PATCH is not defined."""
# Create a temporary file with partial _virtualenv.py content (simulating race condition)
venv_file = tmp_path / "_virtualenv_test.py"
# Write a partial version of _virtualenv.py that has _Finder but not _DISTUTILS_PATCH
# This simulates the state during a race condition where the file is being rewritten
helper_file = Path(__file__).parent / "_test_race_condition_helper.py"
partial_content = helper_file.read_text(encoding="utf-8")
venv_file.write_text(partial_content, encoding="utf-8")
sys.path.insert(0, str(tmp_path))
try:
import _virtualenv_test # noqa: PLC0415
finder = _virtualenv_test.finder
# Try to call find_spec - this should not raise NameError
result = finder.find_spec("distutils.dist", None)
assert result is None, "find_spec should return None when _DISTUTILS_PATCH is not defined"
# Create a mock module object
class MockModule:
__name__ = "distutils.dist"
# Try to call exec_module - this should not raise NameError
def mock_old_exec(_x):
pass
finder.exec_module(mock_old_exec, MockModule())
# Try to call load_module - this should not raise NameError
def mock_old_load(_name):
return MockModule()
result = finder.load_module(mock_old_load, "distutils.dist")
assert result.__name__ == "distutils.dist"
finally:
sys.path.remove(str(tmp_path))
if "_virtualenv_test" in sys.modules:
del sys.modules["_virtualenv_test"]
def test_virtualenv_py_normal_operation():
"""Test that the fix doesn't break normal operation when _DISTUTILS_PATCH is defined."""
# Read the actual _virtualenv.py file
virtualenv_py_path = (
Path(__file__).parent.parent.parent.parent.parent
/ "src"
/ "virtualenv"
/ "create"
/ "via_global_ref"
/ "_virtualenv.py"
)
if not virtualenv_py_path.exists():
return # Skip if we can't find the file
content = virtualenv_py_path.read_text(encoding="utf-8")
# Verify the fix is present
assert "try:" in content
assert "distutils_patch = _DISTUTILS_PATCH" in content
assert "except NameError:" in content
assert "return None" in content or "return" in content
|
