X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/1e58cef17411d62862516f41b48165e4d7e96372..d8dca2a3a9049db7da8b805042833846a188290d:/test/bdd/steps/http_responses.py diff --git a/test/bdd/steps/http_responses.py b/test/bdd/steps/http_responses.py index ccf247a9..2e24ed50 100644 --- a/test/bdd/steps/http_responses.py +++ b/test/bdd/steps/http_responses.py @@ -84,26 +84,8 @@ class GenericResponse: assert set(r.keys()) == {'geocoding', 'geojson', '__geocoding'}, \ f"Unexpected keys in result: {r.keys()}" check_for_attributes(r['geocoding'], 'geojson', 'absent') - r |= r.pop('geocoding') - - - def assert_subfield(self, idx, path, value): - assert path - - field = self.result[idx] - for p in path: - assert isinstance(field, dict) - assert p in field - field = field[p] - - if isinstance(value, float): - assert Almost(value) == float(field) - elif value.startswith("^"): - assert re.fullmatch(value, field) - elif isinstance(field, dict): - assert field, eval('{' + value + '}') - else: - assert str(field) == str(value) + inner = r.pop('geocoding') + r.update(inner) def assert_address_field(self, idx, field, value): @@ -122,7 +104,7 @@ class GenericResponse: self.check_row_field(idx, field, value, base=address) - def match_row(self, row, context=None): + def match_row(self, row, context=None, field=None): """ Match the result fields against the given behave table row. """ if 'ID' in row.headings: @@ -131,12 +113,20 @@ class GenericResponse: todo = range(len(self.result)) for i in todo: + subdict = self.result[i] + if field is not None: + for key in field.split('.'): + self.check_row(i, key in subdict, f"Missing subfield {key}") + subdict = subdict[key] + self.check_row(i, isinstance(subdict, dict), + f"Subfield {key} not a dict") + for name, value in zip(row.headings, row.cells): if name == 'ID': pass elif name == 'osm': - self.check_row_field(i, 'osm_type', OsmType(value[0])) - self.check_row_field(i, 'osm_id', Field(value[1:])) + self.check_row_field(i, 'osm_type', OsmType(value[0]), base=subdict) + self.check_row_field(i, 'osm_id', Field(value[1:]), base=subdict) elif name == 'centroid': if ' ' in value: lon, lat = value.split(' ') @@ -144,12 +134,10 @@ class GenericResponse: lon, lat = context.osm.grid_node(int(value)) else: raise RuntimeError("Context needed when using grid coordinates") - self.check_row_field(i, 'lat', Field(float(lat))) - self.check_row_field(i, 'lon', Field(float(lon))) - elif '+' in name: - self.assert_subfield(i, name.split('+'), value) + self.check_row_field(i, 'lat', Field(float(lat), abs_tol=1e-07), base=subdict) + self.check_row_field(i, 'lon', Field(float(lon), abs_tol=1e-07), base=subdict) else: - self.check_row_field(i, name, Field(value)) + self.check_row_field(i, name, Field(value), base=subdict) def check_row(self, idx, check, msg): @@ -233,24 +221,33 @@ class ReverseResponse(GenericResponse): if child.tag == 'result': assert not self.result, "More than one result in reverse result" self.result.append(dict(child.attrib)) + check_for_attributes(self.result[0], 'display_name', 'absent') + self.result[0]['display_name'] = child.text elif child.tag == 'addressparts': + assert 'address' not in self.result[0], "More than one address in result" address = {} for sub in child: + assert len(sub) == 0, f"Address element '{sub.tag}' has subelements" address[sub.tag] = sub.text self.result[0]['address'] = address elif child.tag == 'extratags': + assert 'extratags' not in self.result[0], "More than one extratags in result" self.result[0]['extratags'] = {} for tag in child: + assert len(tag) == 0, f"Extratags element '{tag.attrib['key']}' has subelements" self.result[0]['extratags'][tag.attrib['key']] = tag.attrib['value'] elif child.tag == 'namedetails': + assert 'namedetails' not in self.result[0], "More than one namedetails in result" self.result[0]['namedetails'] = {} for tag in child: + assert len(tag) == 0, f"Namedetails element '{tag.attrib['desc']}' has subelements" self.result[0]['namedetails'][tag.attrib['desc']] = tag.text elif child.tag == 'geokml': - self.result[0][child.tag] = True + assert 'geokml' not in self.result[0], "More than one geokml in result" + self.result[0]['geokml'] = ET.tostring(child, encoding='unicode') else: assert child.tag == 'error', \ - "Unknown XML tag {} on page: {}".format(child.tag, self.page) + f"Unknown XML tag {child.tag} on page: {self.page}" class StatusResponse(GenericResponse):