notifications-admin/tests/app/utils/test_branding.py

from unittest.mock import PropertyMock

import pytest

from app.models.service import Service
from app.utils.branding import get_email_choices
from tests import organization_json
from tests.conftest import create_email_branding


@pytest.mark.parametrize('function', [get_email_choices])
@pytest.mark.parametrize('org_type, expected_options', [
    ('federal', []),
    ('state', []),
])
def test_get_choices_service_not_assigned_to_org(
    service_one,
    function,
    org_type,
    expected_options,
):
    service_one['organization_type'] = org_type
    service = Service(service_one)

    options = function(service)
    assert list(options) == expected_options


@pytest.mark.parametrize('org_type, branding_id, expected_options', [
    ('federal', None, [
        ('govuk_and_org', 'GOV.UK and Test Organization'),
        ('organization', 'Test Organization'),
    ]),
    ('federal', 'some-branding-id', [
        ('govuk', 'GOV.UK'),  # central orgs can switch back to gsa.gov
        ('govuk_and_org', 'GOV.UK and Test Organization'),
        ('organization', 'Test Organization'),
    ]),
    ('state', None, [
        ('organization', 'Test Organization')
    ]),
    ('state', 'some-branding-id', [
        ('organization', 'Test Organization')
    ]),
    # ('nhs_central', None, [
    #     ('nhs', 'NHS')
    # ]),
    # ('nhs_central', NHS_EMAIL_BRANDING_ID, [
    #     # don't show NHS if it's the current branding
    # ]),
])
@pytest.mark.skip(reason='Update for TTS')
def test_get_email_choices_service_assigned_to_org(
    mocker,
    service_one,
    org_type,
    branding_id,
    expected_options,
    mock_get_service_organization,
    mock_get_email_branding
):
    service = Service(service_one)

    mocker.patch(
        'app.organizations_client.get_organization',
        return_value=organization_json(organization_type=org_type)
    )
    mocker.patch(
        'app.models.service.Service.email_branding_id',
        new_callable=PropertyMock,
        return_value=branding_id
    )

    options = get_email_choices(service)
    assert list(options) == expected_options


@pytest.mark.parametrize('org_type, branding_id, expected_options', [
    ('federal', 'some-branding-id', [
        # don't show gsa.gov options as org default supersedes it
        ('organization', 'Test Organization'),
    ]),
    ('federal', 'org-branding-id', [
        # also don't show org option if it's the current branding
    ]),
    ('state', 'org-branding-id', [
        # don't show org option if it's the current branding
    ]),
])
@pytest.mark.skip(reason='Update for TTS')
def test_get_email_choices_org_has_default_branding(
    mocker,
    service_one,
    org_type,
    branding_id,
    expected_options,
    mock_get_service_organization,
    mock_get_email_branding
):
    service = Service(service_one)

    mocker.patch(
        'app.organizations_client.get_organization',
        return_value=organization_json(
            organization_type=org_type,
            email_branding_id='org-branding-id'
        )
    )
    mocker.patch(
        'app.models.service.Service.email_branding_id',
        new_callable=PropertyMock,
        return_value=branding_id
    )

    options = get_email_choices(service)
    assert list(options) == expected_options


@pytest.mark.parametrize('branding_name, expected_options', [
    ('gsa.gov and something else', [
        ('govuk', 'GOV.UK'),
        ('govuk_and_org', 'GOV.UK and Test Organization'),
        ('organization', 'Test Organization'),
    ]),
    ('gsa.gov and test OrganisatioN', [
        ('govuk', 'GOV.UK'),
        ('organization', 'Test Organization'),
    ])
])
@pytest.mark.skip(reason='Update for TTS')
def test_get_email_choices_branding_name_in_use(
    mocker,
    service_one,
    branding_name,
    expected_options,
    mock_get_service_organization,
):
    service = Service(service_one)

    mocker.patch(
        'app.organizations_client.get_organization',
        return_value=organization_json(organization_type='central')
    )
    mocker.patch(
        'app.models.service.Service.email_branding_id',
        new_callable=PropertyMock,
        return_value='some-branding-id',
    )
    mocker.patch(
        'app.email_branding_client.get_email_branding',
        return_value=create_email_branding('_id', {'name': branding_name})
    )

    options = get_email_choices(service)
    # don't show option if its name is similar to current branding
    assert list(options) == expected_options