Tom Kralidis
224 lines
9.6 KiB
Python
224 lines
9.6 KiB
Python
# =================================================================
|
|
#
|
|
# Authors: Just van den Broecke <justb4@gmail.com>
|
|
# Authors: Tom Kralidis <tomkralidis@gmail.com>
|
|
#
|
|
# Copyright (c) 2019 Just van den Broecke
|
|
# Copyright (c) 2022 Tom Kralidis
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person
|
|
# obtaining a copy of this software and associated documentation
|
|
# files (the "Software"), to deal in the Software without
|
|
# restriction, including without limitation the rights to use,
|
|
# copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following
|
|
# conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
# OTHER DEALINGS IN THE SOFTWARE.
|
|
#
|
|
# =================================================================
|
|
|
|
import json
|
|
import logging
|
|
|
|
import pytest
|
|
|
|
from pygeoapi.api import API
|
|
from pygeoapi.util import yaml_load, geojson_to_geom
|
|
from .util import get_test_file_path, mock_request
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
DEFAULT_CRS = 'http://www.opengis.net/def/crs/OGC/1.3/CRS84'
|
|
|
|
|
|
@pytest.fixture()
|
|
def config():
|
|
with open(get_test_file_path('pygeoapi-test-config-ogr.yml')) as fh:
|
|
return yaml_load(fh)
|
|
|
|
|
|
@pytest.fixture()
|
|
def openapi():
|
|
with open(get_test_file_path('pygeoapi-test-openapi.yml')) as fh:
|
|
return yaml_load(fh)
|
|
|
|
|
|
@pytest.fixture()
|
|
def api_(config):
|
|
return API(config, openapi)
|
|
|
|
|
|
def test_get_collection_items_bbox_crs(config, api_):
|
|
CRS_BBOX_DICT = {
|
|
'http://www.opengis.net/def/crs/OGC/1.3/CRS84': '5.71484, 52.12122, 5.71486, 52.12123', # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/4326': '52.12122, 5.71484, 52.12123, 5.71486', # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/28992': '177430, 459268, 177440, 459278' # noqa
|
|
}
|
|
|
|
COLLECTIONS = ['dutch_addresses_4326', 'dutch_addresses_28992']
|
|
for coll in COLLECTIONS:
|
|
# bbox-crs full extent
|
|
req = mock_request({'bbox': '5.670670, 52.042700, 5.829110, 52.123700', 'bbox-crs': 'http://www.opengis.net/def/crs/OGC/1.3/CRS84'}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 10
|
|
|
|
# bbox-crs partial extent, 1 feature, request with multiple CRSs
|
|
for crs in CRS_BBOX_DICT:
|
|
req = mock_request({'bbox': CRS_BBOX_DICT[crs], 'bbox-crs': crs}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 1
|
|
properties = features[0]['properties']
|
|
assert properties['straatnaam'] == 'Willinkhuizersteeg'
|
|
assert properties['huisnummer'] == '2'
|
|
|
|
# bbox-crs outside extent
|
|
req = mock_request({'bbox': '5, 51.9, 5.1, 52.0', 'bbox-crs': 'http://www.opengis.net/def/crs/OGC/1.3/CRS84'}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 0
|
|
|
|
# bbox-crs outside extent
|
|
req = mock_request({'bbox': '130000, 440000, 140000, 450000', 'bbox-crs': 'http://www.opengis.net/def/crs/EPSG/0/28992'}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 0
|
|
|
|
# bbox-crs outside extent - axis reversed CRS
|
|
req = mock_request({'bbox': '51.9, 5, 52.0, 5.1', 'bbox-crs': 'http://www.opengis.net/def/crs/EPSG/0/4326'}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 0
|
|
|
|
# bbox-crs full extent - axis reversed CRS
|
|
req = mock_request({'bbox': '52.042700, 5.670670, 52.123700, 5.829110', 'bbox-crs': 'http://www.opengis.net/def/crs/EPSG/0/4326'}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 10
|
|
|
|
|
|
def test_get_collection_items_crs(config, api_):
|
|
# 'http://www.opengis.net/def/crs/EPSG/0/4258': [52.12122746, 5.714847], # noqa
|
|
CRS_DICT = {
|
|
'none': [5.714847, 52.12122746], # noqa
|
|
'http://www.opengis.net/def/crs/OGC/1.3/CRS84': [5.714847, 52.12122746], # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/28992': [177439, 459274], # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/4326': [52.12122746, 5.714847], # noqa
|
|
}
|
|
# 'http://www.opengis.net/def/crs/EPSG/0/4258': '52.12122, 5.71484, 52.12123, 5.71486', # noqa
|
|
CRS_BBOX_DICT = {
|
|
'none': '5.71484, 52.12122, 5.71486, 52.12123', # noqa
|
|
'http://www.opengis.net/def/crs/OGC/1.3/CRS84': '5.71484, 52.12122, 5.71486, 52.12123', # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/4326': '52.12122, 5.71484, 52.12123, 5.71486', # noqa
|
|
'http://www.opengis.net/def/crs/EPSG/0/28992': '177430, 459268, 177440, 459278' # noqa
|
|
}
|
|
|
|
COLLECTIONS = ['dutch_addresses_4326', 'dutch_addresses_28992']
|
|
for coll in COLLECTIONS:
|
|
# crs full extent to get target feature
|
|
req = mock_request({}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 10
|
|
feature_id = features[0]['id']
|
|
|
|
# request with multiple CRSs
|
|
for crs in CRS_DICT:
|
|
# Do for query (/items)
|
|
req = mock_request({'crs': crs}) # noqa
|
|
if crs == 'none':
|
|
# Test for default bbox CRS
|
|
req = mock_request({}) # noqa
|
|
crs = DEFAULT_CRS
|
|
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 10
|
|
|
|
test_feature = features[0]
|
|
assert test_feature['id'] == feature_id
|
|
|
|
properties = test_feature['properties']
|
|
assert properties['straatnaam'] == 'Willinkhuizersteeg'
|
|
assert properties['huisnummer'] == '2'
|
|
|
|
# Test if CRS in header and the feature coordinates
|
|
# correspond to CRS parameter.
|
|
assert rsp_headers['Content-Crs'] == f'<{crs}>'
|
|
|
|
test_geom_json = test_feature.get('geometry')
|
|
test_geom = geojson_to_geom(test_geom_json)
|
|
crs_geom = geojson_to_geom({'type': 'Point', 'coordinates': CRS_DICT[crs]}) # noqa
|
|
assert test_geom.equals_exact(crs_geom, 1), f'coords not equal for CRS: {crs} {crs_geom} in COLL: {coll} {test_geom}' # noqa
|
|
|
|
# Same for single Feature 'get'
|
|
req = mock_request({'crs': crs}) # noqa
|
|
rsp_headers, code, response = api_.get_collection_item(req, coll, feature_id) # noqa
|
|
test_feature = json.loads(response)
|
|
|
|
assert test_feature['id'] == feature_id
|
|
|
|
properties = test_feature['properties']
|
|
assert properties['straatnaam'] == 'Willinkhuizersteeg'
|
|
assert properties['huisnummer'] == '2'
|
|
|
|
# Test if CRS in header and the feature coordinates
|
|
# correspond to CRS parameter.
|
|
assert rsp_headers['Content-Crs'] == f'<{crs}>'
|
|
|
|
test_geom_json = test_feature.get('geometry')
|
|
test_geom = geojson_to_geom(test_geom_json)
|
|
crs_geom = geojson_to_geom({'type': 'Point', 'coordinates': CRS_DICT[crs]}) # noqa
|
|
assert test_geom.equals_exact(crs_geom, 1), f'coords not equal for CRS: {crs} {crs_geom} in COLL: {coll} {test_geom}' # noqa
|
|
|
|
# Test combining BBOX and BBOX-CRS
|
|
for bbox_crs in CRS_BBOX_DICT:
|
|
# Do for query (/items)
|
|
req = mock_request({'crs': crs, 'bbox': CRS_BBOX_DICT[bbox_crs], 'bbox-crs': bbox_crs}) # noqa
|
|
if bbox_crs == 'none':
|
|
# Test for default bbox CRS
|
|
req = mock_request({'crs': crs, 'bbox': CRS_BBOX_DICT[bbox_crs]}) # noqa
|
|
bbox_crs = DEFAULT_CRS
|
|
|
|
rsp_headers, code, response = api_.get_collection_items(req, coll) # noqa
|
|
features = json.loads(response)['features']
|
|
|
|
assert len(features) == 1
|
|
|
|
test_feature = features[0]
|
|
assert test_feature['id'] == feature_id
|
|
|
|
properties = test_feature['properties']
|
|
assert properties['straatnaam'] == 'Willinkhuizersteeg'
|
|
assert properties['huisnummer'] == '2'
|
|
|
|
# Test if CRS in header and the feature coordinates
|
|
# correspond to CRS parameter.
|
|
assert rsp_headers['Content-Crs'] == f'<{crs}>'
|
|
|
|
test_geom_json = test_feature.get('geometry')
|
|
test_geom = geojson_to_geom(test_geom_json)
|
|
crs_geom = geojson_to_geom({'type': 'Point', 'coordinates': CRS_DICT[crs]}) # noqa
|
|
assert test_geom.equals_exact(crs_geom, 1), f'coords not equal for CRS: {crs} {crs_geom} in COLL: {coll} {test_geom} bbox-crs={bbox_crs}' # noqa
|