notifications-admin/tests/app/main/views/organisations/test_organisations.py

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

from tests import organisation_json, service_json
from tests.app.main.views.test_agreement import MockS3Object
from tests.conftest import (
    ORGANISATION_ID,
    SERVICE_ONE_ID,
    SERVICE_TWO_ID,
    create_active_user_with_permissions,
    create_platform_admin_user,
    normalize_spaces,
)


def test_organisation_page_shows_all_organisations(
    client_request,
    platform_admin_user,
    mocker
):
    orgs = [
        {'id': 'A3', 'name': 'Test 3', 'active': True, 'count_of_live_services': 0},
        {'id': 'B1', 'name': 'Test 1', 'active': True, 'count_of_live_services': 1},
        {'id': 'C2', 'name': 'Test 2', 'active': False, 'count_of_live_services': 2},
    ]

    get_organisations = mocker.patch(
        'app.models.organisation.AllOrganisations.client_method', return_value=orgs
    )
    client_request.login(platform_admin_user)
    page = client_request.get('.organisations')

    assert normalize_spaces(
        page.select_one('h1').text
    ) == "Organisations"

    assert [
        (
            normalize_spaces(link.text),
            normalize_spaces(hint.text),
            link['href'],
        ) for link, hint in zip(
            page.select('.browse-list-item a'),
            page.select('.browse-list-item .browse-list-hint'),
        )
    ] == [
        ('Test 1', '1 live service', url_for(
            'main.organisation_dashboard', org_id='B1'
        )),
        ('Test 2', '2 live services', url_for(
            'main.organisation_dashboard', org_id='C2'
        )),
        ('Test 3', '0 live services', url_for(
            'main.organisation_dashboard', org_id='A3'
        )),
    ]

    archived = page.select_one('.table-field-status-default.heading-medium')
    assert normalize_spaces(archived.text) == '- archived'
    assert normalize_spaces(archived.parent.text) == 'Test 2 - archived 2 live services'

    assert normalize_spaces(
        page.select_one('a.govuk-button--secondary').text
    ) == 'New organisation'
    get_organisations.assert_called_once_with()


def test_view_organisation_shows_the_correct_organisation(
    client_request,
    mocker
):
    org = {'id': ORGANISATION_ID, 'name': 'Test 1', 'active': True}
    mocker.patch(
        'app.organisations_client.get_organisation', return_value=org
    )
    mocker.patch(
        'app.organisations_client.get_services_and_usage', return_value={'services': {}}
    )

    page = client_request.get(
        '.organisation_dashboard',
        org_id=ORGANISATION_ID,
    )

    assert normalize_spaces(page.select_one('h1').text) == 'Usage'
    assert normalize_spaces(page.select_one('.govuk-hint').text) == (
        'Test 1 has no live services on U.S. Notify'
    )
    assert not page.select('a[download]')


def test_page_to_create_new_organisation(
    client_request,
    platform_admin_user,
    mocker,
):
    client_request.login(platform_admin_user)
    page = client_request.get('.add_organisation')

    assert [
        (input['type'], input['name'], input.get('value'))
        for input in page.select('input')
    ] == [
        ('text', 'name', None),
        ('radio', 'organisation_type', 'federal'),
        ('radio', 'organisation_type', 'state'),
        # ('radio', 'organisation_type', 'nhs_central'),
        # ('radio', 'organisation_type', 'nhs_local'),
        # ('radio', 'organisation_type', 'nhs_gp'),
        # ('radio', 'organisation_type', 'emergency_service'),
        # ('radio', 'organisation_type', 'school_or_college'),
        ('radio', 'organisation_type', 'other'),
        ('radio', 'crown_status', 'crown'),
        ('radio', 'crown_status', 'non-crown'),
        ('hidden', 'csrf_token', mocker.ANY),
    ]


def test_create_new_organisation(
    client_request,
    platform_admin_user,
    mocker,
):
    mock_create_organisation = mocker.patch(
        'app.organisations_client.create_organisation',
        return_value=organisation_json(ORGANISATION_ID),
    )

    client_request.login(platform_admin_user)
    client_request.post(
        '.add_organisation',
        _data={
            'name': 'new name',
            'organisation_type': 'federal',
            'crown_status': 'non-crown',
        },
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=ORGANISATION_ID,
        ),
    )

    mock_create_organisation.assert_called_once_with(
        name='new name',
        organisation_type='federal',
        crown=False,
        agreement_signed=False,
    )


def test_create_new_organisation_validates(
    client_request,
    platform_admin_user,
    mocker,
):
    mock_create_organisation = mocker.patch(
        'app.organisations_client.create_organisation'
    )

    client_request.login(platform_admin_user)
    page = client_request.post(
        '.add_organisation',
        _expected_status=200,
    )
    assert [
        (error['data-error-label'], normalize_spaces(error.text))
        for error in page.select('.govuk-error-message')
    ] == [
        ('name', 'Error: Cannot be empty'),
        ('organisation_type', 'Error: Select the type of organisation'),
        ('crown_status', 'Error: Select whether this organisation is a crown body'),
    ]
    assert mock_create_organisation.called is False


@pytest.mark.parametrize('name, error_message', [
    ('', 'Cannot be empty'),
    ('a', 'at least two alphanumeric characters'),
    ('a' * 256, 'Organisation name must be 255 characters or fewer'),
])
def test_create_new_organisation_fails_with_incorrect_input(
    client_request,
    platform_admin_user,
    mocker,
    name,
    error_message,
):
    mock_create_organisation = mocker.patch(
        'app.organisations_client.create_organisation'
    )

    client_request.login(platform_admin_user)
    page = client_request.post(
        '.add_organisation',
        _data={
            'name': name,
            'organisation_type': 'local',
            'crown_status': 'non-crown',
        },
        _expected_status=200,
    )
    assert mock_create_organisation.called is False
    assert error_message in page.select_one('.govuk-error-message').text


def test_create_new_organisation_fails_with_duplicate_name(
    client_request,
    platform_admin_user,
    mocker,
):
    def _create(**_kwargs):
        json_mock = mocker.Mock(return_value={'message': 'Organisation name already exists'})
        resp_mock = mocker.Mock(status_code=400, json=json_mock)
        http_error = HTTPError(response=resp_mock, message="Default message")
        raise http_error

    mocker.patch(
        'app.organisations_client.create_organisation',
        side_effect=_create
    )

    client_request.login(platform_admin_user)
    page = client_request.post(
        '.add_organisation',
        _data={
            'name': 'Existing org',
            'organisation_type': 'federal',
            'crown_status': 'non-crown',
        },
        _expected_status=200,
    )

    error_message = 'This organisation name is already in use'
    assert error_message in page.select_one('.govuk-error-message').text


@pytest.mark.parametrize('organisation_type, organisation, expected_status', (
    ('nhs_gp', None, 200),
    ('central', None, 403),
    ('nhs_gp', organisation_json(organisation_type='nhs_gp'), 403),
))
@pytest.mark.skip(reason='Update for TTS')
def test_gps_can_create_own_organisations(
    client_request,
    mocker,
    mock_get_service_organisation,
    service_one,
    organisation_type,
    organisation,
    expected_status,
):
    mocker.patch('app.organisations_client.get_organisation', return_value=organisation)
    service_one['organisation_type'] = organisation_type

    page = client_request.get(
        '.add_organisation_from_gp_service',
        service_id=SERVICE_ONE_ID,
        _expected_status=expected_status,
    )

    if expected_status == 403:
        return

    assert page.select_one('input[type=text]')['name'] == 'name'
    assert normalize_spaces(
        page.select_one('label[for=name]').text
    ) == (
        'What’s your practice called?'
    )


@pytest.mark.parametrize('organisation_type, organisation, expected_status', (
    ('nhs_local', None, 200),
    ('nhs_gp', None, 403),
    ('central', None, 403),
    ('nhs_local', organisation_json(organisation_type='nhs_local'), 403),
))
@pytest.mark.skip(reason='Update for TTS')
def test_nhs_local_can_create_own_organisations(
    client_request,
    mocker,
    mock_get_service_organisation,
    service_one,
    organisation_type,
    organisation,
    expected_status,
):
    mocker.patch('app.organisations_client.get_organisation', return_value=organisation)
    mocker.patch(
        'app.models.organisation.AllOrganisations.client_method',
        return_value=[
            organisation_json('t2', 'Trust 2', organisation_type='nhs_local'),
            organisation_json('t1', 'Trust 1', organisation_type='nhs_local'),
            organisation_json('gp1', 'GP 1', organisation_type='nhs_gp'),
            organisation_json('c1', 'Central 1'),
        ],
    )
    service_one['organisation_type'] = organisation_type

    page = client_request.get(
        '.add_organisation_from_nhs_local_service',
        service_id=SERVICE_ONE_ID,
        _expected_status=expected_status,
    )

    if expected_status == 403:
        return

    assert normalize_spaces(page.select_one('main p').text) == (
        'Which NHS Trust or Clinical Commissioning Group do you work for?'
    )
    assert page.select_one('[data-module=live-search]')['data-targets'] == (
        '.govuk-radios__item'
    )
    assert [
        (
            normalize_spaces(radio.select_one('label').text),
            radio.select_one('input')['value']
        )
        for radio in page.select('.govuk-radios__item')
    ] == [
        ('Trust 1', 't1'),
        ('Trust 2', 't2'),
    ]
    assert normalize_spaces(page.select_one('.js-stick-at-bottom-when-scrolling button').text) == (
        'Continue'
    )


@pytest.mark.parametrize('data, expected_service_name', (
    (
        {
            'same_as_service_name': False,
            'name': 'Dr. Example',
        },
        'Dr. Example',
    ),
    (
        {
            'same_as_service_name': True,
            'name': 'This is ignored',
        },
        'service one',
    ),
))
@pytest.mark.skip(reason='Update for TTS')
def test_gps_can_name_their_organisation(
    client_request,
    mocker,
    service_one,
    mock_update_service_organisation,
    data,
    expected_service_name,
):
    service_one['organisation_type'] = 'nhs_gp'
    mock_create_organisation = mocker.patch(
        'app.organisations_client.create_organisation',
        return_value=organisation_json(ORGANISATION_ID),
    )

    client_request.post(
        '.add_organisation_from_gp_service',
        service_id=SERVICE_ONE_ID,
        _data=data,
        _expected_status=302,
        _expected_redirect=url_for(
            'main.service_agreement',
            service_id=SERVICE_ONE_ID,
        )
    )

    mock_create_organisation.assert_called_once_with(
        name=expected_service_name,
        organisation_type='nhs_gp',
        agreement_signed=False,
        crown=False,
    )
    mock_update_service_organisation.assert_called_once_with(SERVICE_ONE_ID, ORGANISATION_ID)


@pytest.mark.parametrize('data, expected_error', (
    (
        {
            'name': 'Dr. Example',
        },
        'Select yes or no',
    ),
    (
        {
            'same_as_service_name': False,
            'name': '',
        },
        'Cannot be empty',
    ),
))
@pytest.mark.skip(reason='Update for TTS')
def test_validation_of_gps_creating_organisations(
    client_request,
    mocker,
    service_one,
    data,
    expected_error,
):
    service_one['organisation_type'] = 'nhs_gp'
    page = client_request.post(
        '.add_organisation_from_gp_service',
        service_id=SERVICE_ONE_ID,
        _data=data,
        _expected_status=200,
    )
    assert expected_error in page.select_one('.govuk-error-message, .error-message').text


@pytest.mark.skip(reason='Update for TTS')
def test_nhs_local_assigns_to_selected_organisation(
    client_request,
    mocker,
    service_one,
    mock_get_organisation,
    mock_update_service_organisation,
):
    mocker.patch(
        'app.models.organisation.AllOrganisations.client_method',
        return_value=[
            organisation_json(ORGANISATION_ID, 'Trust 1', organisation_type='nhs_local'),
        ],
    )
    service_one['organisation_type'] = 'nhs_local'

    client_request.post(
        '.add_organisation_from_nhs_local_service',
        service_id=SERVICE_ONE_ID,
        _data={
            'organisations': ORGANISATION_ID,
        },
        _expected_status=302,
        _expected_redirect=url_for(
            'main.service_agreement',
            service_id=SERVICE_ONE_ID,
        )
    )
    mock_update_service_organisation.assert_called_once_with(SERVICE_ONE_ID, ORGANISATION_ID)


@freeze_time("2020-02-20 20:20")
def test_organisation_services_shows_live_services_and_usage(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mock = mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {'service_id': SERVICE_ONE_ID, 'service_name': '1', 'chargeable_billable_sms': 250122, 'emails_sent': 13000,
             'free_sms_limit': 250000, 'letter_cost': 30.50, 'sms_billable_units': 122, 'sms_cost': 0,
             'sms_remainder': None},
            {'service_id': SERVICE_TWO_ID, 'service_name': '5', 'chargeable_billable_sms': 0, 'emails_sent': 20000,
             'free_sms_limit': 250000, 'letter_cost': 0, 'sms_billable_units': 2500, 'sms_cost': 42.0,
             'sms_remainder': None}
        ]}
    )

    client_request.login(active_user_with_permissions)
    page = client_request.get('.organisation_dashboard', org_id=ORGANISATION_ID)
    mock.assert_called_once_with(ORGANISATION_ID, 2019)

    services = page.select('main h3')
    usage_rows = page.select('main .govuk-grid-column-one-third')
    assert len(services) == 2

    # Totals
    assert normalize_spaces(usage_rows[0].text) == "Emails 33,000 sent"
    assert normalize_spaces(usage_rows[1].text) == "Text messages $42.00 spent"
    assert normalize_spaces(usage_rows[2].text) == "Letters $30.50 spent"

    assert normalize_spaces(services[0].text) == '1'
    assert normalize_spaces(services[1].text) == '5'
    assert services[0].find('a')['href'] == url_for('main.usage', service_id=SERVICE_ONE_ID)

    assert normalize_spaces(usage_rows[3].text) == "13,000 emails sent"
    assert normalize_spaces(usage_rows[4].text) == "122 free text messages sent"
    assert normalize_spaces(usage_rows[5].text) == "$30.50 spent on letters"
    assert services[1].find('a')['href'] == url_for('main.usage', service_id=SERVICE_TWO_ID)
    assert normalize_spaces(usage_rows[6].text) == "20,000 emails sent"
    assert normalize_spaces(usage_rows[7].text) == "$42.00 spent on text messages"
    assert normalize_spaces(usage_rows[8].text) == "$0.00 spent on letters"

    # Ensure there’s no ‘this org has no services message’
    assert not page.select('.govuk-hint')


@freeze_time("2020-02-20 20:20")
def test_organisation_services_shows_live_services_and_usage_with_count_of_1(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {'service_id': SERVICE_ONE_ID, 'service_name': '1', 'chargeable_billable_sms': 1, 'emails_sent': 1,
             'free_sms_limit': 250000, 'letter_cost': 0, 'sms_billable_units': 1, 'sms_cost': 0,
             'sms_remainder': None},
        ]}
    )

    client_request.login(active_user_with_permissions)
    page = client_request.get('.organisation_dashboard', org_id=ORGANISATION_ID)

    usage_rows = page.select('main .govuk-grid-column-one-third')

    # Totals
    assert normalize_spaces(usage_rows[0].text) == "Emails 1 sent"
    assert normalize_spaces(usage_rows[1].text) == "Text messages $0.00 spent"
    assert normalize_spaces(usage_rows[2].text) == "Letters $0.00 spent"

    assert normalize_spaces(usage_rows[3].text) == "1 email sent"
    assert normalize_spaces(usage_rows[4].text) == "1 free text message sent"
    assert normalize_spaces(usage_rows[5].text) == "$0.00 spent on letters"


@freeze_time("2020-02-20 20:20")
@pytest.mark.parametrize('financial_year, expected_selected', (
    (2017, '2017 to 2018 fiscal year'),
    (2018, '2018 to 2019 fiscal year'),
    (2019, '2019 to 2020 fiscal year'),
))
def test_organisation_services_filters_by_financial_year(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
    financial_year,
    expected_selected,
):
    mock = mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": []}
    )
    page = client_request.get(
        '.organisation_dashboard',
        org_id=ORGANISATION_ID,
        year=financial_year,
    )
    mock.assert_called_once_with(ORGANISATION_ID, financial_year)
    assert normalize_spaces(page.select_one('.pill').text) == (
        '2019 to 2020 fiscal year '
        '2018 to 2019 fiscal year '
        '2017 to 2018 fiscal year'
    )
    assert normalize_spaces(page.select_one('.pill-item--selected').text) == (
        expected_selected
    )


@freeze_time("2020-02-20 20:20")
def test_organisation_services_shows_search_bar(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {
                'service_id': SERVICE_ONE_ID,
                'service_name': 'Service 1',
                'chargeable_billable_sms': 250122,
                'emails_sent': 13000,
                'free_sms_limit': 250000,
                'letter_cost': 30.50,
                'sms_billable_units': 122,
                'sms_cost': 1.93,
                'sms_remainder': None
            },
        ] * 8}
    )

    client_request.login(active_user_with_permissions)
    page = client_request.get('.organisation_dashboard', org_id=ORGANISATION_ID)

    services = page.select('.organisation-service')
    assert len(services) == 8

    assert page.select_one('.live-search')['data-targets'] == '.organisation-service'
    assert [
        normalize_spaces(service_name.text)
        for service_name in page.select('.live-search-relevant')
    ] == [
        'Service 1',
        'Service 1',
        'Service 1',
        'Service 1',
        'Service 1',
        'Service 1',
        'Service 1',
        'Service 1',
    ]


@freeze_time("2020-02-20 20:20")
def test_organisation_services_hides_search_bar_for_7_or_fewer_services(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {
                'service_id': SERVICE_ONE_ID,
                'service_name': 'Service 1',
                'chargeable_billable_sms': 250122,
                'emails_sent': 13000,
                'free_sms_limit': 250000,
                'letter_cost': 30.50,
                'sms_billable_units': 122,
                'sms_cost': 1.93,
                'sms_remainder': None
            },
        ] * 7}
    )

    client_request.login(active_user_with_permissions)
    page = client_request.get('.organisation_dashboard', org_id=ORGANISATION_ID)

    services = page.select('.organisation-service')
    assert len(services) == 7
    assert not page.select_one('.live-search')


@freeze_time("2021-11-12 11:09:00.061258")
def test_organisation_services_links_to_downloadable_report(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {
                'service_id': SERVICE_ONE_ID,
                'service_name': 'Service 1',
                'chargeable_billable_sms': 250122,
                'emails_sent': 13000,
                'free_sms_limit': 250000,
                'letter_cost': 30.50,
                'sms_billable_units': 122,
                'sms_cost': 1.93,
                'sms_remainder': None
            },
        ] * 2}
    )
    client_request.login(active_user_with_permissions)
    page = client_request.get('.organisation_dashboard', org_id=ORGANISATION_ID)

    link_to_report = page.select_one('a[download]')
    assert normalize_spaces(link_to_report.text) == 'Download this report (CSV)'
    assert link_to_report.attrs["href"] == url_for(
        '.download_organisation_usage_report',
        org_id=ORGANISATION_ID,
        selected_year=2021
    )


@freeze_time("2021-11-12 11:09:00.061258")
def test_download_organisation_usage_report(
    client_request,
    mock_get_organisation,
    mocker,
    active_user_with_permissions,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_services_and_usage',
        return_value={"services": [
            {
                'service_id': SERVICE_ONE_ID,
                'service_name': 'Service 1',
                'chargeable_billable_sms': 22,
                'emails_sent': 13000,
                'free_sms_limit': 100,
                'letter_cost': 30.5,
                'sms_billable_units': 122,
                'sms_cost': 1.934,
                'sms_remainder': 0
            },
            {
                'service_id': SERVICE_TWO_ID,
                'service_name': 'Service 1',
                'chargeable_billable_sms': 222,
                'emails_sent': 23000,
                'free_sms_limit': 250000,
                'letter_cost': 60.5,
                'sms_billable_units': 322,
                'sms_cost': 3.935,
                'sms_remainder': 0
            },
        ]}
    )
    client_request.login(active_user_with_permissions)
    csv_report = client_request.get(
        '.download_organisation_usage_report',
        org_id=ORGANISATION_ID,
        selected_year=2021,
        _test_page_title=False
    )

    assert csv_report.string == (
        "Service ID,Service Name,Emails sent,Free text message allowance remaining,"
        "Spent on text messages ($),Spent on letters ($)"
        "\r\n596364a0-858e-42c8-9062-a8fe822260eb,Service 1,13000,0,1.93,30.50"
        "\r\n147ad62a-2951-4fa1-9ca0-093cd1a52c52,Service 1,23000,0,3.94,60.50\r\n"
    )


def test_organisation_trial_mode_services_shows_all_non_live_services(
    client_request,
    platform_admin_user,
    mock_get_organisation,
    mocker,
    fake_uuid,
):
    mocker.patch(
        'app.organisations_client.get_organisation_services',
        return_value=[
            service_json(id_='1', name='1', restricted=False, active=True),  # live
            service_json(id_='2', name='2', restricted=True, active=True),  # trial
            service_json(id_='3', name='3', restricted=False, active=False),  # archived
        ]
    )

    client_request.login(platform_admin_user)
    page = client_request.get(
        '.organisation_trial_mode_services',
        org_id=ORGANISATION_ID,
        _test_page_title=False
    )

    services = page.select('.browse-list-item')
    assert len(services) == 2

    assert normalize_spaces(services[0].text) == '2'
    assert normalize_spaces(services[1].text) == '3'
    assert services[0].find('a')['href'] == url_for('main.service_dashboard', service_id='2')
    assert services[1].find('a')['href'] == url_for('main.service_dashboard', service_id='3')


def test_organisation_trial_mode_services_doesnt_work_if_not_platform_admin(
    client_request,
    mock_get_organisation,
):
    client_request.get(
        '.organisation_trial_mode_services',
        org_id=ORGANISATION_ID,
        _expected_status=403
    )


def test_manage_org_users_shows_correct_link_next_to_each_user(
    client_request,
    mock_get_organisation,
    mock_get_users_for_organisation,
    mock_get_invited_users_for_organisation,
):
    page = client_request.get(
        '.manage_org_users',
        org_id=ORGANISATION_ID,
    )

    # No banner confirming a user to be deleted shown
    assert not page.select_one('.banner-dangerous')

    users = page.find_all(class_='user-list-item')

    # The first user is an invited user, so has the link to cancel the invitation.
    # The second two users are active users, so have the link to be removed from the org
    assert normalize_spaces(
        users[0].text
    ) == 'invited_user@test.gsa.gov (invited) Cancel invitation for invited_user@test.gsa.gov'
    assert normalize_spaces(users[1].text) == 'Test User 1 test@gsa.gov Remove Test User 1 test@gsa.gov'
    assert normalize_spaces(users[2].text) == 'Test User 2 testt@gsa.gov Remove Test User 2 testt@gsa.gov'

    assert users[0].a['href'] == url_for(
        '.cancel_invited_org_user',
        org_id=ORGANISATION_ID,
        invited_user_id='73616d70-6c65-4f6f-b267-5f696e766974'
    )
    assert users[1].a['href'] == url_for('.edit_organisation_user', org_id=ORGANISATION_ID, user_id='1234')
    assert users[2].a['href'] == url_for('.edit_organisation_user', org_id=ORGANISATION_ID, user_id='5678')


def test_manage_org_users_shows_no_link_for_cancelled_users(
    client_request,
    mock_get_organisation,
    mock_get_users_for_organisation,
    sample_org_invite,
    mocker,
):
    sample_org_invite['status'] = 'cancelled'
    mocker.patch('app.models.user.OrganisationInvitedUsers.client_method', return_value=[sample_org_invite])

    page = client_request.get(
        '.manage_org_users',
        org_id=ORGANISATION_ID,
    )
    users = page.find_all(class_='user-list-item')

    assert normalize_spaces(users[0].text) == 'invited_user@test.gsa.gov (cancelled invite)'
    assert not users[0].a


@pytest.mark.parametrize('number_of_users', (
    pytest.param(7, marks=pytest.mark.xfail),
    pytest.param(8),
))
def test_manage_org_users_should_show_live_search_if_more_than_7_users(
    client_request,
    mocker,
    mock_get_organisation,
    active_user_with_permissions,
    number_of_users,
):
    mocker.patch(
        'app.models.user.OrganisationInvitedUsers.client_method',
        return_value=[],
    )
    mocker.patch(
        'app.models.user.OrganisationUsers.client_method',
        return_value=[active_user_with_permissions] * number_of_users,
    )

    page = client_request.get(
        '.manage_org_users',
        org_id=ORGANISATION_ID,
    )

    assert page.select_one('div[data-module=live-search]')['data-targets'] == (
        ".user-list-item"
    )
    assert len(page.select('.user-list-item')) == number_of_users

    textbox = page.select_one('[data-module=autofocus] .govuk-input')
    assert 'value' not in textbox
    assert textbox['name'] == 'search'
    # data-module=autofocus is set on a containing element so it
    # shouldn’t also be set on the textbox itself
    assert 'data-module' not in textbox
    assert not page.select_one('[data-force-focus]')
    assert textbox['class'] == [
        'govuk-input', 'govuk-!-width-full',
    ]
    assert normalize_spaces(
        page.select_one('label[for=search]').text
    ) == 'Search by name or email address'


def test_edit_organisation_user_shows_the_delete_confirmation_banner(
    client_request,
    mock_get_organisation,
    mock_get_invites_for_organisation,
    mock_get_users_for_organisation,
    active_user_with_permissions,
):
    page = client_request.get(
        '.edit_organisation_user',
        org_id=ORGANISATION_ID,
        user_id=active_user_with_permissions['id']
    )

    assert normalize_spaces(page.h1) == 'Team members'

    banner = page.select_one('.banner-dangerous')
    assert "Are you sure you want to remove Test User?" in normalize_spaces(banner.contents[0])
    assert banner.form.attrs['action'] == url_for(
        'main.remove_user_from_organisation',
        org_id=ORGANISATION_ID,
        user_id=active_user_with_permissions['id']
    )


def test_remove_user_from_organisation_makes_api_request_to_remove_user(
    client_request,
    mocker,
    mock_get_organisation,
    fake_uuid,
):
    mock_remove_user = mocker.patch('app.organisations_client.remove_user_from_organisation')

    client_request.post(
        '.remove_user_from_organisation',
        org_id=ORGANISATION_ID,
        user_id=fake_uuid,
        _expected_redirect=url_for(
            'main.show_accounts_or_dashboard',
        ),
    )

    mock_remove_user.assert_called_with(ORGANISATION_ID, fake_uuid)


def test_organisation_settings_platform_admin_only(
    client_request,
    mock_get_organisation,
    organisation_one
):
    client_request.get(
        '.organisation_settings',
        org_id=organisation_one['id'],
        _expected_status=403,
    )


def test_organisation_settings_for_platform_admin(
    client_request,
    platform_admin_user,
    mock_get_organisation,
    organisation_one
):
    expected_rows = [
        'Label Value Action',
        'Name Test organisation Change organisation name',
        'Sector Federal government Change sector for the organisation',
        'Crown organisation Yes Change organisation crown status',
        (
            'Data sharing and financial agreement '
            'Not signed Change data sharing and financial agreement for the organisation'
        ),
        'Request to go live notes None Change go live notes for the organisation',
        'Billing details None Change billing details for the organisation',
        'Notes None Change the notes for the organisation',
        'Default email branding GOV.UK Change default email branding for the organisation',
        # 'Default letter branding No branding Change default letter branding for the organisation',
        'Known email domains None Change known email domains for the organisation',
    ]

    client_request.login(platform_admin_user)
    page = client_request.get('.organisation_settings', org_id=organisation_one['id'])

    assert page.find('h1').text == 'Settings'
    rows = page.select('tr')
    assert len(rows) == len(expected_rows)
    for index, row in enumerate(expected_rows):
        assert row == " ".join(rows[index].text.split())
    mock_get_organisation.assert_called_with(organisation_one['id'])


@pytest.mark.parametrize('endpoint, expected_options, expected_selected', (
    (
        '.edit_organisation_type',
        (
            {'value': 'federal', 'label': 'Federal government'},
            {'value': 'state', 'label': 'State government'},
            {'value': 'other', 'label': 'Other'},
        ),
        'federal',
    ),
    (
        '.edit_organisation_crown_status',
        (
            {'value': 'crown', 'label': 'Yes'},
            {'value': 'non-crown', 'label': 'No'},
            {'value': 'unknown', 'label': 'Not sure'},
        ),
        'crown',
    ),
    (
        '.edit_organisation_agreement',
        (
            {
                'value': 'yes',
                'label': 'Yes',
                'hint': 'Users will be told their organisation has already signed the agreement'
            },
            {
                'value': 'no',
                'label': 'No',
                'hint': 'Users will be prompted to sign the agreement before they can go live'
            },
            {
                'value': 'unknown',
                'label': 'No (but we have some service-specific agreements in place)',
                'hint': 'Users will not be prompted to sign the agreement'
            },
        ),
        'no',
    ),
))
@pytest.mark.parametrize('user', (
    pytest.param(
        create_platform_admin_user(),
    ),
    pytest.param(
        create_active_user_with_permissions(),
        marks=pytest.mark.xfail
    ),
))
def test_view_organisation_settings(
    client_request,
    fake_uuid,
    organisation_one,
    mock_get_organisation,
    endpoint,
    expected_options,
    expected_selected,
    user,
):
    client_request.login(user)

    page = client_request.get(endpoint, org_id=organisation_one['id'])

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

    for index, option in enumerate(expected_options):
        option_values = {
            'value': radios[index]['value'],
            'label': normalize_spaces(page.select_one('label[for={}]'.format(radios[index]['id'])).text)
        }
        if 'hint' in option:
            option_values['hint'] = normalize_spaces(
                page.select_one('label[for={}] + .govuk-hint'.format(radios[index]['id'])).text)
        assert option_values == option

    if expected_selected:
        assert page.select_one('input[checked]')['value'] == expected_selected
    else:
        assert not page.select_one('input[checked]')


@pytest.mark.parametrize('endpoint, post_data, expected_persisted', (
    (
        '.edit_organisation_type',
        {'organisation_type': 'federal'},
        {'cached_service_ids': [], 'organisation_type': 'federal'},
    ),
    (
        '.edit_organisation_type',
        {'organisation_type': 'state'},
        {'cached_service_ids': [], 'organisation_type': 'state'},
    ),
    (
        '.edit_organisation_crown_status',
        {'crown_status': 'crown'},
        {'crown': True},
    ),
    (
        '.edit_organisation_crown_status',
        {'crown_status': 'non-crown'},
        {'crown': False},
    ),
    (
        '.edit_organisation_crown_status',
        {'crown_status': 'unknown'},
        {'crown': None},
    ),
    (
        '.edit_organisation_agreement',
        {'agreement_signed': 'yes'},
        {'agreement_signed': True},
    ),
    (
        '.edit_organisation_agreement',
        {'agreement_signed': 'no'},
        {'agreement_signed': False},
    ),
    (
        '.edit_organisation_agreement',
        {'agreement_signed': 'unknown'},
        {'agreement_signed': None},
    ),
))
@pytest.mark.parametrize('user', (
    pytest.param(
        create_platform_admin_user(),
    ),
    pytest.param(
        create_active_user_with_permissions(),
        marks=pytest.mark.xfail
    ),
))
def test_update_organisation_settings(
    mocker,
    client_request,
    fake_uuid,
    organisation_one,
    mock_get_organisation,
    mock_update_organisation,
    endpoint,
    post_data,
    expected_persisted,
    user,
):
    mocker.patch('app.organisations_client.get_organisation_services', return_value=[])
    client_request.login(user)

    client_request.post(
        endpoint,
        org_id=organisation_one['id'],
        _data=post_data,
        _expected_status=302,
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )

    mock_update_organisation.assert_called_once_with(
        organisation_one['id'],
        **expected_persisted,
    )


def test_update_organisation_sector_sends_service_id_data_to_api_client(
    client_request,
    mock_get_organisation,
    organisation_one,
    mock_get_organisation_services,
    mock_update_organisation,
    platform_admin_user,
):
    client_request.login(platform_admin_user)

    client_request.post(
        'main.edit_organisation_type',
        org_id=organisation_one['id'],
        _data={'organisation_type': 'federal'},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )

    mock_update_organisation.assert_called_once_with(
        organisation_one['id'],
        cached_service_ids=['12345', '67890', SERVICE_ONE_ID],
        organisation_type='federal'
    )


@pytest.mark.parametrize('user', (
    pytest.param(
        create_platform_admin_user(),
    ),
    pytest.param(
        create_active_user_with_permissions(),
        marks=pytest.mark.xfail
    ),
))
def test_view_organisation_domains(
    mocker,
    client_request,
    fake_uuid,
    user,
):
    client_request.login(user)

    mocker.patch(
        'app.organisations_client.get_organisation',
        side_effect=lambda org_id: organisation_json(
            org_id,
            'Org 1',
            domains=['example.gsa.gov', 'test.example.gsa.gov'],
        )
    )

    page = client_request.get(
        'main.edit_organisation_domains',
        org_id=ORGANISATION_ID,
    )

    assert [textbox.get('value') for textbox in page.select('input[type=text]')] == [
        'example.gsa.gov',
        'test.example.gsa.gov',
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
        None,
    ]


@pytest.mark.parametrize('post_data, expected_persisted', (
    (
        {
            'domains-0': 'example.gsa.gov',
            'domains-2': 'example.gsa.gov',
            'domains-3': 'EXAMPLE.gsa.gov',
            'domains-5': 'test.gsa.gov',
        },
        {
            'domains': [
                'example.gsa.gov',
                'test.gsa.gov',
            ]
        }
    ),
    (
        {
            'domains-0': '',
            'domains-1': '',
            'domains-2': '',
        },
        {
            'domains': []
        }
    ),
))
@pytest.mark.parametrize('user', (
    pytest.param(
        create_platform_admin_user(),
    ),
    pytest.param(
        create_active_user_with_permissions(),
        marks=pytest.mark.xfail
    ),
))
def test_update_organisation_domains(
    client_request,
    fake_uuid,
    organisation_one,
    mock_get_organisation,
    mock_update_organisation,
    post_data,
    expected_persisted,
    user,
):
    client_request.login(user)

    client_request.post(
        'main.edit_organisation_domains',
        org_id=ORGANISATION_ID,
        _data=post_data,
        _expected_status=302,
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )

    mock_update_organisation.assert_called_once_with(
        ORGANISATION_ID,
        **expected_persisted,
    )


def test_update_organisation_domains_when_domain_already_exists(
    mocker,
    client_request,
    fake_uuid,
    organisation_one,
    mock_get_organisation,
):
    user = create_platform_admin_user()
    client_request.login(user)

    mocker.patch('app.organisations_client.update_organisation', side_effect=HTTPError(
        response=mocker.Mock(
            status_code=400,
            json={'result': 'error', 'message': 'Domain already exists'}
        ),
        message="Domain already exists")
    )

    response = client_request.post(
        'main.edit_organisation_domains',
        org_id=ORGANISATION_ID,
        _data={
            'domains': [
                'example.gsa.gov',
            ]
        },
        _expected_status=200,
    )

    assert response.find("div", class_="banner-dangerous").text.strip() == "This domain is already in use"


def test_update_organisation_name(
    client_request,
    platform_admin_user,
    fake_uuid,
    mock_get_organisation,
    mock_update_organisation,
):
    client_request.login(platform_admin_user)
    client_request.post(
        '.edit_organisation_name',
        org_id=fake_uuid,
        _data={'name': 'TestNewOrgName'},
        _expected_redirect=url_for(
            '.organisation_settings',
            org_id=fake_uuid,
        )
    )
    mock_update_organisation.assert_called_once_with(
        fake_uuid,
        name='TestNewOrgName',
        cached_service_ids=None,
    )


@pytest.mark.parametrize('name, error_message', [
    ('', 'Cannot be empty'),
    ('a', 'at least two alphanumeric characters'),
    ('a' * 256, 'Organisation name must be 255 characters or fewer'),
])
def test_update_organisation_with_incorrect_input(
    client_request,
    platform_admin_user,
    organisation_one,
    mock_get_organisation,
    name,
    error_message
):
    client_request.login(platform_admin_user)
    page = client_request.post(
        '.edit_organisation_name',
        org_id=organisation_one['id'],
        _data={'name': name},
        _expected_status=200,
    )
    assert error_message in page.select_one('.govuk-error-message').text


def test_update_organisation_with_non_unique_name(
    client_request,
    platform_admin_user,
    fake_uuid,
    mock_get_organisation,
    mocker,
):
    mocker.patch(
        'app.organisations_client.update_organisation',
        side_effect=HTTPError(
            response=mocker.Mock(
                status_code=400,
                json={'result': 'error', 'message': 'Organisation name already exists'}
            ),
            message='Organisation name already exists',
        )
    )
    client_request.login(platform_admin_user)
    page = client_request.post(
        '.edit_organisation_name',
        org_id=fake_uuid,
        _data={'name': 'TestNewOrgName'},
        _expected_status=200,
    )

    assert 'This organisation name is already in use' in page.select_one('.govuk-error-message').text


def test_get_edit_organisation_go_live_notes_page(
    client_request,
    platform_admin_user,
    mock_get_organisation,
    organisation_one,
):
    client_request.login(platform_admin_user)
    page = client_request.get(
        '.edit_organisation_go_live_notes',
        org_id=organisation_one['id'],
    )
    assert page.find('textarea', id='request_to_go_live_notes')


@pytest.mark.parametrize('input_note,saved_note', [
    ('Needs permission', 'Needs permission'),
    ('  ', None)
])
def test_post_edit_organisation_go_live_notes_updates_go_live_notes(
    client_request,
    platform_admin_user,
    mock_get_organisation,
    mock_update_organisation,
    organisation_one,
    input_note,
    saved_note,
):
    client_request.login(platform_admin_user)
    client_request.post(
        '.edit_organisation_go_live_notes',
        org_id=organisation_one['id'],
        _data={'request_to_go_live_notes': input_note},
        _expected_redirect=url_for(
            '.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )
    mock_update_organisation.assert_called_once_with(
        organisation_one['id'],
        request_to_go_live_notes=saved_note
    )


def test_organisation_settings_links_to_edit_organisation_notes_page(
    mocker,
    mock_get_organisation,
    organisation_one,
    client_request,
    platform_admin_user,
):
    client_request.login(platform_admin_user)
    page = client_request.get(
        '.organisation_settings', org_id=organisation_one['id']
    )
    assert len(page.find_all(
        'a', attrs={'href': '/organisations/{}/settings/notes'.format(organisation_one['id'])}
    )) == 1


def test_view_edit_organisation_notes(
        client_request,
        platform_admin_user,
        organisation_one,
        mock_get_organisation,
):
    client_request.login(platform_admin_user)
    page = client_request.get(
        'main.edit_organisation_notes',
        org_id=organisation_one['id'],
    )
    assert page.select_one('h1').text == "Edit organisation notes"
    assert page.find('label', class_="form-label").text.strip() == "Notes"
    assert page.find('textarea').attrs["name"] == "notes"


def test_update_organisation_notes(
        client_request,
        platform_admin_user,
        organisation_one,
        mock_get_organisation,
        mock_update_organisation,
):
    client_request.login(platform_admin_user)
    client_request.post(
        'main.edit_organisation_notes',
        org_id=organisation_one['id'],
        _data={'notes': "Very fluffy"},
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )
    mock_update_organisation.assert_called_with(
        organisation_one['id'],
        cached_service_ids=None,
        notes="Very fluffy"
    )


def test_update_organisation_notes_errors_when_user_not_platform_admin(
        client_request,
        organisation_one,
        mock_get_organisation,
        mock_update_organisation,
):
    client_request.post(
        'main.edit_organisation_notes',
        org_id=organisation_one['id'],
        _data={'notes': "Very fluffy"},
        _expected_status=403,
    )


def test_update_organisation_notes_doesnt_call_api_when_notes_dont_change(
        client_request,
        platform_admin_user,
        organisation_one,
        mock_update_organisation,
        mocker
):
    mocker.patch('app.organisations_client.get_organisation', return_value=organisation_json(
        id_=organisation_one['id'],
        name="Test Org",
        notes="Very fluffy"
    ))
    client_request.login(platform_admin_user)
    client_request.post(
        'main.edit_organisation_notes',
        org_id=organisation_one['id'],
        _data={'notes': "Very fluffy"},
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        ),
    )
    assert not mock_update_organisation.called


def test_organisation_settings_links_to_edit_organisation_billing_details_page(
    mocker,
    mock_get_organisation,
    organisation_one,
    client_request,
    platform_admin_user,
):
    client_request.login(platform_admin_user)
    page = client_request.get(
        '.organisation_settings', org_id=organisation_one['id']
    )
    assert len(page.find_all(
        'a', attrs={'href': '/organisations/{}/settings/edit-billing-details'.format(organisation_one['id'])}
    )) == 1


def test_view_edit_organisation_billing_details(
        client_request,
        platform_admin_user,
        organisation_one,
        mock_get_organisation,
):
    client_request.login(platform_admin_user)
    page = client_request.get(
        'main.edit_organisation_billing_details',
        org_id=organisation_one['id'],
    )
    assert page.select_one('h1').text == "Edit organisation billing details"
    labels = page.find_all('label', class_="form-label")
    labels_list = [
        'Contact email addresses',
        'Contact names',
        'Reference',
        'Purchase order number',
        'Notes'
    ]
    for label in labels:
        assert label.text.strip() in labels_list
    textbox_names = page.find_all('input', class_='govuk-input govuk-!-width-full')
    names_list = [
        'billing_contact_email_addresses',
        'billing_contact_names',
        'billing_reference',
        'purchase_order_number',
    ]

    for name in textbox_names:
        assert name.attrs["name"] in names_list

    assert page.find('textarea').attrs["name"] == "notes"


def test_update_organisation_billing_details(
        client_request,
        platform_admin_user,
        organisation_one,
        mock_get_organisation,
        mock_update_organisation,
):
    client_request.login(platform_admin_user)
    client_request.post(
        'main.edit_organisation_billing_details',
        org_id=organisation_one['id'],
        _data={
            'billing_contact_email_addresses': 'accounts@fluff.gsa.gov',
            'billing_contact_names': 'Flannellette von Fluff',
            'billing_reference': '',
            'purchase_order_number': 'PO1234',
            'notes': 'very fluffy, give extra allowance'
        },
        _expected_redirect=url_for(
            'main.organisation_settings',
            org_id=organisation_one['id'],
        )
    )
    mock_update_organisation.assert_called_with(
        organisation_one['id'],
        cached_service_ids=None,
        billing_contact_email_addresses='accounts@fluff.gsa.gov',
        billing_contact_names='Flannellette von Fluff',
        billing_reference='',
        purchase_order_number='PO1234',
        notes='very fluffy, give extra allowance'
    )


def test_update_organisation_billing_details_errors_when_user_not_platform_admin(
        client_request,
        organisation_one,
        mock_get_organisation,
        mock_update_organisation,
):
    client_request.post(
        'main.edit_organisation_billing_details',
        org_id=organisation_one['id'],
        _data={'notes': "Very fluffy"},
        _expected_status=403,
    )


def test_organisation_billing_page_not_accessible_if_not_platform_admin(
    client_request,
    mock_get_organisation,
):
    client_request.get(
        '.organisation_billing',
        org_id=ORGANISATION_ID,
        _expected_status=403
    )


@pytest.mark.parametrize('signed_by_id, signed_by_name, expected_signatory', [
    ('1234', None, 'Test User'),
    (None, 'The Org Manager', 'The Org Manager'),
    ('1234', 'The Org Manager', 'The Org Manager'),
])
def test_organisation_billing_page_when_the_agreement_is_signed_by_a_known_person(
    organisation_one,
    client_request,
    api_user_active,
    mocker,
    platform_admin_user,
    signed_by_id,
    signed_by_name,
    expected_signatory,
):
    api_user_active['id'] = '1234'

    organisation_one['agreement_signed'] = True
    organisation_one['agreement_signed_version'] = 2.5
    organisation_one['agreement_signed_by_id'] = signed_by_id
    organisation_one['agreement_signed_on_behalf_of_name'] = signed_by_name
    organisation_one['agreement_signed_at'] = 'Thu, 20 Feb 2020 06:00:00 GMT'

    mocker.patch('app.organisations_client.get_organisation', return_value=organisation_one)

    client_request.login(platform_admin_user)

    mocker.patch('app.user_api_client.get_user', side_effect=[platform_admin_user, api_user_active])

    page = client_request.get(
        '.organisation_billing',
        org_id=ORGANISATION_ID,
    )

    assert page.h1.string == 'Billing'
    assert '2.5 of the U.S. Notify data sharing and financial agreement on 20 February 2020' in normalize_spaces(
        page.text)
    assert f'{expected_signatory} signed' in page.text
    # assert page.select_one('main a')['href'] == url_for('.organisation_download_agreement', org_id=ORGANISATION_ID)


def test_organisation_billing_page_when_the_agreement_is_signed_by_an_unknown_person(
    organisation_one,
    client_request,
    platform_admin_user,
    mocker,
):
    organisation_one['agreement_signed'] = True
    mocker.patch('app.organisations_client.get_organisation', return_value=organisation_one)

    client_request.login(platform_admin_user)
    page = client_request.get(
        '.organisation_billing',
        org_id=ORGANISATION_ID,
    )

    assert page.h1.string == 'Billing'
    assert (f'{organisation_one["name"]} has accepted the U.S. Notify data '
            'sharing and financial agreement.') in page.text
    # assert page.select_one('main a')['href'] == url_for('.organisation_download_agreement', org_id=ORGANISATION_ID)


@pytest.mark.parametrize('agreement_signed, expected_content', [
    (False, 'needs to accept'),
    (None, 'has not accepted'),
])
def test_organisation_billing_page_when_the_agreement_is_not_signed(
    organisation_one,
    client_request,
    platform_admin_user,
    mocker,
    agreement_signed,
    expected_content,
):
    organisation_one['agreement_signed'] = agreement_signed
    mocker.patch('app.organisations_client.get_organisation', return_value=organisation_one)

    client_request.login(platform_admin_user)
    page = client_request.get(
        '.organisation_billing',
        org_id=ORGANISATION_ID,
    )

    assert page.h1.string == 'Billing'
    assert f'{organisation_one["name"]} {expected_content}' in page.text


@pytest.mark.parametrize('crown, expected_status, expected_file_fetched, expected_file_served', (
    # (
    #     True, 200, 'crown.pdf',
    #     'U.S. Notify data sharing and financial agreement.pdf',
    # ),
    # (
    #     False, 200, 'non-crown.pdf',
    #     'U.S. Notify data sharing and financial agreement (non-crown).pdf',
    # ),
    (
        None, 404, None,
        None,
    ),
))
def test_download_organisation_agreement(
    client_request,
    platform_admin_user,
    mocker,
    crown,
    expected_status,
    expected_file_fetched,
    expected_file_served,
):
    mocker.patch(
        'app.models.organisation.organisations_client.get_organisation',
        return_value=organisation_json(
            crown=crown
        )
    )
    mock_get_s3_object = mocker.patch(
        'app.s3_client.s3_mou_client.get_s3_object',
        return_value=MockS3Object(b'foo')
    )

    client_request.login(platform_admin_user)
    response = client_request.get_response(
        'main.organisation_download_agreement',
        org_id=ORGANISATION_ID,
        _expected_status=expected_status,
    )

    if expected_file_served:
        assert response.get_data() == b'foo'
        assert response.headers['Content-Type'] == 'application/pdf'
        assert response.headers['Content-Disposition'] == (
            f'attachment; filename="{expected_file_served}"'
        )
        mock_get_s3_object.assert_called_once_with('test-mou', expected_file_fetched)
    else:
        assert not expected_file_fetched
        assert mock_get_s3_object.called is False