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
|
from __future__ import annotations
import shutil
import subprocess
from pathlib import Path
import pytest
from flaky import flaky
from virtualenv.discovery.py_info import PythonInfo
from virtualenv.run import cli_run
HERE = Path(__file__).parent
CURRENT = PythonInfo.current_system()
@pytest.fixture(scope="session")
def zipapp_build_env(tmp_path_factory):
create_env_path = None
if CURRENT.implementation != "PyPy":
exe = CURRENT.executable # guaranteed to contain a recent enough pip (tox.ini)
else:
create_env_path = tmp_path_factory.mktemp("zipapp-create-env")
exe, found = None, False
# prefer CPython as builder as pypy is slow
for impl in ["cpython", ""]:
for version in range(11, 6, -1):
try:
# create a virtual environment which is also guaranteed to contain a recent enough pip (bundled)
session = cli_run(
[
"-vvv",
"-p",
f"{impl}3.{version}",
"--activators",
"",
str(create_env_path),
"--no-download",
"--no-periodic-update",
],
)
exe = str(session.creator.exe)
found = True
break
except Exception:
pass
if found:
break
else:
raise RuntimeError("could not find a python to build zipapp")
cmd = [str(Path(exe).parent / "pip"), "install", "pip>=23", "packaging>=23"]
subprocess.check_call(cmd)
yield exe
if create_env_path is not None:
shutil.rmtree(str(create_env_path))
@pytest.fixture(scope="session")
def zipapp(zipapp_build_env, tmp_path_factory):
into = tmp_path_factory.mktemp("zipapp")
path = HERE.parent.parent / "tasks" / "make_zipapp.py"
filename = into / "virtualenv.pyz"
cmd = [zipapp_build_env, str(path), "--dest", str(filename)]
subprocess.check_call(cmd)
yield filename
shutil.rmtree(str(into))
@pytest.fixture(scope="session")
def zipapp_test_env(tmp_path_factory):
base_path = tmp_path_factory.mktemp("zipapp-test")
session = cli_run(["-v", "--activators", "", "--without-pip", str(base_path / "env"), "--no-periodic-update"])
yield session.creator.exe
shutil.rmtree(str(base_path))
@pytest.fixture()
def call_zipapp(zipapp, tmp_path, zipapp_test_env, temp_app_data): # noqa: U100
def _run(*args):
cmd = [str(zipapp_test_env), str(zipapp), "-vv", str(tmp_path / "env")] + list(args)
subprocess.check_call(cmd)
return _run
@flaky(max_runs=2, min_passes=1)
def test_zipapp_help(call_zipapp, capsys):
call_zipapp("-h")
out, err = capsys.readouterr()
assert not err
@pytest.mark.parametrize("seeder", ["app-data", "pip"])
def test_zipapp_create(call_zipapp, seeder):
call_zipapp("--seeder", seeder)
|
