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)