mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-24 01:11:38 -05:00
add GET /organisation and GET /organisation/id endpoints
This commit is contained in:
39
tests/app/organisation/test_rest.py
Normal file
39
tests/app/organisation/test_rest.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from flask import json
|
||||
|
||||
from app.models import Organisation
|
||||
|
||||
from tests import create_authorization_header
|
||||
|
||||
|
||||
def test_get_organisations(notify_api, notify_db, notify_db_session):
|
||||
org1 = Organisation(colour='#FFFFFF', logo='/path/image.png', name='Org1')
|
||||
org2 = Organisation(colour='#000000', logo='/path/other.png', name='Org2')
|
||||
notify_db.session.add_all([org1, org2])
|
||||
notify_db.session.commit()
|
||||
|
||||
with notify_api.test_request_context(), notify_api.test_client() as client:
|
||||
auth_header = create_authorization_header()
|
||||
response = client.get('/organisation', headers=[auth_header])
|
||||
|
||||
assert response.status_code == 200
|
||||
organisations = json.loads(response.get_data(as_text=True))['organisations']
|
||||
assert len(organisations) == 2
|
||||
assert {org['id'] for org in organisations} == {str(org1.id), str(org2.id)}
|
||||
|
||||
|
||||
def test_get_organisation_by_id(notify_api, notify_db, notify_db_session):
|
||||
org = Organisation(colour='#FFFFFF', logo='/path/image.png', name='My Org')
|
||||
notify_db.session.add(org)
|
||||
notify_db.session.commit()
|
||||
|
||||
with notify_api.test_request_context(), notify_api.test_client() as client:
|
||||
auth_header = create_authorization_header()
|
||||
response = client.get('/organisation/{}'.format(org.id), headers=[auth_header])
|
||||
|
||||
assert response.status_code == 200
|
||||
organisation = json.loads(response.get_data(as_text=True))['organisation']
|
||||
assert set(organisation.keys()) == {'colour', 'logo', 'name', 'id'}
|
||||
assert organisation['colour'] == '#FFFFFF'
|
||||
assert organisation['logo'] == '/path/image.png'
|
||||
assert organisation['name'] == 'My Org'
|
||||
assert organisation['id'] == str(org.id)
|
||||
Reference in New Issue
Block a user