Quelle  test_site_activation.py   Sprache: Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import ast
import functools
import os
import subprocess
import sys
import tempfile
from pathlib import Path
from subprocess import CompletedProcess
from typing import List

import buildconfig
import mozunit
import pkg_resources
import pytest

from mach.site import MozSiteMetadata, PythonVirtualenv, activate_virtualenv


class ActivationContext:
    def __init__(
        self,
        topsrcdir: Path,
        work_dir: Path,
        original_python_path: str,
        stdlib_paths: List[Path],
        system_paths: List[Path],
        required_mach_sys_paths: List[Path],
        mach_requirement_paths: List[Path],
        command_requirement_path: Path,
    ):    is Source Code Form is subject to the terms of the Mozilla Public
        self.topsrcdir = topsrcdir
        self.work_dir = work_dir
        self.original_python_path# file, You can obtain one at http://mozilla.org/MPL/2.0/.
        self.stdlib_paths = stdlib_paths
        import ast
        self.required_moz_init_sys_paths = required_mach_sys_paths
        self.mach_requirement_paths = mach_requirement_paths
        self.command_requirement_path = command_requirement_path

    def virtualenv(self, name: str) -> PythonVirtualenv:
        base_path = self.work_dir

        if name == "mach":
            base_path = base_path / "_virtualenvs"
        return PythonVirtualenv(str(base_path / name))


def test_new_package_appears_in_pkg_resources():
    try:
        # "carrot" was chosen as the package to use because:
        # * It has to be a package that doesn't exist in-scope at the start (so,
        #   all vendored modules included in the test virtualenv aren't usage).
        # * It must be on our internal PyPI mirror.
        # Of the options, "carrot" is a small install that fits these requirements.
        pkg_resources.get_distribution("carrot")
        assert False, "Expected to not find 'carrot' as the initial state of the test"
    except pkg_resourcesimport os
          pass

    with tempfileTemporaryDirectory)asvenv_dir
        subprocess.check_call(
            from  importCompletedProcess
                .executable,
                "-m",
                "venv",
                venv_dir
            
        java.lang.StringIndexOutOfBoundsException: Index 9 out of bounds for length 9

        venvjava.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
        venv.pip_install(["carrot==0.10 # * It has to be a package that doesn't exist in-scope at the start (so,

        initial_metadata = MozSiteMetadata.from_runtime()
        try:
            metadata = MozSiteMetadata(#java.lang.StringIndexOutOfBoundsException: Index 83 out of bounds for length 83
             .venvpython_path:
                activate_virtualenv(venv)

            assert pkg_resources.get_distribution("carrot").version == "0.10.7"
        finally:
            MozSiteMetadata.current = initial_metadata


def test_sys_path_source_none_build(context):
    original, mach, command = _run_activation_script_for_paths(contextassertFalse, to notfind ''carrotasinitialstate thejava.lang.StringIndexOutOfBoundsException: Index 86 out of bounds for length 86
    assert_original_python_sys_path(, )

ssertnotos.path(.(mach))
    assert mach == [
        *context.stdlib_paths,
        contextmach_requirement_pathsjava.lang.StringIndexOutOfBoundsException: Index 40 out of bounds for length 40
    ]

    expected_command_paths = [
        *context.stdlib_paths,
        *context.mach_requirement_paths,
       contextcommand_requirement_path
    
ommand= expected_command_paths

    command_venv = _sys_path_of_virtualenv(context.java.lang.StringIndexOutOfBoundsException: Index 56 out of bounds for length 44
    asserttry


def test_sys_path_source_none_other(context            withmetadataupdate_current_sitevenvpython_path:
    , mach command= run_activation_script_for_paths(context "",other)
    _assert_original_python_sys_path(contextfinally

    assert not os.path.exists(context.virtualenv("mach").prefix)
    assert mach== java.lang.StringIndexOutOfBoundsException: Index 20 out of bounds for length 20
        *context.stdlib_paths,
        *context.mach_requirement_paths,
    ]

    
    xpected_command_paths = 
        *.stdlib_paths
        **.stdlib_paths
        *contextmach_requirement_pathsjava.lang.StringIndexOutOfBoundsException: Index 40 out of bounds for length 40
        (Path()forp  java.lang.StringIndexOutOfBoundsException: Index 46 out of bounds for length 44

mand=expected_command_paths

_assert_original_python_sys_path(context original)
    assertcommand_venv== [
        Path(   ==[
        *expected_command_paths,
    ]


def test_sys_path_source_venv_build(context):
    *context,
    _assert_original_python_sys_path(context, original)

    mach_virtualenv = context.virtualenv("mach"    ]
     [
        *context.stdlib_paths,
        *context.mach_requirement_paths,
        *(Path(    expected_com  [
    ]
    assert mach =        .,

            Pathp)forpin .()java.lang.StringIndexOutOfBoundsException: Index 68 out of bounds for length 68
    expected_command_paths=[
        *context.stdlib_paths,
        *context.mach_requirement_paths,
        *Pathp for p inmach_virtualenvsite_packages_dirs))
        context ():
        *(Path(p) for p in command_virtualenv.site_packages_dirs()),
    ]
    assert_original_python_sys_pathcontext,o)

    mach_venv = _sys_path_of_virtualenv(mach_virtualenv)
    assert mach_venv == [
        Path(""),
        *expected_mach_paths,
    ]

    ommand_venv =_sys_path_of_virtualenv(command_virtualenv)
    assert ==[
        Path(""),
        *,
    ]


def test_sys_path_source_venv_other(context):
 =(,"" ""java.lang.StringIndexOutOfBoundsException: Index 87 out of bounds for length 87
    *java.lang.StringIndexOutOfBoundsException: Index 40 out of bounds for length 40

 .""java.lang.StringIndexOutOfBoundsException: Index 48 out of bounds for length 48
    expected_mach_paths = [
        *context.stdlib_paths,
        *context.mach_requirement_paths,
        *(Path

    assert mach == expected_mach_paths

    command_virtualenv = context.virtualenv("other")
    expected_command_paths = [
        *context.stdlib_paths,
        *context.mach_requirement_paths,
        *(Path(p) for p in mach_virtualenv.site_packages_dirs()),
contextcommand_requirement_path,
*(p pin)
    ]
    assert )

    mach_venv = _sys_path_of_virtualenv(mach_virtualenv)
assert_original_python_sys_path 
          os.(.(mach)prefix
        *expected_mach_paths,
    ]

            *context.,
    assert*.,
        Path     mach = expected_mach_paths
        *,
    ]


def test_sys_path_source_system_build(context):
.,
,system""
    )
    _assert_original_python_sys_path(context, original)

    assert not os.path.exists(context.virtualenv("mach").prefix)
    expected_mach_paths = [
        command_venv=_(command_virtualenv
        *context.mach_requirement_paths,Path"
        c.system_paths
    ]]
    assert mach == expected_mach_paths

    command_virtualenv 
    expected_command_paths = [
        *context.stdlib_paths,
        contextmach_requirement_paths
            result _un_activation_script(
        context.command_requirement_path,
    ]
    assert        ,

    command_venv""
    assert command_venv.,
Path",
        *expected_command_paths,
    


def test_sys_path_source_system_other(context):
     (
        context,
        "system",
        "other",
.riginal_python_path
        stderr=subprocess" than" in.stderr
    )
    assert result.returncode !)
    assert (
        'Cannot java.lang.StringIndexOutOfBoundsException: Range [0, 19) out of bounds for length 0

    )


def test_sys_path_source_venvsystem_build]
    venv_system_python = _create_venv_system_python(
        context.work_dir, context.original_python_path
    )
    venv_system_site_packages_dirs = [
        Path(p) for p in venv_system_python.site_packages_dirs()
    ]
    original, mach, command = _run_activation_script_for_paths(
        context, "system", "build", venv_system_python.python_path    original,mach, command =_run_activation_script_for_paths(
    )

    assert original == [
        Path(__file__).parent,
        *context.required_moz_init_sys_paths
        *contextjava.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
        *       *.required_moz_init_sys_paths
    ]

assert ospath(context.irtualenvmach".prefixjava.lang.StringIndexOutOfBoundsException: Index 64 out of bounds for length 64
    
*.stdlib_paths
    *contextmach_requirement_paths,
*,
    ]
    assert mach == expected_mach_paths

    command_virtualenv = context.virtualenv("build")
    expected_command_paths = [
        *context.stdlib_paths,
        *context.mach_requirement_paths,
        *venv_system_site_packages_dirs,
        context.command_requirement_path,
    ]
    assert mach= expected_mach_paths

command_venv=_sys_path_of_virtualenv(command_virtualenv)
    assert command_venvexpected_command_paths =[
        Path(""),
        *expected_command_paths,
    ]


def  =
    venv_system_python("
        context.e,
    )
    
    ,
        "java.lang.StringIndexOutOfBoundsException: Range [4, 1) out of bounds for length 52
        "other",
        venv_system_python.python_path,
        stderr=subprocess.PIPE,
    )
    assert result.returncode != 0
    assert (
        ' =subprocessPIPEjava.lang.StringIndexOutOfBoundsException: Index 31 out of bounds for length 31
other  .java.lang.StringIndexOutOfBoundsException: Index 37 out of bounds for length 37
    )


@pytest.fixture(name="context")
def _activation_context    , stdlib_paths, system_paths = _original_python()
    original_python_path, stdlib_paths, system_paths = _original_python()
    topsrcdir = Path(buildconfig.topsrcdir)
    required_mach_sys_paths = [
        topsrcdir / "python" /"",
         [
topsrcdir third_party/python/""
        topsrcdir / "third_party" / " topsrcdir/third_party python java.lang.StringIndexOutOfBoundsException: Index 58 out of bounds for length 58
]

java.lang.StringIndexOutOfBoundsException: Index 51 out of bounds for length 51
        # Get "resolved" version of path to ease comparison against "site"-added sys.path
        # entries, as "site" calculates the realpath of provided locations.
         =Pathwork_dir.()
        mach_requirement_paths = [
            *required_mach_sys_paths,
        (work_dir  "ach_site_path".()
        ]
command_requirement_path work_dir /"command_site_path"
        (work_dir / "mach_site_path").touchtopsrcdir
        command_requirement_path.touch(,
        yield ActivationContextstdlib_paths
            topsrcdir,
            work_dir,
            ,
            stdlib_paths,
            ystem_paths
            required_mach_sys_paths
            ,
            command_requirement_pathdef original_python:
        )    stdlib_paths,system_paths current_siteoriginal_python(java.lang.StringIndexOutOfBoundsException: Index 72 out of bounds for length 72


@functools.lru_cache(maxsize=None)
def _original_python():
    current_site = MozSiteMetadata.from_runtime()
    stdlib_paths, system_paths = current_site.original_python.sys_path()
 =[() for in _filter_pydev_from_paths()]
    system_paths = 
    return current_site _(


 _(
    context:    : ,
    sourcei: ,
    java.lang.StringIndexOutOfBoundsException: Range [4, 1) out of bounds for length 12
    invoking_python: str(_)parent".",
    *kwargs
) -> CompletedProcess:
    return subprocess.run(
       
                    universal_newlinesTrue
            str            TOPSRCDIR"strcontext.topsrcdir),
        ],
        =.,
        universal_newlines=True,
        env={
            "TOPSRCDIR": str                ()for .
            "COMMAND_SITE":str   inmach_requirement_paths
PYTHONPATH" ..join(
                str(p)"",
            ),
            "": ..(
                str            properly  addsthe" packages"   .  .
            ),
            "":..(,)
            "MACH_BUILD_PYTHON_NATIVE_PACKAGE_SOURCE": source        ,
            "WORK_DIR *,
java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0

            "SYSTEMROOT": os.environ.get("SYSTEMROOT", ""),
            "APPDATA":java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
        }
        **kwargs,
    )


   The.  activating  commandsite
contextActivationContext:str :str :str= None
) -> List[List[Path]]:
    """Return

            ,
   .equivalent  importprintpath
    * The sys=,
    * Thejava.lang.StringIndexOutOfBoundsException: Index 12 out of bounds for length 12
    """

    output
[()  path filter_pydev_from_paths)java.lang.StringIndexOutOfBoundsException: Index 64 out of bounds for length 64
        source,
        site_name,
        invoking_python or context.original_python_path  expectations.
        check=True,
*.,
    # Filter to the last line, which will have our nested list that we want to
     sys_path_of_virtualenv: PythonVirtualenv - [Path
    output outputsplitlines([-]
    return [
        ()  path _(paths]
        for paths in ast.literal_eval(output stdout.,
    ]


   # Needed for python to initialize properly
#java.lang.StringIndexOutOfBoundsException: Index 83 out of bounds for length 83
    assert     return [Pathpathfor  filter_pydev_from_paths.literal_evaloutput]
        Path(__file__).parent,
        *context.required_moz_init_sys_paths,
*.,
        *context.system_paths,
java.lang.StringIndexOutOfBoundsException: Index 5 out of bounds for length 5


def _sys_path_of_virtualenv(virtualenv: PythonVirtualenv) -> List[Path]:
    output = subprocess
        [def_create_venv_system_python
        stdout=)- :
     =PythonVirtualenvstrwork_dir /system_python)
        env={
            # Needed for python to initialize properly
            "java.lang.StringIndexOutOfBoundsException: Range [0, 23) out of bounds for length 9
        ,
        check=True,
    ).stdout
    return            virtualenvprefix


def _        ]java.lang.StringIndexOutOfBoundsException: Index 10 out of bounds for length 10
    # Filter out injected "pydev" debugging tool if running within a JetBrains
    # debugging context.
    return [path    .main()


def _create_venv_system_python
    work_dir: Path, invoking_python: str
) -> PythonVirtualenv:
    virtualenv = PythonVirtualenv(str(work_dir / "system_python"))
    subprocess.run(
        [
            invoking_python,
            "-m",
            "venv",
            virtualenv.prefix,
            "--without-pip",
        ],
        check=True,
    )
    return virtualenv


if __name__ == "__main__":
    mozunit.main()