darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-08 08:28:15 -04:00
Code Issues Packages Projects Releases Wiki Activity

notify-api-412 use black to enforce python coding style

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-25 09:12:23 -07:00
parent c6eb007386
commit 8c9721d8e2
201 changed files with 31660 additions and 28105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

179
tests/app/utils/test_branding.py
View File

@@ -8,48 +8,58 @@ 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', []),
])
@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_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')
@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,
@@ -57,37 +67,52 @@ def test_get_email_choices_service_assigned_to_org(
branding_id,
expected_options,
mock_get_service_organization,
mock_get_email_branding
mock_get_email_branding,
):
service = Service(service_one)
mocker.patch(
'app.organizations_client.get_organization',
return_value=organization_json(organization_type=org_type)
"app.organizations_client.get_organization",
return_value=organization_json(organization_type=org_type),
)
mocker.patch(
'app.models.service.Service.email_branding_id',
"app.models.service.Service.email_branding_id",
new_callable=PropertyMock,
return_value=branding_id
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')
@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,
@@ -95,39 +120,47 @@ def test_get_email_choices_org_has_default_branding(
branding_id,
expected_options,
mock_get_service_organization,
mock_get_email_branding
mock_get_email_branding,
):
service = Service(service_one)
mocker.patch(
'app.organizations_client.get_organization',
"app.organizations_client.get_organization",
return_value=organization_json(
organization_type=org_type,
email_branding_id='org-branding-id'
)
organization_type=org_type, email_branding_id="org-branding-id"
),
)
mocker.patch(
'app.models.service.Service.email_branding_id',
"app.models.service.Service.email_branding_id",
new_callable=PropertyMock,
return_value=branding_id
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')
@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,
@@ -138,17 +171,17 @@ def test_get_email_choices_branding_name_in_use(
service = Service(service_one)
mocker.patch(
'app.organizations_client.get_organization',
return_value=organization_json(organization_type='central')
"app.organizations_client.get_organization",
return_value=organization_json(organization_type="central"),
)
mocker.patch(
'app.models.service.Service.email_branding_id',
"app.models.service.Service.email_branding_id",
new_callable=PropertyMock,
return_value='some-branding-id',
return_value="some-branding-id",
)
mocker.patch(
'app.email_branding_client.get_email_branding',
return_value=create_email_branding('_id', {'name': branding_name})
"app.email_branding_client.get_email_branding",
return_value=create_email_branding("_id", {"name": branding_name}),
)
options = get_email_choices(service)