Jens
faeb58ef6d
GeoIntel release gates / Compile, test, contracts and builds (push) Successful in 1m49s
GeoIntel release gates / Python and npm vulnerability policy (push) Successful in 21s
GeoIntel release gates / Production AI image, SBOM and container scan (push) Successful in 5m39s
GeoIntel release gates / Deploy exact gated revision to Unraid (push) Failing after 58m43s
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from types import SimpleNamespace
|
|
from uuid import uuid4
|
|
|
|
from app.models import Project
|
|
from app.services.dataset_service import DatasetService
|
|
|
|
|
|
class FakeUploadFile:
|
|
filename = "real-orthophoto.tif"
|
|
content_type = "image/tiff"
|
|
|
|
def __init__(self) -> None:
|
|
self._content = b"fake-raster"
|
|
|
|
async def read(self, size: int) -> bytes:
|
|
chunk, self._content = self._content[:size], self._content[size:]
|
|
return chunk
|
|
|
|
|
|
class FakeSession:
|
|
def __init__(self, project_id):
|
|
self.project_id = project_id
|
|
self.added = []
|
|
|
|
def get(self, model, item_id):
|
|
if model is Project and item_id == self.project_id:
|
|
return SimpleNamespace(id=item_id)
|
|
return None
|
|
|
|
def add(self, item):
|
|
self.added.append(item)
|
|
|
|
def commit(self):
|
|
return None
|
|
|
|
def refresh(self, _item):
|
|
return None
|
|
|
|
|
|
def test_raster_upload_maps_metadata_bounds_resolution_and_bands(monkeypatch) -> None:
|
|
project_id = uuid4()
|
|
db = FakeSession(project_id)
|
|
|
|
async def persist_upload_file(**_kwargs):
|
|
return {
|
|
"storage_path": "/tmp/real-orthophoto.tif",
|
|
"original_filename": "real-orthophoto.tif",
|
|
"stored_filename": "real-orthophoto.tif",
|
|
"content_type": "image/tiff",
|
|
"size_bytes": 11,
|
|
"checksum_sha256": "checksum",
|
|
}
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.dataset_service.StorageService.persist_upload_file",
|
|
persist_upload_file,
|
|
)
|
|
monkeypatch.setattr(
|
|
"app.services.dataset_service.extract_raster_metadata",
|
|
lambda _path: {
|
|
"driver": "GTiff",
|
|
"crs": "EPSG:31370",
|
|
"bounds": [193277.5, 205708.2, 193777.5, 206208.2],
|
|
"resolution": [0.9765625, 0.9765625],
|
|
"dtype": ["uint8", "uint8", "uint8"],
|
|
},
|
|
)
|
|
|
|
response = asyncio.run(
|
|
DatasetService.upload_dataset(
|
|
db=db,
|
|
project_id=project_id,
|
|
file=FakeUploadFile(),
|
|
dataset_type="raster",
|
|
source="user_upload",
|
|
)
|
|
)
|
|
|
|
assert response.status == "ready"
|
|
assert response.crs == "EPSG:31370"
|
|
assert response.bounds_json == {
|
|
"minx": 193277.5,
|
|
"miny": 205708.2,
|
|
"maxx": 193777.5,
|
|
"maxy": 206208.2,
|
|
}
|
|
assert db.added[0].bounds_json == response.bounds_json
|
|
assert db.added[0].resolution_json == {"x": 0.9765625, "y": 0.9765625}
|
|
assert db.added[0].bands_json == {"dtype": ["uint8", "uint8", "uint8"]}
|