Refactor test so that it does not have to change every time we add a new organisation.

This commit is contained in:
Rebecca Law
2017-06-30 11:26:17 +01:00
committed by venusbb
parent 1bd5e88979
commit 3e45f7d71f

View File

@@ -10,4 +10,5 @@ def test_get_dvla_organisations(client):
assert response.status_code == 200
dvla_organisations = json.loads(response.get_data(as_text=True))
assert dvla_organisations == {'001': 'HM Government', '500': 'Land Registry'}
assert dvla_organisations['001'] == 'HM Government'
assert dvla_organisations['500'] == 'Land Registry'