diff options
author | deshevoy <deshevoy@yandex-team.ru> | 2022-02-10 16:46:57 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:46:57 +0300 |
commit | 28148f76dbfcc644d96427d41c92f36cbf2fdc6e (patch) | |
tree | b83306b6e37edeea782e9eed673d89286c4fef35 /contrib/python/pytest/py3/_pytest/tmpdir.py | |
parent | e988f30484abe5fdeedcc7a5d3c226c01a21800c (diff) | |
download | ydb-28148f76dbfcc644d96427d41c92f36cbf2fdc6e.tar.gz |
Restoring authorship annotation for <deshevoy@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/python/pytest/py3/_pytest/tmpdir.py')
-rw-r--r-- | contrib/python/pytest/py3/_pytest/tmpdir.py | 162 |
1 files changed, 81 insertions, 81 deletions
diff --git a/contrib/python/pytest/py3/_pytest/tmpdir.py b/contrib/python/pytest/py3/_pytest/tmpdir.py index 679e5d0cfd..a6bd383a9c 100644 --- a/contrib/python/pytest/py3/_pytest/tmpdir.py +++ b/contrib/python/pytest/py3/_pytest/tmpdir.py @@ -1,38 +1,38 @@ """Support for providing temporary directories to test functions.""" -import os -import re +import os +import re import sys -import tempfile +import tempfile from pathlib import Path from typing import Optional - -import attr -import py - -from .pathlib import LOCK_TIMEOUT -from .pathlib import make_numbered_dir -from .pathlib import make_numbered_dir_with_cleanup + +import attr +import py + +from .pathlib import LOCK_TIMEOUT +from .pathlib import make_numbered_dir +from .pathlib import make_numbered_dir_with_cleanup from .pathlib import rm_rf from _pytest.compat import final from _pytest.config import Config from _pytest.deprecated import check_ispytest from _pytest.fixtures import fixture from _pytest.fixtures import FixtureRequest -from _pytest.monkeypatch import MonkeyPatch - - +from _pytest.monkeypatch import MonkeyPatch + + @final @attr.s(init=False) class TempPathFactory: - """Factory for temporary directories under the common base temp directory. - + """Factory for temporary directories under the common base temp directory. + The base directory can be configured using the ``--basetemp`` option. """ - + _given_basetemp = attr.ib(type=Optional[Path]) - _trace = attr.ib() + _trace = attr.ib() _basetemp = attr.ib(type=Optional[Path]) - + def __init__( self, given_basetemp: Optional[Path], @@ -52,21 +52,21 @@ class TempPathFactory: self._trace = trace self._basetemp = basetemp - @classmethod + @classmethod def from_config( cls, config: Config, *, _ispytest: bool = False, ) -> "TempPathFactory": """Create a factory according to pytest configuration. :meta private: - """ + """ check_ispytest(_ispytest) - return cls( + return cls( given_basetemp=config.option.basetemp, trace=config.trace.get("tmpdir"), _ispytest=True, - ) - + ) + def _ensure_relative_to_basetemp(self, basename: str) -> str: basename = os.path.normpath(basename) if (self.getbasetemp() / basename).resolve().parent != self.getbasetemp(): @@ -89,19 +89,19 @@ class TempPathFactory: The path to the new directory. """ basename = self._ensure_relative_to_basetemp(basename) - if not numbered: - p = self.getbasetemp().joinpath(basename) + if not numbered: + p = self.getbasetemp().joinpath(basename) p.mkdir(mode=0o700) - else: + else: p = make_numbered_dir(root=self.getbasetemp(), prefix=basename, mode=0o700) - self._trace("mktemp", p) - return p - + self._trace("mktemp", p) + return p + def getbasetemp(self) -> Path: """Return the base temporary directory, creating it if needed.""" if self._basetemp is not None: - return self._basetemp - + return self._basetemp + if self._given_basetemp is not None: basetemp = self._given_basetemp if basetemp.exists(): @@ -146,16 +146,16 @@ class TempPathFactory: self._basetemp = basetemp self._trace("new basetemp", basetemp) return basetemp - + @final @attr.s(init=False) class TempdirFactory: """Backward comptibility wrapper that implements :class:``py.path.local`` for :class:``TempPathFactory``.""" - + _tmppath_factory = attr.ib(type=TempPathFactory) - + def __init__( self, tmppath_factory: TempPathFactory, *, _ispytest: bool = False ) -> None: @@ -164,67 +164,67 @@ class TempdirFactory: def mktemp(self, basename: str, numbered: bool = True) -> py.path.local: """Same as :meth:`TempPathFactory.mktemp`, but returns a ``py.path.local`` object.""" - return py.path.local(self._tmppath_factory.mktemp(basename, numbered).resolve()) - + return py.path.local(self._tmppath_factory.mktemp(basename, numbered).resolve()) + def getbasetemp(self) -> py.path.local: """Backward compat wrapper for ``_tmppath_factory.getbasetemp``.""" - return py.path.local(self._tmppath_factory.getbasetemp().resolve()) - - + return py.path.local(self._tmppath_factory.getbasetemp().resolve()) + + def get_user() -> Optional[str]: - """Return the current user name, or None if getuser() does not work + """Return the current user name, or None if getuser() does not work in the current environment (see #1010).""" - import getpass - - try: - return getpass.getuser() - except (ImportError, KeyError): - return None - - + import getpass + + try: + return getpass.getuser() + except (ImportError, KeyError): + return None + + def pytest_configure(config: Config) -> None: - """Create a TempdirFactory and attach it to the config object. - - This is to comply with existing plugins which expect the handler to be - available at pytest_configure time, but ideally should be moved entirely - to the tmpdir_factory session fixture. - """ - mp = MonkeyPatch() + """Create a TempdirFactory and attach it to the config object. + + This is to comply with existing plugins which expect the handler to be + available at pytest_configure time, but ideally should be moved entirely + to the tmpdir_factory session fixture. + """ + mp = MonkeyPatch() tmppath_handler = TempPathFactory.from_config(config, _ispytest=True) t = TempdirFactory(tmppath_handler, _ispytest=True) - config._cleanup.append(mp.undo) - mp.setattr(config, "_tmp_path_factory", tmppath_handler, raising=False) - mp.setattr(config, "_tmpdirhandler", t, raising=False) - - + config._cleanup.append(mp.undo) + mp.setattr(config, "_tmp_path_factory", tmppath_handler, raising=False) + mp.setattr(config, "_tmpdirhandler", t, raising=False) + + @fixture(scope="session") def tmpdir_factory(request: FixtureRequest) -> TempdirFactory: """Return a :class:`_pytest.tmpdir.TempdirFactory` instance for the test session.""" # Set dynamically by pytest_configure() above. return request.config._tmpdirhandler # type: ignore - - + + @fixture(scope="session") def tmp_path_factory(request: FixtureRequest) -> TempPathFactory: """Return a :class:`_pytest.tmpdir.TempPathFactory` instance for the test session.""" # Set dynamically by pytest_configure() above. return request.config._tmp_path_factory # type: ignore - - + + def _mk_tmp(request: FixtureRequest, factory: TempPathFactory) -> Path: - name = request.node.name - name = re.sub(r"[\W]", "_", name) - MAXVAL = 30 - name = name[:MAXVAL] - return factory.mktemp(name, numbered=True) - - + name = request.node.name + name = re.sub(r"[\W]", "_", name) + MAXVAL = 30 + name = name[:MAXVAL] + return factory.mktemp(name, numbered=True) + + @fixture def tmpdir(tmp_path: Path) -> py.path.local: """Return a temporary directory path object which is unique to each test function invocation, created as a sub directory of the base temporary directory. - + By default, a new base temporary directory is created each test session, and old bases are removed after 3 sessions, to aid in debugging. If ``--basetemp`` is used then it is cleared each session. See :ref:`base @@ -232,23 +232,23 @@ def tmpdir(tmp_path: Path) -> py.path.local: The returned object is a `py.path.local`_ path object. - .. _`py.path.local`: https://py.readthedocs.io/en/latest/path.html - """ + .. _`py.path.local`: https://py.readthedocs.io/en/latest/path.html + """ return py.path.local(tmp_path) - - + + @fixture def tmp_path(request: FixtureRequest, tmp_path_factory: TempPathFactory) -> Path: """Return a temporary directory path object which is unique to each test function invocation, created as a sub directory of the base temporary directory. - + By default, a new base temporary directory is created each test session, and old bases are removed after 3 sessions, to aid in debugging. If ``--basetemp`` is used then it is cleared each session. See :ref:`base temporary directory`. - + The returned object is a :class:`pathlib.Path` object. - """ - - return _mk_tmp(request, tmp_path_factory) + """ + + return _mk_tmp(request, tmp_path_factory) |