aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/pytest/py3/_pytest/tmpdir.py
diff options
context:
space:
mode:
authordeshevoy <deshevoy@yandex-team.ru>2022-02-10 16:46:57 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:46:57 +0300
commit28148f76dbfcc644d96427d41c92f36cbf2fdc6e (patch)
treeb83306b6e37edeea782e9eed673d89286c4fef35 /contrib/python/pytest/py3/_pytest/tmpdir.py
parente988f30484abe5fdeedcc7a5d3c226c01a21800c (diff)
downloadydb-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.py162
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)