Initial public release

backend/tests/test_sprint122_raster_upload_metadata_mapping.py

@@ -0,0 +1,92 @@
+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"]}