mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-04 18:31:13 -05:00
reformat
This commit is contained in:
@@ -5,199 +5,196 @@ from tests.app.db import create_email_branding
|
||||
|
||||
|
||||
def test_get_email_branding_options(admin_request, notify_db_session):
|
||||
email_branding1 = EmailBranding(colour='#FFFFFF', logo='/path/image.png', name='Org1')
|
||||
email_branding2 = EmailBranding(colour='#000000', logo='/path/other.png', name='Org2')
|
||||
email_branding1 = EmailBranding(
|
||||
colour="#FFFFFF", logo="/path/image.png", name="Org1"
|
||||
)
|
||||
email_branding2 = EmailBranding(
|
||||
colour="#000000", logo="/path/other.png", name="Org2"
|
||||
)
|
||||
notify_db_session.add_all([email_branding1, email_branding2])
|
||||
notify_db_session.commit()
|
||||
|
||||
email_branding = admin_request.get(
|
||||
'email_branding.get_email_branding_options'
|
||||
)['email_branding']
|
||||
email_branding = admin_request.get("email_branding.get_email_branding_options")[
|
||||
"email_branding"
|
||||
]
|
||||
|
||||
assert len(email_branding) == 2
|
||||
assert {
|
||||
email_branding['id'] for email_branding in email_branding
|
||||
} == {
|
||||
str(email_branding1.id), str(email_branding2.id)
|
||||
assert {email_branding["id"] for email_branding in email_branding} == {
|
||||
str(email_branding1.id),
|
||||
str(email_branding2.id),
|
||||
}
|
||||
|
||||
|
||||
def test_get_email_branding_by_id(admin_request, notify_db_session):
|
||||
email_branding = EmailBranding(colour='#FFFFFF', logo='/path/image.png', name='Some Org', text='My Org')
|
||||
email_branding = EmailBranding(
|
||||
colour="#FFFFFF", logo="/path/image.png", name="Some Org", text="My Org"
|
||||
)
|
||||
notify_db_session.add(email_branding)
|
||||
notify_db_session.commit()
|
||||
|
||||
response = admin_request.get(
|
||||
'email_branding.get_email_branding_by_id',
|
||||
"email_branding.get_email_branding_by_id",
|
||||
_expected_status=200,
|
||||
email_branding_id=email_branding.id
|
||||
email_branding_id=email_branding.id,
|
||||
)
|
||||
|
||||
assert set(response['email_branding'].keys()) == {'colour', 'logo', 'name', 'id', 'text',
|
||||
'brand_type'}
|
||||
assert response['email_branding']['colour'] == '#FFFFFF'
|
||||
assert response['email_branding']['logo'] == '/path/image.png'
|
||||
assert response['email_branding']['name'] == 'Some Org'
|
||||
assert response['email_branding']['text'] == 'My Org'
|
||||
assert response['email_branding']['id'] == str(email_branding.id)
|
||||
assert response['email_branding']['brand_type'] == str(email_branding.brand_type)
|
||||
assert set(response["email_branding"].keys()) == {
|
||||
"colour",
|
||||
"logo",
|
||||
"name",
|
||||
"id",
|
||||
"text",
|
||||
"brand_type",
|
||||
}
|
||||
assert response["email_branding"]["colour"] == "#FFFFFF"
|
||||
assert response["email_branding"]["logo"] == "/path/image.png"
|
||||
assert response["email_branding"]["name"] == "Some Org"
|
||||
assert response["email_branding"]["text"] == "My Org"
|
||||
assert response["email_branding"]["id"] == str(email_branding.id)
|
||||
assert response["email_branding"]["brand_type"] == str(email_branding.brand_type)
|
||||
|
||||
|
||||
def test_post_create_email_branding(admin_request, notify_db_session):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'colour': '#0000ff',
|
||||
'logo': '/images/test_x2.png',
|
||||
'brand_type': BRANDING_ORG
|
||||
"name": "test email_branding",
|
||||
"colour": "#0000ff",
|
||||
"logo": "/images/test_x2.png",
|
||||
"brand_type": BRANDING_ORG,
|
||||
}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
assert data['name'] == response['data']['name']
|
||||
assert data['colour'] == response['data']['colour']
|
||||
assert data['logo'] == response['data']['logo']
|
||||
assert data['name'] == response['data']['text']
|
||||
assert data['brand_type'] == response['data']['brand_type']
|
||||
assert data["name"] == response["data"]["name"]
|
||||
assert data["colour"] == response["data"]["colour"]
|
||||
assert data["logo"] == response["data"]["logo"]
|
||||
assert data["name"] == response["data"]["text"]
|
||||
assert data["brand_type"] == response["data"]["brand_type"]
|
||||
|
||||
|
||||
def test_post_create_email_branding_without_brand_type_defaults(admin_request, notify_db_session):
|
||||
def test_post_create_email_branding_without_brand_type_defaults(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'colour': '#0000ff',
|
||||
'logo': '/images/test_x2.png',
|
||||
"name": "test email_branding",
|
||||
"colour": "#0000ff",
|
||||
"logo": "/images/test_x2.png",
|
||||
}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
assert BRANDING_ORG == response['data']['brand_type']
|
||||
assert BRANDING_ORG == response["data"]["brand_type"]
|
||||
|
||||
|
||||
def test_post_create_email_branding_without_logo_is_ok(admin_request, notify_db_session):
|
||||
def test_post_create_email_branding_without_logo_is_ok(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'colour': '#0000ff',
|
||||
"name": "test email_branding",
|
||||
"colour": "#0000ff",
|
||||
}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
"email_branding.create_email_branding",
|
||||
_data=data,
|
||||
_expected_status=201,
|
||||
)
|
||||
assert not response['data']['logo']
|
||||
assert not response["data"]["logo"]
|
||||
|
||||
|
||||
def test_post_create_email_branding_colour_is_valid(admin_request, notify_db_session):
|
||||
data = {
|
||||
'logo': 'images/text_x2.png',
|
||||
'name': 'test branding'
|
||||
}
|
||||
data = {"logo": "images/text_x2.png", "name": "test branding"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
assert response['data']['logo'] == data['logo']
|
||||
assert response['data']['name'] == 'test branding'
|
||||
assert response['data']['colour'] is None
|
||||
assert response['data']['text'] == 'test branding'
|
||||
assert response["data"]["logo"] == data["logo"]
|
||||
assert response["data"]["name"] == "test branding"
|
||||
assert response["data"]["colour"] is None
|
||||
assert response["data"]["text"] == "test branding"
|
||||
|
||||
|
||||
def test_post_create_email_branding_with_text(admin_request, notify_db_session):
|
||||
data = {
|
||||
'text': 'text for brand',
|
||||
'logo': 'images/text_x2.png',
|
||||
'name': 'test branding'
|
||||
"text": "text for brand",
|
||||
"logo": "images/text_x2.png",
|
||||
"name": "test branding",
|
||||
}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
assert response['data']['logo'] == data['logo']
|
||||
assert response['data']['name'] == 'test branding'
|
||||
assert response['data']['colour'] is None
|
||||
assert response['data']['text'] == 'text for brand'
|
||||
assert response["data"]["logo"] == data["logo"]
|
||||
assert response["data"]["name"] == "test branding"
|
||||
assert response["data"]["colour"] is None
|
||||
assert response["data"]["text"] == "text for brand"
|
||||
|
||||
|
||||
def test_post_create_email_branding_with_text_and_name(admin_request, notify_db_session):
|
||||
def test_post_create_email_branding_with_text_and_name(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
data = {
|
||||
'name': 'name for brand',
|
||||
'text': 'text for brand',
|
||||
'logo': 'images/text_x2.png'
|
||||
"name": "name for brand",
|
||||
"text": "text for brand",
|
||||
"logo": "images/text_x2.png",
|
||||
}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
assert response['data']['logo'] == data['logo']
|
||||
assert response['data']['name'] == 'name for brand'
|
||||
assert response['data']['colour'] is None
|
||||
assert response['data']['text'] == 'text for brand'
|
||||
assert response["data"]["logo"] == data["logo"]
|
||||
assert response["data"]["name"] == "name for brand"
|
||||
assert response["data"]["colour"] is None
|
||||
assert response["data"]["text"] == "text for brand"
|
||||
|
||||
|
||||
def test_post_create_email_branding_with_text_as_none_and_name(admin_request, notify_db_session):
|
||||
data = {
|
||||
'name': 'name for brand',
|
||||
'text': None,
|
||||
'logo': 'images/text_x2.png'
|
||||
}
|
||||
def test_post_create_email_branding_with_text_as_none_and_name(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
data = {"name": "name for brand", "text": None, "logo": "images/text_x2.png"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
assert response['data']['logo'] == data['logo']
|
||||
assert response['data']['name'] == 'name for brand'
|
||||
assert response['data']['colour'] is None
|
||||
assert response['data']['text'] is None
|
||||
assert response["data"]["logo"] == data["logo"]
|
||||
assert response["data"]["name"] == "name for brand"
|
||||
assert response["data"]["colour"] is None
|
||||
assert response["data"]["text"] is None
|
||||
|
||||
|
||||
def test_post_create_email_branding_returns_400_when_name_is_missing(admin_request, notify_db_session):
|
||||
data = {
|
||||
'text': 'some text',
|
||||
'logo': 'images/text_x2.png'
|
||||
}
|
||||
def test_post_create_email_branding_returns_400_when_name_is_missing(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
data = {"text": "some text", "logo": "images/text_x2.png"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=400
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=400
|
||||
)
|
||||
|
||||
assert response['errors'][0]['message'] == 'name is a required property'
|
||||
assert response["errors"][0]["message"] == "name is a required property"
|
||||
|
||||
|
||||
@pytest.mark.parametrize('data_update', [
|
||||
({'name': 'test email_branding 1'}),
|
||||
({'logo': 'images/text_x3.png', 'colour': '#ffffff'}),
|
||||
({'logo': 'images/text_x3.png'}),
|
||||
({'logo': 'images/text_x3.png'}),
|
||||
({'logo': 'images/text_x3.png'}),
|
||||
])
|
||||
def test_post_update_email_branding_updates_field(admin_request, notify_db_session, data_update):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'logo': 'images/text_x2.png'
|
||||
}
|
||||
@pytest.mark.parametrize(
|
||||
"data_update",
|
||||
[
|
||||
({"name": "test email_branding 1"}),
|
||||
({"logo": "images/text_x3.png", "colour": "#ffffff"}),
|
||||
({"logo": "images/text_x3.png"}),
|
||||
({"logo": "images/text_x3.png"}),
|
||||
({"logo": "images/text_x3.png"}),
|
||||
],
|
||||
)
|
||||
def test_post_update_email_branding_updates_field(
|
||||
admin_request, notify_db_session, data_update
|
||||
):
|
||||
data = {"name": "test email_branding", "logo": "images/text_x2.png"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
email_branding_id = response['data']['id']
|
||||
email_branding_id = response["data"]["id"]
|
||||
|
||||
admin_request.post(
|
||||
'email_branding.update_email_branding',
|
||||
"email_branding.update_email_branding",
|
||||
_data=data_update,
|
||||
email_branding_id=email_branding_id
|
||||
email_branding_id=email_branding_id,
|
||||
)
|
||||
|
||||
email_branding = EmailBranding.query.all()
|
||||
@@ -209,28 +206,28 @@ def test_post_update_email_branding_updates_field(admin_request, notify_db_sessi
|
||||
assert email_branding[0].text == email_branding[0].name
|
||||
|
||||
|
||||
@pytest.mark.parametrize('data_update', [
|
||||
({'text': 'text email branding'}),
|
||||
({'text': 'new text', 'name': 'new name'}),
|
||||
({'text': None, 'name': 'test name'}),
|
||||
])
|
||||
def test_post_update_email_branding_updates_field_with_text(admin_request, notify_db_session, data_update):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'logo': 'images/text_x2.png'
|
||||
}
|
||||
@pytest.mark.parametrize(
|
||||
"data_update",
|
||||
[
|
||||
({"text": "text email branding"}),
|
||||
({"text": "new text", "name": "new name"}),
|
||||
({"text": None, "name": "test name"}),
|
||||
],
|
||||
)
|
||||
def test_post_update_email_branding_updates_field_with_text(
|
||||
admin_request, notify_db_session, data_update
|
||||
):
|
||||
data = {"name": "test email_branding", "logo": "images/text_x2.png"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=201
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=201
|
||||
)
|
||||
|
||||
email_branding_id = response['data']['id']
|
||||
email_branding_id = response["data"]["id"]
|
||||
|
||||
admin_request.post(
|
||||
'email_branding.update_email_branding',
|
||||
"email_branding.update_email_branding",
|
||||
_data=data_update,
|
||||
email_branding_id=email_branding_id
|
||||
email_branding_id=email_branding_id,
|
||||
)
|
||||
|
||||
email_branding = EmailBranding.query.all()
|
||||
@@ -242,31 +239,30 @@ def test_post_update_email_branding_updates_field_with_text(admin_request, notif
|
||||
|
||||
|
||||
def test_create_email_branding_reject_invalid_brand_type(admin_request):
|
||||
data = {
|
||||
'name': 'test email_branding',
|
||||
'brand_type': 'NOT A TYPE'
|
||||
|
||||
}
|
||||
data = {"name": "test email_branding", "brand_type": "NOT A TYPE"}
|
||||
response = admin_request.post(
|
||||
'email_branding.create_email_branding',
|
||||
_data=data,
|
||||
_expected_status=400
|
||||
"email_branding.create_email_branding", _data=data, _expected_status=400
|
||||
)
|
||||
|
||||
assert response['errors'][0]['message'] == 'brand_type NOT A TYPE is not one of [org, both, org_banner]'
|
||||
assert (
|
||||
response["errors"][0]["message"]
|
||||
== "brand_type NOT A TYPE is not one of [org, both, org_banner]"
|
||||
)
|
||||
|
||||
|
||||
def test_update_email_branding_reject_invalid_brand_type(admin_request, notify_db_session):
|
||||
def test_update_email_branding_reject_invalid_brand_type(
|
||||
admin_request, notify_db_session
|
||||
):
|
||||
email_branding = create_email_branding()
|
||||
data = {
|
||||
'brand_type': 'NOT A TYPE'
|
||||
|
||||
}
|
||||
data = {"brand_type": "NOT A TYPE"}
|
||||
response = admin_request.post(
|
||||
'email_branding.update_email_branding',
|
||||
"email_branding.update_email_branding",
|
||||
_data=data,
|
||||
_expected_status=400,
|
||||
email_branding_id=email_branding.id
|
||||
email_branding_id=email_branding.id,
|
||||
)
|
||||
|
||||
assert response['errors'][0]['message'] == 'brand_type NOT A TYPE is not one of [org, both, org_banner]'
|
||||
assert (
|
||||
response["errors"][0]["message"]
|
||||
== "brand_type NOT A TYPE is not one of [org, both, org_banner]"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user