CatalogueRevamping
/
spatial-d4science
14
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move query tests to own file

This commit is contained in:
amercader 2022-09-01 12:00:34 +02:00
parent 95c7b67132
commit 9cb4fb5fb3
3 changed files with 150 additions and 145 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ckanext/spatial/tests/base.py
View File

@ -1,7 +1,5 @@
# -*- coding: utf-8 -*-
import pytest
geojson_examples = {
"point": '{"type":"Point","coordinates":[100.0,0.0]}',
"point_2": '{"type":"Point","coordinates":[20,10]}',

143
ckanext/spatial/tests/test_api.py
View File

@ -1,152 +1,9 @@
import pytest
from ckan.model import Session
from ckan.lib.search import SearchError
import ckan.tests.helpers as helpers
import ckan.tests.factories as factories
from ckanext.spatial.tests.base import SpatialTestBase
extents = {
"nz": '{"type":"Polygon","coordinates":[[[174,-38],[176,-38],[176,-40],[174,-40],[174,-38]]]}',
"ohio": '{"type": "Polygon","coordinates": [[[-84,38],[-84,40],[-80,42],[-80,38],[-84,38]]]}',
"dateline": '{"type":"Polygon","coordinates":[[[169,70],[169,60],[192,60],[192,70],[169,70]]]}',
"dateline2": '{"type":"Polygon","coordinates":[[[170,60],[-170,60],[-170,70],[170,70],[170,60]]]}',
}
@pytest.mark.usefixtures("clean_db", "clean_index", "harvest_setup")
@pytest.mark.ckan_config("ckanext.spatial.search_backend", "solr")
class TestBBoxSearch(SpatialTestBase):
def test_spatial_query(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": self.geojson_examples["point"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-180,-90,180,90"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_outside_bbox(self):
factories.Dataset(
extras=[{"key": "spatial", "value": self.geojson_examples["point"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-10,-20,10,20"}
)
assert result["count"] == 0
def test_spatial_query_wrong_bbox(self):
with pytest.raises(SearchError):
helpers.call_action(
"package_search",
extras={"ext_bbox": "-10,-20,10,a"},
)
def test_spatial_query_nz(self):
dataset = factories.Dataset(extras=[{"key": "spatial", "value": extents["nz"]}])
result = helpers.call_action(
"package_search", extras={"ext_bbox": "56,-54,189,-28"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_nz_wrap(self):
dataset = factories.Dataset(extras=[{"key": "spatial", "value": extents["nz"]}])
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-203,-54,-167,-28"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_ohio(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["ohio"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-110,37,-78,53"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_ohio_wrap(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["ohio"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "258,37,281,51"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_1(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-197,56,-128,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_2(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "162,54,237,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_3(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline2"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-197,56,-128,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_4(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline2"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "162,54,237,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
@pytest.mark.usefixtures(
"with_plugins",

150
ckanext/spatial/tests/test_spatial_search.py Normal file
View File

@ -0,0 +1,150 @@
import pytest
from ckan.lib.search import SearchError
import ckan.tests.helpers as helpers
import ckan.tests.factories as factories
from ckanext.spatial.tests.base import SpatialTestBase
extents = {
"nz": '{"type":"Polygon","coordinates":[[[174,-38],[176,-38],[176,-40],[174,-40],[174,-38]]]}',
"ohio": '{"type": "Polygon","coordinates": [[[-84,38],[-84,40],[-80,42],[-80,38],[-84,38]]]}',
"dateline": '{"type":"Polygon","coordinates":[[[169,70],[169,60],[192,60],[192,70],[169,70]]]}',
"dateline2": '{"type":"Polygon","coordinates":[[[170,60],[-170,60],[-170,70],[170,70],[170,60]]]}',
}
@pytest.mark.usefixtures("clean_db", "clean_index", "harvest_setup")
@pytest.mark.ckan_config("ckanext.spatial.search_backend", "solr")
class TestBBoxSearch(SpatialTestBase):
def test_spatial_query(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": self.geojson_examples["point"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-180,-90,180,90"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_outside_bbox(self):
factories.Dataset(
extras=[{"key": "spatial", "value": self.geojson_examples["point"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-10,-20,10,20"}
)
assert result["count"] == 0
def test_spatial_query_wrong_bbox(self):
with pytest.raises(SearchError):
helpers.call_action(
"package_search",
extras={"ext_bbox": "-10,-20,10,a"},
)
def test_spatial_query_nz(self):
dataset = factories.Dataset(extras=[{"key": "spatial", "value": extents["nz"]}])
result = helpers.call_action(
"package_search", extras={"ext_bbox": "56,-54,189,-28"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_nz_wrap(self):
dataset = factories.Dataset(extras=[{"key": "spatial", "value": extents["nz"]}])
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-203,-54,-167,-28"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_ohio(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["ohio"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-110,37,-78,53"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_ohio_wrap(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["ohio"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "258,37,281,51"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_1(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-197,56,-128,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_2(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "162,54,237,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_3(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline2"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "-197,56,-128,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]
def test_spatial_query_dateline_4(self):
dataset = factories.Dataset(
extras=[{"key": "spatial", "value": extents["dateline2"]}]
)
result = helpers.call_action(
"package_search", extras={"ext_bbox": "162,54,237,70"}
)
assert result["count"] == 1
assert result["results"][0]["id"] == dataset["id"]