def test_geo(self):
geo_response = CouncilViewSet.as_view({"get": "geo"})(self.request,
pk="ABC")
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="ABC")
# geo_response should contain geometry
self.assertEqual(True, ("geometry" in geo_response.data))
self.assertEqual("MultiPolygon", geo_response.data["geometry"]["type"])
# (non-geo) response should not contain geometry
self.assertEqual(True, ("geometry" not in response.data))
self.assertEqual(response.data["name"],
geo_response.data["properties"]["name"])
def test_geo(self):
geo_response = CouncilViewSet.as_view({'get': 'geo'})(self.request,
pk='X01000001')
response = CouncilViewSet.as_view({'get': 'retrieve'})(self.request,
pk='X01000001')
# geo_response should contain geometry
self.assertEqual(True, ('geometry' in geo_response.data))
self.assertEqual('MultiPolygon', geo_response.data['geometry']['type'])
# (non-geo) response should not contain gemoetry
self.assertEqual(True, ('geometry' not in response.data))
self.assertEqual(response.data['name'],
geo_response.data['properties']['name'])
def test_identifiers_in_api_response(self):
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="X01000001")
self.assertDictEqual(
response.data,
{
"address": "",
"council_id": "X01000001",
"email": "",
"identifiers": ["ABC"],
"name": "X01000001",
"phone": "",
"postcode": "",
"url": "http://testserver/api/beta/councils/X01000001/",
"website": "",
"registration_contacts": None,
"electoral_services_contacts": {
"email": "",
"phone_numbers": [""],
"address": "",
"postcode": "",
"website": "",
},
},
)
def test_redirect_from_identifier(self):
"""
Check that a non-PK ID that is a valid identifier is redurected to
the canonical URL for that instance.
:return:
"""
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="ABC")
self.assertEqual(response.status_code, 301)
self.assertEqual(response.url,
"http://testserver/api/beta/councils/X01000001/")
def test_null_area(self):
response = CouncilViewSet.as_view({"get": "geo"})(self.request,
pk="X01000002")
self.assertEqual(None, response.data["geometry"])
def test_bad_council(self):
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="FOO")
# should return 404 if council does not exist
self.assertEqual(404, response.status_code)
def test_valid_council(self):
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="X01000001")
self.assertEqual(200, response.status_code)
self.assertEqual("X01000001", response.data["council_id"])
def test_list(self):
response = CouncilViewSet.as_view({"get": "list"})(self.request)
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.data))
def test_valid_council(self):
response = CouncilViewSet.as_view({"get": "retrieve"})(self.request,
pk="ABC")
self.assertEqual(200, response.status_code)
self.assertEqual("ABC", response.data["council_id"])
self.assertEqual("England", response.data["nation"])
def test_null_area(self):
response = CouncilViewSet.as_view({'get': 'geo'})(self.request,
pk='X01000002')
self.assertEqual(None, response.data['geometry'])
def test_bad_council(self):
response = CouncilViewSet.as_view({'get': 'retrieve'})(self.request,
pk='FOO')
# should return 404 if council does not exist
self.assertEqual(404, response.status_code)
def test_valid_council(self):
response = CouncilViewSet.as_view({'get': 'retrieve'})(self.request,
pk='X01000001')
self.assertEqual(200, response.status_code)
self.assertEqual('X01000001', response.data['council_id'])