2025-11-25 15:18:37 +01:00

82 lines
2.7 KiB
Python

import pytest
from geoservercloud import GeoServerCloud
from conftest import GEOSERVER_URL
WORKSPACE = "cog"
@pytest.fixture(scope="function")
def geoserver():
geoserver = GeoServerCloud(url=GEOSERVER_URL)
geoserver.create_workspace(WORKSPACE, set_default_workspace=True)
yield geoserver
geoserver.delete_workspace(WORKSPACE)
def test_create_cog_coverage(geoserver):
"""Test creating a COG coverage store and coverage"""
store_name = "land_shallow_topo_21600_NW_cog"
coverage_name = "land_shallow_topo_NW"
# Create COG coverage store
store_xml = f"""<coverageStore>
<name>{store_name}</name>
<type>GeoTIFF</type>
<enabled>true</enabled>
<workspace><name>{WORKSPACE}</name></workspace>
<url>cog://https://test-data-cog-public.s3.amazonaws.com/public/land_shallow_topo_21600_NW_cog.tif</url>
<metadata>
<entry key="CogSettings.Key">
<cogSettings>
<rangeReaderSettings>HTTP</rangeReaderSettings>
</cogSettings>
</entry>
</metadata>
</coverageStore>"""
rest_client = geoserver.rest_service.rest_client
endpoints = geoserver.rest_service.rest_endpoints
response = rest_client.post(
endpoints.coveragestores(WORKSPACE),
data=store_xml,
headers={"Content-Type": "application/xml"},
)
assert response.status_code == 201
# Create coverage
coverage_xml = f"""<coverage>
<name>{coverage_name}</name>
<nativeName>{store_name}</nativeName>
</coverage>"""
response = rest_client.post(
endpoints.coverages(WORKSPACE, store_name),
data=coverage_xml,
headers={"Content-Type": "application/xml"},
)
assert response.status_code == 201
# Verify the coverage was created - try listing coverages first
list_response = rest_client.get(endpoints.coverages(WORKSPACE, store_name))
assert (
list_response.status_code == 200
), f"Failed to get coverages: {list_response.status_code} - {list_response.text}"
# Check specific coverage
response = rest_client.get(endpoints.coverage(WORKSPACE, store_name, coverage_name))
assert response.status_code == 200
# Verify coverage properties
coverage_data = response.json()["coverage"]
assert coverage_data["name"] == coverage_name
assert coverage_data["nativeName"] == coverage_name
assert coverage_data["enabled"] == True
# Test WMS GetMap request
wms_response = rest_client.get(
f"/wms?SERVICE=WMS&VERSION=1.1.0&REQUEST=GetMap&LAYERS={WORKSPACE}:{coverage_name}&STYLES=&BBOX=-180,-90,180,90&WIDTH=256&HEIGHT=256&FORMAT=image/jpeg&SRS=EPSG:4326"
)
assert wms_response.status_code == 200
assert wms_response.headers.get("content-type").startswith("image/jpeg")