mirror of
https://github.com/copier-org/copier.git
synced 2025-05-31 23:25:15 +00:00
33 lines
707 B
Python
33 lines
707 B
Python
from os import SEEK_END
|
|
from tempfile import mkdtemp
|
|
import six
|
|
import shutil
|
|
|
|
import pytest
|
|
from unittest import mock
|
|
|
|
|
|
@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()
|
|
request.addfinalizer(lambda: shutil.rmtree(dst))
|
|
return 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
|