darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 23:55:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

ensure old route still returns json in same shape

Browse Source
This commit is contained in:
Leo Hemsted
2018-02-06 12:10:00 +00:00
parent 37a44ab3a9
commit 0c67a61604
2 changed files with 47 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
tests/app/email_branding/test_rest.py
View File

@@ -1,7 +1,11 @@
import json
import pytest
from app.models import EmailBranding
from tests import create_authorization_header
def test_get_email_branding_options(admin_request, notify_db, notify_db_session):
email_branding1 = EmailBranding(colour='#FFFFFF', logo='/path/image.png', name='Org1')
@@ -21,6 +25,29 @@ def test_get_email_branding_options(admin_request, notify_db, notify_db_session)
}
def test_get_email_branding_options_from_old_endpoint(client, notify_db, 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')
notify_db.session.add_all([email_branding1, email_branding2])
notify_db.session.commit()
response = client.get(
'/organisation',
headers=[create_authorization_header()]
)
assert response.status_code == 200
json_resp = json.loads(response.get_data(as_text=True))
email_branding = json_resp['organisations']
assert len(email_branding) == 2
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, notify_db_session):
email_branding = EmailBranding(colour='#FFFFFF', logo='/path/image.png', name='My Org')
notify_db.session.add(email_branding)
@@ -39,6 +66,21 @@ def test_get_email_branding_by_id(admin_request, notify_db, notify_db_session):
assert response['email_branding']['id'] == str(email_branding.id)
def test_get_email_branding_by_id_from_old_endpoint(client, notify_db, notify_db_session):
email_branding = EmailBranding(colour='#FFFFFF', logo='/path/image.png', name='My Org')
notify_db.session.add(email_branding)
notify_db.session.commit()
response = client.get(
'/organisation/{}'.format(email_branding.id),
headers=[create_authorization_header()]
)
assert response.status_code == 200
json_resp = json.loads(response.get_data(as_text=True))
assert json_resp['organisation']['id'] == str(email_branding.id)
def test_post_create_email_branding(admin_request, notify_db_session):
data = {
'name': 'test email_branding',