tests/helpers/helpers.py - master - niceSOFT/python3-hatchling

tests/helpers/helpers.py
184 строки6 KB
Vladimir Petrigo
Fix hardcoded reliance on pip (#1882)
23 июл 2026, 05:00
23 июл 2026, 05:00b1af6e8
from __future__ import annotations

import importlib
import json
import os
import re
import sys
from datetime import datetime, timezone
from functools import lru_cache
from textwrap import dedent as _dedent
from typing import TYPE_CHECKING
from unittest.mock import call

import tomli_w

from hatch.config.user import RootConfig
from hatch.env.utils import add_verbosity_flag
from hatch.python.core import InstalledDistribution
from hatch.python.resolve import get_distribution
from hatch.utils.toml import load_toml_file

if TYPE_CHECKING:
    from hatch.utils.fs import Path


def dedent(text):
    return _dedent(text[1:])


@lru_cache
def tarfile_extraction_compat_options():
    return {"filter": "data"} if sys.version_info >= (3, 12) else {}


def remove_trailing_spaces(text):
    return "".join(f"{line.rstrip()}\n" for line in text.splitlines(True))


def extract_requirements(lines):
    for raw_line in lines:
        line = raw_line.rstrip()
        if line and not line.startswith("#"):
            yield line


def get_current_timestamp():
    return datetime.now(timezone.utc).timestamp()


def assert_plugin_installation(subprocess_run, dependencies: list[str], *, verbosity=0, count=1):
    from uv import find_uv_bin

    uv_bin = find_uv_bin()
    command = [
        uv_bin,
        "pip",
        "install",
        "--disable-pip-version-check",
        "--python",
        sys.executable,
    ]
    add_verbosity_flag(command, verbosity, adjustment=-1)
    command.extend(dependencies)

    assert subprocess_run.call_args_list == [call(command, shell=False)] * count


def assert_files(directory, expected_files, *, check_contents=True):
    start = str(directory)
    expected_relative_files = {str(f.path): f.contents for f in expected_files}
    seen_relative_file_paths = set()

    for root, _, files in os.walk(directory):
        relative_path = os.path.relpath(root, start)

        # First iteration
        if relative_path == ".":
            relative_path = ""

        for file_name in files:
            relative_file_path = os.path.join(relative_path, file_name)
            seen_relative_file_paths.add(relative_file_path)

            if check_contents and relative_file_path in expected_relative_files:
                file_path = os.path.join(start, relative_file_path)
                expected_contents = expected_relative_files[relative_file_path]
                try:
                    with open(file_path, encoding="utf-8") as f:
                        assert f.read() == expected_contents, relative_file_path
                except UnicodeDecodeError:
                    with open(file_path, "rb") as f:
                        assert f.read() == expected_contents, (relative_file_path, expected_contents)
            else:  # no cov
                pass

    expected_relative_file_paths = set(expected_relative_files)

    missing_files = expected_relative_file_paths - seen_relative_file_paths
    assert not missing_files, f"Missing files: {', '.join(sorted(missing_files))}"

    extra_files = seen_relative_file_paths - expected_relative_file_paths
    assert not extra_files, f"Extra files: {', '.join(sorted(extra_files))}"


def assert_output_match(output: str, pattern: str, *, exact: bool = True):
    flags = re.MULTILINE if exact else re.MULTILINE | re.DOTALL
    assert re.search(dedent(pattern), output, flags=flags) is not None, output


def get_template_files(template_name, project_name, **kwargs):
    kwargs["project_name"] = project_name
    kwargs["project_name_normalized"] = project_name.lower().replace(".", "-")
    kwargs["package_name"] = kwargs["project_name_normalized"].replace("-", "_")

    config = RootConfig({})
    kwargs.setdefault("author", config.template.name)
    kwargs.setdefault("email", config.template.email)
    kwargs.setdefault("year", str(datetime.now(timezone.utc).year))

    return __load_template_module(template_name)(**kwargs)


@lru_cache
def __load_template_module(template_name):
    template = importlib.import_module(f"..templates.{template_name}", __name__)
    return template.get_files


def update_project_environment(project, name, config):
    project_file = project.root / "pyproject.toml"
    raw_config = load_toml_file(str(project_file))

    env_config = raw_config.setdefault("tool", {}).setdefault("hatch", {}).setdefault("envs", {}).setdefault(name, {})
    env_config.update(config)

    project.config.envs[name] = project.config.envs.get(name, project.config.envs["default"]).copy()
    project.config.envs[name].update(env_config)

    with open(str(project_file), "w", encoding="utf-8") as f:
        f.write(tomli_w.dumps(raw_config))


def write_distribution(directory: Path, name: str):
    dist = get_distribution(name)
    path = directory / dist.name
    path.ensure_dir_exists()
    python_path = path / dist.python_path
    python_path.parent.ensure_dir_exists()
    python_path.touch()

    metadata = {"source": dist.source, "python_path": dist.python_path}
    metadata_file = path / InstalledDistribution.metadata_filename()
    metadata_file.write_text(json.dumps(metadata))

    return InstalledDistribution(path, dist, metadata)


def downgrade_distribution_metadata(dist_dir: Path):
    metadata_file = dist_dir / InstalledDistribution.metadata_filename()
    metadata = json.loads(metadata_file.read_text())
    dist = InstalledDistribution(dist_dir, get_distribution(dist_dir.name), metadata)

    source = metadata["source"]
    python_path = metadata["python_path"]
    version = dist.version
    new_version = downgrade_version(version)
    new_source = source.replace(version, new_version)
    metadata["source"] = new_source

    # We also modify the Python path because some directory structures are determined
    # by the archive name which is itself determined by the source
    metadata["python_path"] = python_path.replace(version, new_version)
    if python_path != metadata["python_path"]:
        new_python_path = dist_dir / metadata["python_path"]
        new_python_path.parent.ensure_dir_exists()
        (dist_dir / python_path).rename(new_python_path)

    metadata_file.write_text(json.dumps(metadata))
    return metadata


def downgrade_version(version: str) -> str:
    major_version = version.split(".")[0]
    return version.replace(major_version, str(int(major_version) - 1), 1)