tests/app/dvla_organisation/test_rest.py

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'
Revert "Revert "add DVLA organisations to API"" 2017-04-21 12:39:42 +01:00			`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))`
Refactor test so that it does not have to change every time we add a new organisation. 2017-06-30 11:26:17 +01:00			`assert dvla_organisations['001'] == 'HM Government'`
			`assert dvla_organisations['500'] == 'Land Registry'`