Matthieu Darboisandpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
36049d86a2
* feature: add musllinux_1_2 support * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix[test]: clean-up docker images in test_*linux*_only.py We are reaching disk space limits (14 GB) when running tests. Add a fixture to clean-up docker images after tests. This fixture is applied on tests that pull a specific image for one test only in order not to take too much time pulling that image many times. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
from typing import Generator
|
|
|
|
import pytest
|
|
|
|
from cibuildwheel.util import detect_ci_provider
|
|
|
|
from .utils import platform
|
|
|
|
|
|
def pytest_addoption(parser) -> None:
|
|
parser.addoption(
|
|
"--run-emulation", action="store_true", default=False, help="run emulation tests"
|
|
)
|
|
parser.addoption("--run-podman", action="store_true", default=False, help="run podman tests")
|
|
parser.addoption(
|
|
"--run-cp38-universal2",
|
|
action="store_true",
|
|
default=False,
|
|
help="macOS cp38 uses the universal2 installer",
|
|
)
|
|
|
|
|
|
@pytest.fixture(
|
|
params=[{"CIBW_BUILD_FRONTEND": "pip"}, {"CIBW_BUILD_FRONTEND": "build"}], ids=["pip", "build"]
|
|
)
|
|
def build_frontend_env(request) -> dict[str, str]:
|
|
return request.param # type: ignore[no-any-return]
|
|
|
|
|
|
@pytest.fixture()
|
|
def docker_cleanup() -> Generator[None, None, None]:
|
|
def get_images() -> set[str]:
|
|
images = subprocess.run(
|
|
["docker", "image", "ls", "--format", "{{json .ID}}"],
|
|
text=True,
|
|
check=True,
|
|
stdout=subprocess.PIPE,
|
|
).stdout
|
|
return {json.loads(image.strip()) for image in images.splitlines() if image.strip()}
|
|
|
|
if detect_ci_provider() is None or platform != "linux":
|
|
try:
|
|
yield
|
|
finally:
|
|
pass
|
|
return
|
|
images_before = get_images()
|
|
try:
|
|
yield
|
|
finally:
|
|
images_after = get_images()
|
|
for image in images_after - images_before:
|
|
subprocess.run(["docker", "rmi", image], check=False)
|