93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
|
import contextlib
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import Dict
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
import setuptools._distutils.util
|
||
|
|
|
||
|
|
from cibuildwheel.errors import FatalError
|
||
|
|
from cibuildwheel.util import CIProvider, detect_ci_provider
|
||
|
|
from cibuildwheel.windows import PythonConfiguration, setup_setuptools_cross_compile
|
||
|
|
|
||
|
|
# monkeypatching os.name is too flaky. E.g. It works on my machine, but fails in pipeline
|
||
|
|
if not sys.platform.startswith("win"):
|
||
|
|
pytest.skip("Windows-only tests", allow_module_level=True)
|
||
|
|
|
||
|
|
|
||
|
|
@contextlib.contextmanager
|
||
|
|
def patched_environment(monkeypatch: pytest.MonkeyPatch, environment: Dict[str, str]):
|
||
|
|
with monkeypatch.context() as mp:
|
||
|
|
for envvar, val in environment.items():
|
||
|
|
mp.setenv(name=envvar, value=val)
|
||
|
|
yield
|
||
|
|
|
||
|
|
|
||
|
|
def test_x86(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
|
||
|
|
arch = "32"
|
||
|
|
environment: Dict[str, str] = {}
|
||
|
|
|
||
|
|
configuration = PythonConfiguration("irrelevant", arch, "irrelevant", None)
|
||
|
|
|
||
|
|
setup_setuptools_cross_compile(tmp_path, configuration, tmp_path, environment)
|
||
|
|
with patched_environment(monkeypatch, environment):
|
||
|
|
target_platform = setuptools._distutils.util.get_platform()
|
||
|
|
|
||
|
|
assert environment["VSCMD_ARG_TGT_ARCH"] == "x86"
|
||
|
|
assert target_platform == "win32"
|
||
|
|
|
||
|
|
|
||
|
|
def test_x64(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
|
||
|
|
arch = "64"
|
||
|
|
environment: Dict[str, str] = {}
|
||
|
|
|
||
|
|
configuration = PythonConfiguration("irrelevant", arch, "irrelevant", None)
|
||
|
|
|
||
|
|
setup_setuptools_cross_compile(tmp_path, configuration, tmp_path, environment)
|
||
|
|
with patched_environment(monkeypatch, environment):
|
||
|
|
target_platform = setuptools._distutils.util.get_platform()
|
||
|
|
|
||
|
|
assert environment["VSCMD_ARG_TGT_ARCH"] == "x64"
|
||
|
|
assert target_platform == "win-amd64"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.skipif(
|
||
|
|
detect_ci_provider() == CIProvider.azure_pipelines, reason="arm64 not recognised on azure"
|
||
|
|
)
|
||
|
|
def test_arm(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
|
||
|
|
arch = "ARM64"
|
||
|
|
environment: Dict[str, str] = {}
|
||
|
|
|
||
|
|
configuration = PythonConfiguration("irrelevant", arch, "irrelevant", None)
|
||
|
|
|
||
|
|
setup_setuptools_cross_compile(tmp_path, configuration, tmp_path, environment)
|
||
|
|
with patched_environment(monkeypatch, environment):
|
||
|
|
target_platform = setuptools._distutils.util.get_platform()
|
||
|
|
|
||
|
|
assert environment["VSCMD_ARG_TGT_ARCH"] == "arm64"
|
||
|
|
assert target_platform == "win-arm64"
|
||
|
|
|
||
|
|
|
||
|
|
def test_env_set(tmp_path: Path):
|
||
|
|
arch = "32"
|
||
|
|
environment = {"VSCMD_ARG_TGT_ARCH": "x64"}
|
||
|
|
|
||
|
|
configuration = PythonConfiguration("irrelevant", arch, "irrelevant", None)
|
||
|
|
|
||
|
|
with pytest.raises(FatalError, match="VSCMD_ARG_TGT_ARCH"):
|
||
|
|
setup_setuptools_cross_compile(tmp_path, configuration, tmp_path, environment)
|
||
|
|
|
||
|
|
|
||
|
|
def test_env_blank(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
|
||
|
|
arch = "32"
|
||
|
|
environment = {"VSCMD_ARG_TGT_ARCH": ""}
|
||
|
|
|
||
|
|
configuration = PythonConfiguration("irrelevant", arch, "irrelevant", None)
|
||
|
|
|
||
|
|
setup_setuptools_cross_compile(tmp_path, configuration, tmp_path, environment)
|
||
|
|
with patched_environment(monkeypatch, environment):
|
||
|
|
target_platform = setuptools._distutils.util.get_platform()
|
||
|
|
|
||
|
|
assert environment["VSCMD_ARG_TGT_ARCH"] == "x86"
|
||
|
|
assert target_platform == "win32"
|