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