mirror of
https://github.com/copier-org/copier.git
synced 2025-05-05 23:42:55 +00:00
c4726b4fe7
This way it's easier to track where each one is created. Most usefil while debugging tests.
38 lines
932 B
Python
38 lines
932 B
Python
import shutil
|
|
from os import SEEK_END
|
|
from pathlib import Path
|
|
from tempfile import mkdtemp
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
import six
|
|
from plumbum import local
|
|
|
|
# Workaround to let plumbum.local and plumbum.cmd run in virtualenv
|
|
local.env.path.insert(0, local.cwd / ".venv" / "bin")
|
|
|
|
|
|
@pytest.fixture()
|
|
def dst(request):
|
|
"""Return a real temporary folder path which is unique to each test
|
|
function invocation. This folder is deleted after the test has finished.
|
|
"""
|
|
dst = mkdtemp(prefix=f"{__name__}.dst.")
|
|
request.addfinalizer(lambda: shutil.rmtree(dst, ignore_errors=True))
|
|
return Path(dst)
|
|
|
|
|
|
class AppendableStringIO(six.StringIO):
|
|
def append(self, text):
|
|
pos = self.tell()
|
|
self.seek(0, SEEK_END)
|
|
self.write(text)
|
|
self.seek(pos)
|
|
|
|
|
|
@pytest.fixture()
|
|
def stdin():
|
|
buffer = AppendableStringIO()
|
|
with mock.patch("sys.stdin", buffer):
|
|
yield buffer
|