def test_run_json_load(mocker, mock_user_config, template_name, context, replay_test_dir, replay_file): """Test that replay.load runs json.load under the hood and that the context is correctly loaded from the file in replay_dir. """ spy_get_replay_file = mocker.spy(replay, 'get_file_name') mock_json_load = mocker.patch('json.load', side_effect=json.load) loaded_context = replay.load(replay_test_dir, template_name) assert not mock_user_config.called spy_get_replay_file.assert_called_once_with(replay_test_dir, template_name) assert mock_json_load.call_count == 1 (infile_handler,), kwargs = mock_json_load.call_args assert infile_handler.name == replay_file assert loaded_context == context
def test_io_error_if_no_replay_file(mocker, replay_test_dir): """Test that replay.load raises if it cannot find a replay file.""" with pytest.raises(IOError): replay.load(replay_test_dir, 'no_replay')
def test_type_error_if_no_template_name(replay_test_dir): """Test that replay.load raises if the tempate_name is not a valid str.""" with pytest.raises(TypeError): replay.load(replay_test_dir, None)
def test_value_error_if_key_missing_in_context(mocker, replay_test_dir): """Test that replay.load raises if the loaded context does not contain 'cookiecutter'. """ with pytest.raises(ValueError): replay.load(replay_test_dir, 'invalid_replay')
def test_io_error_if_no_replay_file(mocker, mock_user_config): """Test that replay.load raises if it cannot find a replay file.""" with pytest.raises(IOError): replay.load('no_replay')
def test_value_error_if_key_missing_in_context(mocker): """Test that replay.load raises if the loaded context does not contain 'cookiecutter'. """ with pytest.raises(ValueError): replay.load('invalid_replay')
def test_type_error_if_no_template_name(): """Test that replay.load raises if the tempate_name is not a valid str.""" with pytest.raises(TypeError): replay.load(None)
def test_io_error_if_no_replay_file(mocker, replay_test_dir): """Test that replay.load raises if it cannot find a replay file.""" with pytest.raises(IOError): replay.load(replay_test_dir, 'no_replay', 'cookiecutter')
def cookiecutter( template, checkout=None, no_input=False, extra_context=None, replay=None, overwrite_if_exists=False, output_dir='.', config_file=None, default_config=False, password=None, directory=None, skip_if_file_exists=False, accept_hooks=True, ): """ Run Cookiecutter just as if using it from the command line. :param template: A directory containing a project template directory, or a URL to a git repository. :param checkout: The branch, tag or commit ID to checkout after clone. :param no_input: Prompt the user at command line for manual configuration? :param extra_context: A dictionary of context that overrides default and user configuration. :param replay: Do not prompt for input, instead read from saved json. If ``True`` read from the ``replay_dir``. if it exists :param output_dir: Where to output the generated project dir into. :param config_file: User configuration file path. :param default_config: Use default values rather than a config file. :param password: The password to use when extracting the repository. :param directory: Relative path to a cookiecutter template in a repository. :param accept_hooks: Accept pre and post hooks if set to `True`. """ if replay and ((no_input is not False) or (extra_context is not None)): err_msg = ( "You can not use both replay and no_input or extra_context " "at the same time." ) raise InvalidModeException(err_msg) config_dict = get_user_config( config_file=config_file, default_config=default_config, ) repo_dir, cleanup = determine_repo_dir( template=template, abbreviations=config_dict['abbreviations'], clone_to_dir=config_dict['cookiecutters_dir'], checkout=checkout, no_input=no_input, password=password, directory=directory, ) template_name = os.path.basename(os.path.abspath(repo_dir)) if replay: if isinstance(replay, bool): context = load(config_dict['replay_dir'], template_name) else: path, template_name = os.path.split(os.path.splitext(replay)[0]) context = load(path, template_name) else: context_file = os.path.join(repo_dir, 'cookiecutter.json') logger.debug('context_file is %s', context_file) context = generate_context( context_file=context_file, default_context=config_dict['default_context'], extra_context=extra_context, ) # prompt the user to manually configure at the command line. # except when 'no-input' flag is set context['cookiecutter'] = prompt_for_config(context, no_input) # include template dir or url in the context dict context['cookiecutter']['_template'] = template # include output+dir in the context dict context['cookiecutter']['_output_dir'] = os.path.abspath(output_dir) dump(config_dict['replay_dir'], template_name, context) # Create project from local context and project template. result = generate_files( repo_dir=repo_dir, context=context, overwrite_if_exists=overwrite_if_exists, skip_if_file_exists=skip_if_file_exists, output_dir=output_dir, accept_hooks=accept_hooks, ) # Cleanup (if required) if cleanup: rmtree(repo_dir) return result
def get_replay(self): """Retrieve dict of user input values.""" if self.template_name: return replay.load(tempfile.gettempdir(), self.template_name)