From 3e45f7d71fbd154a1039836228098efb62457f1b Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Fri, 30 Jun 2017 11:26:17 +0100 Subject: [PATCH] Refactor test so that it does not have to change every time we add a new organisation. --- tests/app/dvla_organisation/test_rest.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/app/dvla_organisation/test_rest.py b/tests/app/dvla_organisation/test_rest.py index e40c843bf..8bf4f71ab 100644 --- a/tests/app/dvla_organisation/test_rest.py +++ b/tests/app/dvla_organisation/test_rest.py @@ -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'