def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
postcode = self.request.GET.get("pc")
lon = self.request.GET.get("lon")
lat = self.request.GET.get("lat")
mapit = MapIt()
context["postcode"] = postcode
context["all_councils"] = Council.objects.all()
try:
if lon and lat:
gss_codes = mapit.wgs84_point_to_gss_codes(lon, lat)
elif postcode:
gss_codes = mapit.postcode_point_to_gss_codes(postcode)
else:
return context
councils = Council.objects.filter(gss_code__in=gss_codes)
combined_authorities = [
council.combined_authority for council in councils
if council.combined_authority
]
context["councils"] = list(councils) + combined_authorities
except (
NotFoundException,
BadRequestException,
InternalServerErrorException,
ForbiddenException,
) as error:
context["error"] = error
return context
def test_invalid_postcode(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"code": 400,
"error": "Postcode invalid",
}
mapit_session.get.return_value.status_code = 400
mapit = MapIt()
with self.assertRaisesMessage(BadRequestException, "Postcode invalid"):
actual = mapit.postcode_point_to_gss_codes(
"BO11AB") # different value to avoid cache
def test_unknown_postcode(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"code": 404,
"error": "Postcode not found",
}
mapit_session.get.return_value.status_code = 404
mapit = MapIt()
with self.assertRaisesMessage(NotFoundException, "Postcode not found"):
actual = mapit.postcode_point_to_gss_codes(
"BO11AC") # different value to avoid cache
def test_wgs84_point_to_gss_codes(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"11111": {
"id": 11111,
"codes": {
"gss": "E14000111",
"unit_id": "11111"
},
"name": "Borsetshire Council",
"country": "E",
"type": "CTY",
}
}
mapit = MapIt()
actual = mapit.wgs84_point_to_gss_codes(-0.132814, 51.501351)
self.assertEqual(actual, ["E14000111"])
def test_postcode_point_to_gss_codes(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"areas": {
"11111": {
"id": 11111,
"codes": {
"gss": "E14000111",
"unit_id": "11111"
},
"name": "Borsetshire Council",
"country": "E",
"type": "CTY",
}
}
}
mapit = MapIt()
actual = mapit.postcode_point_to_gss_codes("BO11AA")
self.assertEqual(actual, ["E14000111"])
def test_mapit_id_to_touches(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"151155": {
"name": "East Garioch",
"id": 151155,
"country": "S",
"type": "UTW",
"codes": {
"unit_id": "27271",
"gss": "S13002859"
},
"generation_low": 31,
"type_name": "Unitary Authority ward (UTW)",
"parent_area": 2648,
"country_name": "Scotland",
"all_names": {},
"generation_high": 40,
},
"151152": {
"name": "Mid Formartine",
"id": 151152,
"country": "S",
"type": "UTW",
"codes": {
"unit_id": "43274",
"gss": "S13002855"
},
"generation_low": 31,
"type_name": "Unitary Authority ward (UTW)",
"parent_area": 2648,
"country_name": "Scotland",
"all_names": {},
"generation_high": 40,
},
}
mapit = MapIt()
actual = mapit.mapit_id_to_touches(2650)
self.assertCountEqual(actual, ["S13002855", "S13002859"])
def test_gss_code_to_mapit_id(self, mapit_session):
mapit_session.get.return_value.json.return_value = {
"id": 2650,
"type": "UTA",
"all_names": {
"O": ["Ordnance Survey", "Aberdeen City"]
},
"name": "Aberdeen City Council",
"country_name": "Scotland",
"generation_high": 40,
"generation_low": 1,
"codes": {
"unit_id": "30421",
"local-authority-canonical": "ABE",
"ons": "00QA",
"gss": "S12000033",
},
"country": "S",
"parent_area": None,
"type_name": "Unitary Authority",
}
mapit = MapIt()
actual = mapit.gss_code_to_mapit_id("S12000033")
self.assertEqual(actual, 2650)
def test_500_error(self, mapit_session):
mapit_session.get.return_value.status_code = 500
mapit = MapIt()
with self.assertRaises(InternalServerErrorException):
mapit.mapit_id_to_touches(2440)
def test_403_error(self, mapit_session):
mapit_session.get.return_value.status_code = 403
mapit = MapIt()
with self.assertRaises(ForbiddenException):
mapit.mapit_id_to_touches(2430)