from flask import json

from tests import create_authorization_header


def test_get_dvla_organisations(client):
    auth_header = create_authorization_header()

    response = client.get('/dvla_organisations', headers=[auth_header])

    assert response.status_code == 200
    dvla_organisations = json.loads(response.get_data(as_text=True))
    assert dvla_organisations['001'] == 'HM Government'
    assert dvla_organisations['500'] == 'Land Registry'