michaeljoseph / cookiecutter

    mock_ensure_failure.assert_called_once_with(replay_test_dir)

def test_run_json_dump(mocker, mock_ensure_success, mock_user_config,

                       template_name, context, replay_test_dir, replay_file):

    """Test that replay.dump runs json.dump under the hood and that the context

    is correctly written to the expected file in the replay_dir.

    """

    spy_get_replay_file = mocker.spy(replay, 'get_file_name')

    mock_json_dump = mocker.patch('json.dump', side_effect=json.dump)

    replay.dump(replay_test_dir, template_name, context)

    assert not mock_user_config.called

    mock_ensure_success.assert_called_once_with(replay_test_dir)

    spy_get_replay_file.assert_called_once_with(replay_test_dir, template_name)

    assert mock_json_dump.call_count == 1

    (dumped_context, outfile_handler), kwargs = mock_json_dump.call_args

    assert outfile_handler.name == replay_file

    assert dumped_context == context

        replay.load(replay_test_dir, 'no_replay')

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