cookiecutter / cookiecutter

import os

import shutil

import pytest

from cookiecutter import utils

from cookiecutter.config import DEFAULT_CONFIG

USER_CONFIG = """

cookiecutters_dir: '{cookiecutters_dir}'

replay_dir: '{replay_dir}'

"""

# In YAML, double quotes mean to use escape sequences.

# Single quotes mean we will have unescaped backslahes.

# http://blogs.perl.org/users/tinita/2018/03/

# strings-in-yaml---to-quote-or-not-to-quote.html

def backup_dir(original_dir, backup_dir):

    """Generate backup directory based on original directory."""

    # If the default original_dir is pre-existing, move it to a temp location

    if not os.path.isdir(original_dir):

        return False

    # Remove existing backups before backing up. If they exist, they're stale.

    if os.path.isdir(backup_dir):

        utils.rmtree(backup_dir)

    shutil.copytree(original_dir, backup_dir)

    return True

def restore_backup_dir(original_dir, backup_dir, original_dir_found):

    """Restore default contents."""

    # Carefully delete the created original_dir only in certain

    # conditions.

    original_dir_is_dir = os.path.isdir(original_dir)

    if original_dir_found:

        # Delete the created original_dir as long as a backup

        # exists

        if original_dir_is_dir and os.path.isdir(backup_dir):

            utils.rmtree(original_dir)

    else:

        # Delete the created original_dir.

        # There's no backup because it never existed

        if original_dir_is_dir:

            utils.rmtree(original_dir)

    # Restore the user's default original_dir contents

    if os.path.isdir(backup_dir):

        shutil.copytree(backup_dir, original_dir)

    if os.path.isdir(original_dir):

        utils.rmtree(backup_dir)

@pytest.fixture(scope='function')

def clean_system(request):

    """Fixture. Simulates a clean system with no configured or cloned cookiecutters.

    It runs code which can be regarded as setup code as known from a unittest

    TestCase. Additionally it defines a local function referring to values

    which have been stored to local variables in the setup such as the location

    of the cookiecutters on disk. This function is registered as a teardown

    hook with `request.addfinalizer` at the very end of the fixture. Pytest

    runs the named hook as soon as the fixture is out of scope, when the test

    finished to put it another way.

    During setup:

    * Back up the `~/.cookiecutterrc` config file to `~/.cookiecutterrc.backup`

    * Back up the `~/.cookiecutters/` dir to `~/.cookiecutters.backup/`

    * Back up the `~/.cookiecutter_replay/` dir to

      `~/.cookiecutter_replay.backup/`

    * Starts off a test case with no pre-existing `~/.cookiecutterrc` or

      `~/.cookiecutters/` or `~/.cookiecutter_replay/`

    During teardown:

    * Delete `~/.cookiecutters/` only if a backup is present at

      `~/.cookiecutters.backup/`

    * Delete `~/.cookiecutter_replay/` only if a backup is present at

      `~/.cookiecutter_replay.backup/`

    * Restore the `~/.cookiecutterrc` config file from

      `~/.cookiecutterrc.backup`

    * Restore the `~/.cookiecutters/` dir from `~/.cookiecutters.backup/`

    * Restore the `~/.cookiecutter_replay/` dir from

      `~/.cookiecutter_replay.backup/`

    """

    # If ~/.cookiecutterrc is pre-existing, move it to a temp location

    user_config_path = os.path.expanduser('~/.cookiecutterrc')

    user_config_path_backup = os.path.expanduser('~/.cookiecutterrc.backup')

    if os.path.exists(user_config_path):

        user_config_found = True

        shutil.copy(user_config_path, user_config_path_backup)

        os.remove(user_config_path)

    else:

        user_config_found = False

    # If the default cookiecutters_dir is pre-existing, move it to a

    # temp location

    cookiecutters_dir = os.path.expanduser('~/.cookiecutters')

    cookiecutters_dir_backup = os.path.expanduser('~/.cookiecutters.backup')

    cookiecutters_dir_found = backup_dir(cookiecutters_dir, cookiecutters_dir_backup)

    # If the default cookiecutter_replay_dir is pre-existing, move it to a

    # temp location

    cookiecutter_replay_dir = os.path.expanduser('~/.cookiecutter_replay')

    cookiecutter_replay_dir_backup = os.path.expanduser('~/.cookiecutter_replay.backup')

    cookiecutter_replay_dir_found = backup_dir(

        cookiecutter_replay_dir, cookiecutter_replay_dir_backup

    )

    def restore_backup():

        # If it existed, restore ~/.cookiecutterrc

        # We never write to ~/.cookiecutterrc, so this logic is simpler.

        if user_config_found and os.path.exists(user_config_path_backup):

            shutil.copy(user_config_path_backup, user_config_path)

            os.remove(user_config_path_backup)

        # Carefully delete the created ~/.cookiecutters dir only in certain

        # conditions.

        restore_backup_dir(

            cookiecutters_dir, cookiecutters_dir_backup, cookiecutters_dir_found

        )

        # Carefully delete the created ~/.cookiecutter_replay dir only in

        # certain conditions.

        restore_backup_dir(

            cookiecutter_replay_dir,

            cookiecutter_replay_dir_backup,

            cookiecutter_replay_dir_found,

        )

    request.addfinalizer(restore_backup)

@pytest.fixture(scope='session')

def user_dir(tmp_path_factory):

    """Fixture that simulates the user's home directory."""

    return tmp_path_factory.mktemp('user_dir')

@pytest.fixture(scope='session')

def user_config_data(user_dir):

    """Fixture that creates 2 Cookiecutter user config dirs.

     It will create it in the user's home directory.

    * `cookiecutters_dir`

    * `cookiecutter_replay`

    :returns: Dict with name of both user config dirs

    """

    cookiecutters_dir = user_dir.joinpath('cookiecutters')

    cookiecutters_dir.mkdir()

    replay_dir = user_dir.joinpath('cookiecutter_replay')

    replay_dir.mkdir()

    return {

        'cookiecutters_dir': str(cookiecutters_dir),

        'replay_dir': str(replay_dir),

    }

@pytest.fixture(scope='session')

def user_config_file(user_dir, user_config_data):

    """Fixture that creates a config file called `config`.

     It will create it in the user's home directory, with YAML from

     `user_config_data`.

    :param user_dir: Simulated user's home directory

    :param user_config_data: Dict of config values

    :returns: String of path to config file

    """

    config_file = user_dir.joinpath('config')

    config_text = USER_CONFIG.format(**user_config_data)

    config_file.write_text(config_text)

    return str(config_file)

@pytest.fixture

def output_dir(tmp_path):

    """Fixture to prepare test output directory."""

    output_path = tmp_path.joinpath("output")

    output_path.mkdir()

    return str(output_path)

@pytest.fixture

def clone_dir(tmp_path):

    """Simulate creation of a directory called `clone_dir` inside of `tmp_path`. \

    Returns a str to said directory."""

    clone_dir = tmp_path.joinpath("clone_dir")

    clone_dir.mkdir()

    return clone_dir