notifications-admin/tests/app/main/views/test_template_folders.py

import uuid

import pytest
from flask import url_for
from notifications_python_client.errors import HTTPError

from tests.conftest import (
    SERVICE_ONE_ID,
    TEMPLATE_ONE_ID,
    active_caseworking_user,
    active_user_view_permissions,
    active_user_with_permissions,
    normalize_spaces,
)

ROOT_FOLDER_ID = '__NONE__'
PARENT_FOLDER_ID = '7e979e79-d970-43a5-ac69-b625a8d147b0'
CHILD_FOLDER_ID = '92ee1ee0-e4ee-4dcc-b1a7-a5da9ebcfa2b'
GRANDCHILD_FOLDER_ID = 'fafe723f-1d39-4a10-865f-e551e03d8886'
FOLDER_TWO_ID = 'bbbb222b-2b22-2b22-222b-b222b22b2222'


def _folder(name, folder_id=None, parent=None):
    return {
        'name': name,
        'id': folder_id or str(uuid.uuid4()),
        'parent_id': parent,
    }


def _template(template_type, name, parent=None, template_id=None):
    return {
        'id': template_id or str(uuid.uuid4()),
        'name': name,
        'template_type': template_type,
        'folder': parent,
    }


@pytest.mark.parametrize('parent_folder_id', [None, PARENT_FOLDER_ID])
def test_add_page_shows_option_for_folder(
    client_request,
    service_one,
    parent_folder_id,
    mocker,
    mock_get_service_templates,
    mock_get_organisations_and_services_for_user,
):
    service_one['permissions'] += ['edit_folders']
    mocker.patch('app.service_api_client.get_service', return_value={"data": service_one})

    page = client_request.get(
        'main.add_template_by_type',
        service_id=service_one['id'],
        template_folder_id=parent_folder_id,
        _test_page_title=False
    )

    radios = page.select('input[type=radio]')
    labels = page.select('label')

    assert [x['value'] for x in radios] == ['email', 'sms', 'copy-existing', 'folder']
    assert [x.text.strip() for x in labels] == [
        'Email template',
        'Text message template',
        'Copy of an existing template',
        'Folder'
    ]


@pytest.mark.parametrize('parent_folder_id', [None, PARENT_FOLDER_ID])
def test_get_add_template_folder_page(client_request, service_one, parent_folder_id):
    service_one['permissions'] += ['edit_folders']

    page = client_request.get(
        'main.add_template_folder',
        service_id=service_one['id'],
        template_folder_id=parent_folder_id
    )

    assert page.select_one('input[name=name]') is not None


def test_add_template_folder_page_rejects_if_service_doesnt_have_permission(client_request, service_one):
    client_request.get('main.add_template_folder', service_id=service_one['id'], _expected_status=403)
    client_request.post('main.add_template_folder', service_id=service_one['id'], _expected_status=403)


@pytest.mark.parametrize('parent_folder_id', [None, PARENT_FOLDER_ID])
def test_post_add_template_folder_page(client_request, service_one, mocker, parent_folder_id):
    mock_create = mocker.patch('app.template_folder_api_client.create_template_folder')

    service_one['permissions'] += ['edit_folders']

    client_request.post(
        'main.add_template_folder',
        service_id=service_one['id'],
        template_folder_id=parent_folder_id,
        _data={'name': 'foo'},
        _expected_redirect=url_for(
            'main.choose_template',
            service_id=service_one['id'],
            template_folder_id=parent_folder_id,
            _external=True,
        )
    )

    mock_create.assert_called_once_with(service_one['id'], name='foo', parent_id=parent_folder_id)


@pytest.mark.parametrize(
    (
        'expected_title_tag,'
        'expected_page_title,'
        'expected_parent_link_args,'
        'extra_args,'
        'expected_nav_links,'
        'expected_items, '
        'expected_displayed_items, '
        'expected_searchable_text, '
        'expected_empty_message '
    ),
    [
        (
            'Templates – service one – GOV.UK Notify',
            'Templates',
            [],
            {},
            ['Text message', 'Email', 'Letter'],
            [
                'folder_one 2 folders',
                'folder_one / folder_one_one 1 template, 1 folder',
                'folder_one / folder_one_one / folder_one_one_one 1 template',
                'folder_one / folder_one_one / folder_one_one_one / sms_template_nested Text message template',
                'folder_one / folder_one_one / letter_template_nested Letter template',
                'folder_one / folder_one_two Empty',
                'folder_two Empty',
                'sms_template_one Text message template',
                'sms_template_two Text message template',
                'email_template_one Email template',
                'email_template_two Email template',
                'letter_template_one Letter template',
                'letter_template_two Letter template',
            ],
            [
                'folder_one 2 folders',
                'folder_two Empty',
                'sms_template_one Text message template',
                'sms_template_two Text message template',
                'email_template_one Email template',
                'email_template_two Email template',
                'letter_template_one Letter template',
                'letter_template_two Letter template',
            ],
            [
                'folder_one',
                'folder_one_one',
                'folder_one_one_one',
                'sms_template_nested',
                'letter_template_nested',
                'folder_one_two',
                'folder_two',
                'sms_template_one',
                'sms_template_two',
                'email_template_one',
                'email_template_two',
                'letter_template_one',
                'letter_template_two',
            ],
            None,
        ),
        (
            'Templates – service one – GOV.UK Notify',
            'Templates',
            [],
            {'template_type': 'sms'},
            ['All', 'Email', 'Letter'],
            [
                'folder_one 1 folder',
                'folder_one / folder_one_one 1 folder',
                'folder_one / folder_one_one / folder_one_one_one 1 template',
                'folder_one / folder_one_one / folder_one_one_one / sms_template_nested Text message template',
                'sms_template_one Text message template',
                'sms_template_two Text message template',
            ],
            [
                'folder_one 1 folder',
                'sms_template_one Text message template',
                'sms_template_two Text message template',
            ],
            [
                'folder_one',
                'folder_one_one',
                'folder_one_one_one',
                'sms_template_nested',
                'sms_template_one',
                'sms_template_two',
            ],
            None,
        ),
        (
            'folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one',
            [{'template_type': 'all'}],
            {'template_folder_id': PARENT_FOLDER_ID},
            ['Text message', 'Email', 'Letter'],
            [
                'folder_one_one 1 template, 1 folder',
                'folder_one_one / folder_one_one_one 1 template',
                'folder_one_one / folder_one_one_one / sms_template_nested Text message template',
                'folder_one_one / letter_template_nested Letter template',
                'folder_one_two Empty',
            ],
            [
                'folder_one_one 1 template, 1 folder',
                'folder_one_two Empty',
            ],
            [
                'folder_one_one',
                'folder_one_one_one',
                'sms_template_nested',
                'letter_template_nested',
                'folder_one_two',
            ],
            None,
        ),
        (
            'folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one',
            [{'template_type': 'sms'}],
            {'template_type': 'sms', 'template_folder_id': PARENT_FOLDER_ID},
            ['All', 'Email', 'Letter'],
            [
                'folder_one_one 1 folder',
                'folder_one_one / folder_one_one_one 1 template',
                'folder_one_one / folder_one_one_one / sms_template_nested Text message template',
            ],
            [
                'folder_one_one 1 folder',
            ],
            [
                'folder_one_one',
                'folder_one_one_one',
                'sms_template_nested',
            ],
            None,
        ),
        (
            'folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one',
            [{'template_type': 'email'}],
            {'template_type': 'email', 'template_folder_id': PARENT_FOLDER_ID},
            ['All', 'Text message', 'Letter'],
            [],
            [],
            [],
            'There are no email templates in this folder',
        ),
        (
            'folder_one_one – folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one / folder_one_one',
            [
                {'template_type': 'all'},
                {'template_type': 'all', 'template_folder_id': PARENT_FOLDER_ID},
            ],
            {'template_folder_id': CHILD_FOLDER_ID},
            ['Text message', 'Email', 'Letter'],
            [
                'folder_one_one_one 1 template',
                'folder_one_one_one / sms_template_nested Text message template',
                'letter_template_nested Letter template',
            ],
            [
                'folder_one_one_one 1 template',
                'letter_template_nested Letter template',
            ],
            [
                'folder_one_one_one',
                'sms_template_nested',
                'letter_template_nested',
            ],
            None,
        ),
        (
            'folder_one_one_one – folder_one_one – folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one / folder_one_one / folder_one_one_one',
            [
                {'template_type': 'all'},
                {'template_type': 'all', 'template_folder_id': PARENT_FOLDER_ID},
                {'template_type': 'all', 'template_folder_id': CHILD_FOLDER_ID},
            ],
            {'template_folder_id': GRANDCHILD_FOLDER_ID},
            ['Text message', 'Email', 'Letter'],
            [
                'sms_template_nested Text message template',
            ],
            [
                'sms_template_nested Text message template',
            ],
            [
                'sms_template_nested',
            ],
            None,
        ),
        (
            'folder_one_one_one – folder_one_one – folder_one – Templates – service one – GOV.UK Notify',
            'Templates / folder_one / folder_one_one / folder_one_one_one',
            [
                {'template_type': 'email'},
                {'template_type': 'email', 'template_folder_id': PARENT_FOLDER_ID},
                {'template_type': 'email', 'template_folder_id': CHILD_FOLDER_ID},
            ],
            {
                'template_type': 'email',
                'template_folder_id': GRANDCHILD_FOLDER_ID,
            },
            ['All', 'Text message', 'Letter'],
            [],
            [],
            [],
            'There are no email templates in this folder',
        ),
        (
            'folder_two – Templates – service one – GOV.UK Notify',
            'Templates / folder_two',
            [{'template_type': 'all'}],
            {'template_folder_id': FOLDER_TWO_ID},
            ['Text message', 'Email', 'Letter'],
            [],
            [],
            [],
            'This folder is empty',
        ),
        (
            'folder_two – Templates – service one – GOV.UK Notify',
            'Templates / folder_two',
            [{'template_type': 'sms'}],
            {'template_folder_id': FOLDER_TWO_ID, 'template_type': 'sms'},
            ['All', 'Email', 'Letter'],
            [],
            [],
            [],
            'This folder is empty',
        ),
    ]
)
def test_should_show_templates_folder_page(
    client_request,
    mock_get_template_folders,
    mock_has_no_jobs,
    service_one,
    mocker,
    expected_title_tag,
    expected_page_title,
    expected_parent_link_args,
    extra_args,
    expected_nav_links,
    expected_items,
    expected_displayed_items,
    expected_searchable_text,
    expected_empty_message,
):
    mock_get_template_folders.return_value = [
        _folder('folder_two', FOLDER_TWO_ID),
        _folder('folder_one', PARENT_FOLDER_ID),
        _folder('folder_one_two', parent=PARENT_FOLDER_ID),
        _folder('folder_one_one', CHILD_FOLDER_ID, parent=PARENT_FOLDER_ID),
        _folder('folder_one_one_one', GRANDCHILD_FOLDER_ID, parent=CHILD_FOLDER_ID),
    ]
    mock_get_service_templates = mocker.patch(
        'app.service_api_client.get_service_templates',
        return_value={'data': [
            _template('sms', 'sms_template_one'),
            _template('sms', 'sms_template_two'),
            _template('email', 'email_template_one'),
            _template('email', 'email_template_two'),
            _template('letter', 'letter_template_one'),
            _template('letter', 'letter_template_two'),
            _template('letter', 'letter_template_nested', parent=CHILD_FOLDER_ID),
            _template('sms', 'sms_template_nested', parent=GRANDCHILD_FOLDER_ID),
        ]}
    )

    service_one['permissions'] += ['letter', 'edit_folders']

    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _test_page_title=False,
        **extra_args
    )

    assert normalize_spaces(page.select_one('title').text) == expected_title_tag
    assert normalize_spaces(page.select_one('h1').text) == expected_page_title

    assert len(page.select('h1 a')) == len(expected_parent_link_args)

    for index, parent_link in enumerate(page.select('h1 a')):
        assert parent_link['href'] == url_for(
            'main.choose_template',
            service_id=SERVICE_ONE_ID,
            **expected_parent_link_args[index]
        )

    links_in_page = page.select('.pill a')

    assert len(links_in_page) == len(expected_nav_links)

    for index, expected_link in enumerate(expected_nav_links):
        assert links_in_page[index].text.strip() == expected_link

    all_page_items = page.select('.template-list-item')
    checkboxes = page.select('input[name=templates_and_folders]')
    unique_checkbox_values = set(item['value'] for item in checkboxes)
    assert len(all_page_items) == len(expected_items)
    assert len(checkboxes) == len(expected_items)
    assert len(unique_checkbox_values) == len(expected_items)

    for index, expected_item in enumerate(expected_items):
        assert normalize_spaces(all_page_items[index].text) == expected_item

    displayed_page_items = page.find_all(lambda tag: (
        tag.has_attr('class')
        and 'template-list-item' in tag['class']
        and 'template-list-item-hidden-by-default' not in tag['class']
    ))
    assert len(displayed_page_items) == len(expected_displayed_items)

    for index, expected_item in enumerate(expected_displayed_items):
        assert '/' not in expected_item  # Yo dawg I heard you like tests…
        assert normalize_spaces(displayed_page_items[index].text) == expected_item

    all_searchable_text = page.select('#template-list .template-list-item .live-search-relevant')
    assert len(all_searchable_text) == len(expected_searchable_text)

    for index, expected_item in enumerate(expected_searchable_text):
        assert normalize_spaces(all_searchable_text[index].text) == expected_item

    if expected_empty_message:
        assert normalize_spaces(page.select_one('.template-list-empty').text) == (
            expected_empty_message
        )
    else:
        assert not page.select('.template-list-empty')

    mock_get_service_templates.assert_called_once_with(SERVICE_ONE_ID)


@pytest.mark.parametrize("template_type", ["email", "sms"])
def test_add_template_by_type_should_redirect_to_add_service_template(
    client_request,
    service_one,
    template_type,
    mock_get_service_templates,
    mock_get_organisations_and_services_for_user,
):
    service_one['permissions'] += ['edit_folders']
    client_request.post(
        'main.add_template_by_type',
        service_id=SERVICE_ONE_ID,
        template_folder_id=PARENT_FOLDER_ID,
        _data={'template_type': template_type},
        _expected_redirect=url_for('main.add_service_template',
                                   service_id=SERVICE_ONE_ID,
                                   template_type=template_type,
                                   template_folder_id=PARENT_FOLDER_ID,
                                   _external=True),
    )


def test_add_template_by_type_should_redirect_to_view_template_for_letter(
        client_request,
        service_one,
        mock_get_service_templates,
        mock_get_organisations_and_services_for_user,
        mock_create_service_template
):
    service_one['permissions'] += ['edit_folders']
    service_one['permissions'] += ['letter']
    client_request.post(
        'main.add_template_by_type',
        service_id=SERVICE_ONE_ID,
        template_folder_id=PARENT_FOLDER_ID,
        _data={'template_type': 'letter'},
        _expected_redirect=url_for('main.view_template',
                                   service_id=SERVICE_ONE_ID,
                                   template_id='Untitled',
                                   _external=True),
    )
    mock_create_service_template.assert_called_once_with('Untitled',
                                                         'letter',
                                                         'Body',
                                                         SERVICE_ONE_ID,
                                                         'Main heading',
                                                         'normal',
                                                         PARENT_FOLDER_ID)


def test_can_create_email_template_with_parent_folder(
        client_request,
        mock_create_service_template
):
    data = {
        'name': "new name",
        'subject': "Food incoming!",
        'template_content': "here's a burrito 🌯",
        'template_type': 'email',
        'service': SERVICE_ONE_ID,
        'process_type': 'normal',
        'parent_folder_id': PARENT_FOLDER_ID
    }
    client_request.post('.add_service_template',
                        service_id=SERVICE_ONE_ID,
                        template_type='email',
                        template_folder_id=PARENT_FOLDER_ID,
                        _data=data,
                        _expected_redirect=url_for("main.view_template",
                                                   service_id=SERVICE_ONE_ID,
                                                   template_id="new%20name",
                                                   _external=True)
                        )
    mock_create_service_template.assert_called_once_with(
        data['name'],
        data['template_type'],
        data['template_content'],
        SERVICE_ONE_ID,
        data['subject'],
        data['process_type'],
        data['parent_folder_id'])


def test_get_manage_folder_page(
    client_request,
    service_one,
    mock_get_template_folders,
):
    folder_id = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': folder_id, 'name': 'folder_two', 'parent_id': None},
    ]
    service_one['permissions'] += ['edit_folders']

    page = client_request.get(
        'main.manage_template_folder',
        service_id=service_one['id'],
        template_folder_id=folder_id,
        _test_page_title=False,
    )
    assert normalize_spaces(page.select_one('title').text) == (
        'folder_two – Templates – service one – GOV.UK Notify'
    )
    assert page.select_one('input[name=name]')['value'] == 'folder_two'
    delete_link = page.find('a', string="Delete this folder")
    expected_delete_url = "/services/{}/templates/folders/{}/delete".format(service_one['id'], folder_id)

    assert expected_delete_url in delete_link["href"]


def test_manage_folder_page_404s(client_request, service_one, mock_get_template_folders):
    service_one['permissions'] += ['edit_folders']
    client_request.get(
        'main.manage_template_folder',
        service_id=service_one['id'],
        template_folder_id=str(uuid.uuid4()),
        _expected_status=404,
    )


def test_get_manage_folder_page_no_permissions(client_request, service_one, mock_get_template_folders):
    folder_id = str(uuid.uuid4())

    client_request.get(
        'main.manage_template_folder',
        service_id=service_one['id'],
        template_folder_id=folder_id,
        _expected_status=403
    )


def test_rename_folder(client_request, service_one, mock_get_template_folders, mocker):
    mock_update = mocker.patch('app.template_folder_api_client.update_template_folder')
    folder_id = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': folder_id, 'name': 'folder_two', 'parent_id': None},
    ]
    service_one['permissions'] += ['edit_folders']

    client_request.post(
        'main.manage_template_folder',
        service_id=service_one['id'],
        template_folder_id=folder_id,
        _data={"name": "new beautiful name"},
        _expected_redirect=url_for("main.choose_template",
                                   service_id=service_one['id'],
                                   template_folder_id=folder_id,
                                   _external=True)
    )

    mock_update.assert_called_once_with(
        service_one['id'],
        folder_id,
        name="new beautiful name"
    )


def test_delete_template_folder_should_request_confirmation(
    client_request, service_one, mock_get_template_folders, mocker
):
    service_one['permissions'] += ['edit_folders']
    folder_id = str(uuid.uuid4())
    mock_get_template_folders.side_effect = [[
        {'id': folder_id, 'name': 'sacrifice', 'parent_id': None},
    ], []]
    mocker.patch(
        'app.models.service.Service.get_templates',
        return_value=[],
    )
    page = client_request.get(
        'main.delete_template_folder', service_id=service_one['id'],
        template_folder_id=folder_id,
        _test_page_title=False,
    )
    assert normalize_spaces(page.select('.banner-dangerous')[0].text) == (
        'Are you sure you want to delete the ‘sacrifice’ folder? '
        'Yes, delete'
    )

    assert page.select_one('input[name=name]')['value'] == 'sacrifice'

    assert len(page.select('form')) == 2
    assert len(page.select('button')) == 2

    assert 'action' not in page.select('form')[0]
    assert page.select('form button')[0].text == 'Yes, delete'

    assert page.select('form')[1]['action'] == url_for(
        'main.manage_template_folder',
        service_id=service_one['id'],
        template_folder_id=folder_id,
    )
    assert page.select('form button')[1].text == 'Save'


def test_delete_template_folder_should_detect_non_empty_folder_on_get(
    client_request, service_one, mock_get_template_folders, mocker
):
    service_one['permissions'] += ['edit_folders']
    folder_id = str(uuid.uuid4())
    template_id = str(uuid.uuid4())
    mock_get_template_folders.side_effect = [
        [{'id': folder_id, 'name': "can't touch me", 'parent_id': None}],
        []
    ]
    mocker.patch(
        'app.models.service.Service.get_templates',
        return_value=[{'id': template_id, 'name': 'template'}],
    )
    client_request.get(
        'main.delete_template_folder', service_id=service_one['id'],
        template_folder_id=folder_id,
        _expected_redirect=url_for(
            "main.choose_template",
            template_type="all",
            service_id=service_one['id'],
            template_folder_id=folder_id,
            _external=True
        ),
        _expected_status=302
    )


@pytest.mark.parametrize('parent_folder_id', (
    None,
    PARENT_FOLDER_ID,
))
def test_delete_folder(client_request, service_one, mock_get_template_folders, mocker, parent_folder_id):
    mock_delete = mocker.patch('app.template_folder_api_client.delete_template_folder')
    folder_id = str(uuid.uuid4())
    mock_get_template_folders.side_effect = [[
        {'id': folder_id, 'name': 'sacrifice', 'parent_id': parent_folder_id},
    ], []]
    mocker.patch(
        'app.models.service.Service.get_templates',
        return_value=[],
    )
    service_one['permissions'] += ['edit_folders']

    client_request.post(
        'main.delete_template_folder',
        service_id=service_one['id'],
        template_folder_id=folder_id,
        _expected_redirect=url_for(
            "main.choose_template",
            service_id=service_one['id'],
            template_folder_id=parent_folder_id,
            _external=True,
        )
    )

    mock_delete.assert_called_once_with(service_one['id'], folder_id)


@pytest.mark.parametrize('user', [
    pytest.param(
        active_user_with_permissions
    ),
    pytest.param(
        active_user_view_permissions,
        marks=pytest.mark.xfail(raises=AssertionError)
    ),
    pytest.param(
        active_caseworking_user,
        marks=pytest.mark.xfail(raises=AssertionError)
    ),
])
@pytest.mark.parametrize('extra_service_permissions', [
    pytest.param(
        ['edit_folders']
    ),
    pytest.param(
        [],
        marks=pytest.mark.xfail(raises=AssertionError)
    ),
])
def test_should_show_checkboxes_for_selecting_templates(
    client_request,
    mocker,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_has_no_jobs,
    fake_uuid,
    user,
    extra_service_permissions,
):
    service_one['permissions'] += extra_service_permissions
    client_request.login(user(fake_uuid))

    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
    )
    checkboxes = page.select('input[name=templates_and_folders]')

    assert len(checkboxes) == 4

    assert checkboxes[0]['value'] == TEMPLATE_ONE_ID
    assert checkboxes[0]['id'] == 'templates-or-folder-{}'.format(TEMPLATE_ONE_ID)

    for index in (1, 2, 3):
        assert checkboxes[index]['value'] != TEMPLATE_ONE_ID
        assert TEMPLATE_ONE_ID not in checkboxes[index]['id']


@pytest.mark.parametrize('user,extra_service_permissions', [
    (active_user_with_permissions, []),
    (active_user_view_permissions, ['edit_folders']),
    (active_caseworking_user, ['edit_folders']),
])
def test_should_not_show_radios_and_buttons_for_move_destination_if_incorrect_permissions(
    client_request,
    mocker,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_has_no_jobs,
    fake_uuid,
    user,
    extra_service_permissions,
):
    service_one['permissions'] += extra_service_permissions

    client_request.login(user(fake_uuid))

    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
    )
    radios = page.select('input[type=radio]')
    radio_div = page.find('div', {'id': 'move_to_folder_radios'})
    assert radios == page.select('input[name=move_to]')

    assert not radios
    assert not radio_div
    assert page.find_all('button', {'name': 'operation'}) == []


def test_should_show_radios_and_buttons_for_move_destination_if_correct_permissions(
    client_request,
    mocker,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_has_no_jobs,
    fake_uuid,
    active_user_with_permissions
):
    service_one['permissions'] += ['edit_folders']

    client_request.login(active_user_with_permissions)

    FOLDER_TWO_ID = str(uuid.uuid4())
    FOLDER_ONE_TWO_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
        {'id': CHILD_FOLDER_ID, 'name': 'folder_one_one', 'parent_id': PARENT_FOLDER_ID},
        {'id': FOLDER_ONE_TWO_ID, 'name': 'folder_one_two', 'parent_id': PARENT_FOLDER_ID},
    ]
    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
    )
    radios = page.select('#move_to_folder_radios input[type=radio]')
    radio_div = page.find('div', {'id': 'move_to_folder_radios'})
    assert radios == page.select('input[name=move_to]')

    assert [x['value'] for x in radios] == [
        ROOT_FOLDER_ID, PARENT_FOLDER_ID, CHILD_FOLDER_ID, FOLDER_ONE_TWO_ID, FOLDER_TWO_ID,
    ]
    assert [x.text.strip() for x in radio_div.select('label')] == [
        'Templates', 'folder_one', 'folder_one_one', 'folder_one_two', 'folder_two',
    ]
    assert set(x['value'] for x in page.find_all('button', {'name': 'operation'})) == {
        'unknown',
        'move-to-existing-folder',
        'move-to-new-folder',
        'add-new-folder',
        'add-new-template',
    }


def test_move_to_shouldnt_select_a_folder_by_default(
    client_request,
    mocker,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_has_no_jobs,
    fake_uuid,
    active_user_with_permissions
):
    service_one['permissions'] += ['edit_folders']

    client_request.login(active_user_with_permissions)

    FOLDER_TWO_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]
    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
    )
    checked_radio = page.find('input', attrs={'name': 'move_to', 'checked': 'checked'})
    assert checked_radio is None


def test_should_be_able_to_move_to_existing_folder(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
):
    service_one['permissions'] += ['edit_folders']
    FOLDER_TWO_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]
    client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data={
            'operation': 'move-to-existing-folder',
            'move_to': PARENT_FOLDER_ID,
            'templates_and_folders': [
                FOLDER_TWO_ID,
                TEMPLATE_ONE_ID,
            ],
        },
        _expected_status=302,
        _expected_redirect=url_for(
            'main.choose_template',
            service_id=SERVICE_ONE_ID,
            _external=True,
        ),
    )
    mock_move_to_template_folder.assert_called_once_with(
        service_id=SERVICE_ONE_ID,
        folder_id=PARENT_FOLDER_ID,
        folder_ids={FOLDER_TWO_ID},
        template_ids={TEMPLATE_ONE_ID},
    )


@pytest.mark.parametrize('user,extra_service_permissions', [
    (active_user_view_permissions, ['edit_folders']),
    (active_user_with_permissions, [])
])
def test_should_not_be_able_to_move_to_existing_folder_if_dont_have_permission(
    client_request,
    service_one,
    fake_uuid,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
    user,
    extra_service_permissions,
):
    service_one['permissions'] += extra_service_permissions
    client_request.login(user(fake_uuid))
    FOLDER_TWO_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]
    client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data={
            'operation': 'move-to-existing-folder',
            'move_to': PARENT_FOLDER_ID,
            'templates_and_folders': [
                FOLDER_TWO_ID,
                TEMPLATE_ONE_ID,
            ],
        },
        _expected_status=403
    )
    assert mock_move_to_template_folder.called is False


def test_move_folder_form_shows_current_folder_hint_when_in_a_folder(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
):
    service_one['permissions'] += ['edit_folders']
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'parent_folder', 'parent_id': None},
        {'id': CHILD_FOLDER_ID, 'name': 'child_folder', 'parent_id': PARENT_FOLDER_ID},
    ]
    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        template_folder_id=PARENT_FOLDER_ID,
        _test_page_title=False
    )

    page.find("input", attrs={"name": "move_to", "value": PARENT_FOLDER_ID})

    move_form_labels = page.find('div', id='move_to_folder_radios').find_all('label')

    assert len(move_form_labels) == 3
    assert normalize_spaces(move_form_labels[0].text) == 'Templates'
    assert normalize_spaces(move_form_labels[1].text) == 'parent_folder current folder'
    assert normalize_spaces(move_form_labels[2].text) == 'child_folder'


def test_move_folder_form_does_not_show_current_folder_hint_at_the_top_level(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
):
    service_one['permissions'] += ['edit_folders']
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'parent_folder', 'parent_id': None},
        {'id': CHILD_FOLDER_ID, 'name': 'child_folder', 'parent_id': PARENT_FOLDER_ID},
    ]
    page = client_request.get(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _test_page_title=False
    )

    page.find("input", attrs={"name": "move_to", "value": PARENT_FOLDER_ID})

    move_form_labels = page.find('div', id='move_to_folder_radios').find_all('label')

    assert len(move_form_labels) == 3
    assert normalize_spaces(move_form_labels[0].text) == 'Templates'
    assert normalize_spaces(move_form_labels[1].text) == 'parent_folder'
    assert normalize_spaces(move_form_labels[2].text) == 'child_folder'


def test_should_be_able_to_move_a_sub_item(
    client_request,
    service_one,
    fake_uuid,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
):
    service_one['permissions'] += ['edit_folders']
    GRANDCHILD_FOLDER_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': CHILD_FOLDER_ID, 'name': 'folder_one_one', 'parent_id': PARENT_FOLDER_ID},
        {'id': GRANDCHILD_FOLDER_ID, 'name': 'folder_one_one_one', 'parent_id': CHILD_FOLDER_ID},
    ]
    client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        template_folder_id=PARENT_FOLDER_ID,
        _data={
            'operation': 'move-to-existing-folder',
            'move_to': ROOT_FOLDER_ID,
            'templates_and_folders': [GRANDCHILD_FOLDER_ID],
        },
        _expected_status=302,
    )
    mock_move_to_template_folder.assert_called_once_with(
        service_id=SERVICE_ONE_ID,
        folder_id=None,
        folder_ids={GRANDCHILD_FOLDER_ID},
        template_ids=set(),
    )


@pytest.mark.parametrize('data', [
    # move to existing, but add new folder name given
    {
        'operation': 'move-to-existing-folder',
        'templates_and_folders': [],
        'add_new_folder_name': 'foo',
        'move_to': PARENT_FOLDER_ID
    },
    # move to existing, but move to new folder name given
    {
        'operation': 'move-to-existing-folder',
        'templates_and_folders': [TEMPLATE_ONE_ID],
        'move_to_new_folder_name': 'foo',
        'move_to': PARENT_FOLDER_ID
    },
    # move to existing, but no templates to move
    {
        'operation': 'move-to-existing-folder',
        'templates_and_folders': [],
        'move_to_new_folder_name': '',
        'move_to': PARENT_FOLDER_ID
    },
    # move to new, but nothing selected to move
    {
        'operation': 'move-to-new-folder',
        'templates_and_folders': [],
        'move_to_new_folder_name': 'foo',
        'move_to': None
    },
    # add a new template, but also select move destination
    {
        'operation': 'add-new-template',
        'templates_and_folders': [],
        'move_to_new_folder_name': '',
        'move_to': PARENT_FOLDER_ID,
        'add_template_by_template_type': 'email',
    },
    # add a new template, but also move to root folder
    {
        'operation': 'add-new-template',
        'templates_and_folders': [],
        'move_to_new_folder_name': '',
        'move_to': ROOT_FOLDER_ID,
        'add_template_by_template_type': 'email',
    },
    # add a new template, but don't select anything
    {
        'operation': 'add-new-template',
    },
])
def test_no_action_if_user_fills_in_ambiguous_fields(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
    mock_create_template_folder,
    data,
):
    service_one['permissions'] += ['edit_folders', 'letter']

    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'parent folder', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]

    page = client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data=data,
        _expected_status=200,
        _expected_redirect=None,
    )

    assert mock_move_to_template_folder.called is False
    assert mock_create_template_folder.called is False

    assert page.select_one('button[value={}]'.format(data['operation']))

    assert [
        'email',
        'sms',
        'letter',
        'copy-existing',
    ] == [
        radio['value']
        for radio in page.select('#add_new_template_form input[type=radio]')
    ]

    assert [
        ROOT_FOLDER_ID,
        FOLDER_TWO_ID,
        PARENT_FOLDER_ID,
    ] == [
        radio['value']
        for radio in page.select('#move_to_folder_radios input[type=radio]')
    ]


def test_new_folder_is_created_if_only_new_folder_is_filled_out(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
    mock_create_template_folder
):
    data = {
        'move_to_new_folder_name': '',
        'add_new_folder_name': 'new folder',
        'operation': 'add-new-folder'
    }

    service_one['permissions'] += ['edit_folders']

    client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data=data,
        _expected_status=302,
        _expected_redirect=url_for(
            'main.choose_template',
            service_id=service_one['id'],
            template_folder_id=None,
            _external=True,
        ),
    )

    assert mock_move_to_template_folder.called is False
    mock_create_template_folder.assert_called_once_with(
        SERVICE_ONE_ID,
        name='new folder',
        parent_id=None
    )


def test_should_be_able_to_move_to_new_folder(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
    mock_create_template_folder,
):
    service_one['permissions'] += ['edit_folders']
    new_folder_id = mock_create_template_folder.return_value
    FOLDER_TWO_ID = str(uuid.uuid4())
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'parent folder', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]

    client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        template_folder_id=None,
        _data={
            'operation': 'move-to-new-folder',
            'move_to_new_folder_name': 'new folder',
            'templates_and_folders': [
                FOLDER_TWO_ID,
                TEMPLATE_ONE_ID,
            ],
        },
        _expected_status=302,
        _expected_redirect=url_for('main.choose_template', service_id=SERVICE_ONE_ID, _external=True),
    )

    mock_create_template_folder.assert_called_once_with(
        SERVICE_ONE_ID,
        name='new folder',
        parent_id=None
    )
    mock_move_to_template_folder.assert_called_once_with(
        service_id=SERVICE_ONE_ID,
        folder_id=new_folder_id,
        folder_ids={FOLDER_TWO_ID},
        template_ids={TEMPLATE_ONE_ID},
    )


def test_radio_button_with_no_value_shows_custom_error_message(
    client_request,
    service_one,
    mock_get_service_templates,
    mock_get_template_folders,
    mock_move_to_template_folder,
    mock_create_template_folder,
):
    service_one['permissions'] += ['edit_folders']

    page = client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data={'operation': 'add-new-template'},
        _expected_status=200,
        _expected_redirect=None,
    )

    assert mock_move_to_template_folder.called is False
    assert mock_create_template_folder.called is False

    assert page.select_one('span.error-message').text.strip() == 'Select the type of template you want to add'


@pytest.mark.parametrize('data, error_msg', [
    # nothing selected when moving
    (
        {'operation': 'move-to-new-folder', 'templates_and_folders': [], 'move_to_new_folder_name': 'foo'},
        'Select at least one template or folder'
    ),
    (
        {'operation': 'move-to-existing-folder', 'templates_and_folders': [], 'move_to': PARENT_FOLDER_ID},
        'Select at least one template or folder'
    ),
    # api error (eg moving folder to itself)
    (
        {'operation': 'move-to-existing-folder', 'templates_and_folders': [FOLDER_TWO_ID], 'move_to': FOLDER_TWO_ID},
        'Some api error msg'
    )

])
def test_show_custom_error_message(
        client_request,
        service_one,
        mock_get_service_templates,
        mock_get_template_folders,
        mock_move_to_template_folder,
        mock_create_template_folder,
        data,
        error_msg,
):
    service_one['permissions'] += ['edit_folders']
    mock_get_template_folders.return_value = [
        {'id': PARENT_FOLDER_ID, 'name': 'folder_one', 'parent_id': None},
        {'id': FOLDER_TWO_ID, 'name': 'folder_two', 'parent_id': None},
    ]
    mock_move_to_template_folder.side_effect = HTTPError(message='Some api error msg')

    page = client_request.post(
        'main.choose_template',
        service_id=SERVICE_ONE_ID,
        _data=data,
        _expected_status=200,
        _expected_redirect=None,
    )

    assert page.select_one('div.banner-dangerous').text.strip() == error_msg