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