add tests

Author: vincentsarago

tests/routes/test_collections.py

@@ -19,7 +19,7 @@ def test_collections(app):
 
 
 def test_collections_landsat(app):
-    """Test /collections endpoint."""
+    """Test /collections/{collectionId} endpoint."""
     response = app.get("/collections/public.landsat_wrs")
     assert response.status_code == 200
     assert response.headers["content-type"] == "application/json"
@@ -44,3 +44,19 @@ def test_collections_landsat(app):
     assert response.status_code == 422
     body = response.json()
     assert body["detail"] == "Invalid Table format 'landsat_wrs'."
+
+
+def test_collections_queryables(app):
+    """Test /collections/{collectionId}/queryables endpoint."""
+    response = app.get("/collections/public.landsat_wrs/queryables")
+    assert response.status_code == 200
+    assert response.headers["content-type"] == "application/schema+json"
+    body = response.json()
+    assert body["title"] == "public.landsat_wrs"
+    assert body["type"] == "object"
+    assert ["title", "properties", "type", "$schema", "$id"] == list(body)
+
+    response = app.get("/collections/public.landsat_wrs/queryables?f=html")
+    assert response.status_code == 200
+    assert "text/html" in response.headers["content-type"]
+    assert "Queryables" in response.text

tifeatures/dbmodel.py

@@ -14,23 +14,28 @@ class Column(BaseModel):
     description: Optional[str]
 
     @property
-    def json_type(self):
+    def json_type(self) -> str:
         """Return JSON field type."""
-        type = self.type
+        pgtype = self.type
+
         if any(
             [
-                type.startswith("int"),
-                type.startswith("num"),
-                type.startswith("float"),
+                pgtype.startswith("int"),
+                pgtype.startswith("num"),
+                pgtype.startswith("float"),
             ]
         ):
             return "number"
-        if type.startswith("bool"):
+
+        if pgtype.startswith("bool"):
             return "boolean"
-        if type.endswith("[]"):
+
+        if pgtype.endswith("[]"):
             return "array"
-        if any([type.startswith("json"), type.startswith("geo")]):
+
+        if any([pgtype.startswith("json"), pgtype.startswith("geo")]):
             return "object"
+
         return "string"
 
 

tifeatures/factory.py

@@ -21,7 +21,7 @@
 from tifeatures.layer import CollectionLayer
 from tifeatures.layer import Table as TableLayer
 from tifeatures.resources.enums import MediaType, ResponseType
-from tifeatures.resources.response import GeoJSONResponse
+from tifeatures.resources.response import GeoJSONResponse, SchemaJSONResponse
 from tifeatures.settings import APISettings
 
 from fastapi import APIRouter, Depends, Path, Query
@@ -438,6 +438,7 @@ def collection(
             response_model=model.Queryables,
             response_model_exclude_none=True,
             response_model_by_alias=True,
+            response_class=SchemaJSONResponse,
             responses={
                 200: {
                     "content": {

tifeatures/resources/response.py

@@ -7,3 +7,9 @@ class GeoJSONResponse(responses.JSONResponse):
     """GeoJSON Response"""
 
     media_type = "application/geo+json"
+
+
+class SchemaJSONResponse(responses.JSONResponse):
+    """Schema Response"""
+
+    media_type = "application/schema+json"