sfeos-tools 0.1.0__tar.gz → 0.1.1__tar.gz

{sfeos_tools-0.1.0/sfeos_tools.egg-info → sfeos_tools-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfeos-tools
-Version: 0.1.0
+Version: 0.1.1
 Summary: CLI tools for managing stac-fastapi-elasticsearch-opensearch deployments
 Author: CloudFerro S.A.
 Author-email: Jonathan Healy <jon@healy-hypersaptial.dev>
@@ -35,11 +35,12 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: GIS
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
@@ -48,13 +49,14 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: click>=8.0.0
 Provides-Extra: elasticsearch
-Requires-Dist: stac-fastapi-elasticsearch; extra == "elasticsearch"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "elasticsearch"
 Provides-Extra: opensearch
-Requires-Dist: stac-fastapi-opensearch; extra == "opensearch"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "opensearch"
 Provides-Extra: dev
-Requires-Dist: stac-fastapi-elasticsearch; extra == "dev"
-Requires-Dist: stac-fastapi-opensearch; extra == "dev"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "dev"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "dev"
 Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: pytest-cov>=2.0; extra == "dev"
 Requires-Dist: black>=21.0; extra == "dev"
 Requires-Dist: isort>=5.0; extra == "dev"
@@ -66,6 +68,22 @@ Dynamic: license-file
 
 CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch) deployments.
 
+<!-- markdownlint-disable MD033 MD041 -->
+
+
+<p align="left">
+  <img src="https://raw.githubusercontent.com/stac-utils/stac-fastapi-elasticsearch-opensearch/refs/heads/main/assets/sfeos.png" width=1000>
+</p>
+
+<!-- **Jump to:** [Project Introduction](#project-introduction---what-is-sfeos) | [Quick Start](#quick-start) | [Table of Contents](#table-of-contents) -->
+
+  [![Downloads](https://static.pepy.tech/badge/sfeos-tools?color=blue)](https://pepy.tech/project/sfeos-tools)
+  [![GitHub contributors](https://img.shields.io/github/contributors/healy-hyperspatial/sfeos-tools?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/graphs/contributors)
+  [![GitHub stars](https://img.shields.io/github/stars/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/stargazers)
+  [![GitHub forks](https://img.shields.io/github/forks/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/network/members)
+   [![PyPI version](https://img.shields.io/pypi/v/sfeos-tools.svg?color=blue)](https://pypi.org/project/sfeos-tools/)
+  [![STAC](https://img.shields.io/badge/STAC-1.1.0-blue.svg)](https://github.com/radiantearth/stac-spec/tree/v1.1.0)
+
 ## Table of Contents
 
 - [Installation](#installation)

{sfeos_tools-0.1.0 → sfeos_tools-0.1.1}/README.md

@@ -2,6 +2,22 @@
 
 CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch) deployments.
 
+<!-- markdownlint-disable MD033 MD041 -->
+
+
+<p align="left">
+  <img src="https://raw.githubusercontent.com/stac-utils/stac-fastapi-elasticsearch-opensearch/refs/heads/main/assets/sfeos.png" width=1000>
+</p>
+
+<!-- **Jump to:** [Project Introduction](#project-introduction---what-is-sfeos) | [Quick Start](#quick-start) | [Table of Contents](#table-of-contents) -->
+
+  [![Downloads](https://static.pepy.tech/badge/sfeos-tools?color=blue)](https://pepy.tech/project/sfeos-tools)
+  [![GitHub contributors](https://img.shields.io/github/contributors/healy-hyperspatial/sfeos-tools?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/graphs/contributors)
+  [![GitHub stars](https://img.shields.io/github/stars/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/stargazers)
+  [![GitHub forks](https://img.shields.io/github/forks/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/network/members)
+   [![PyPI version](https://img.shields.io/pypi/v/sfeos-tools.svg?color=blue)](https://pypi.org/project/sfeos-tools/)
+  [![STAC](https://img.shields.io/badge/STAC-1.1.0-blue.svg)](https://github.com/radiantearth/stac-spec/tree/v1.1.0)
+
 ## Table of Contents
 
 - [Installation](#installation)

{sfeos_tools-0.1.0 → sfeos_tools-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sfeos-tools"
-version = "0.1.0"
+version = "0.1.1"
 description = "CLI tools for managing stac-fastapi-elasticsearch-opensearch deployments"
 readme = "README.md"
 authors = [
@@ -21,11 +21,12 @@ classifiers = [
     "Intended Audience :: Developers",
     "Intended Audience :: Science/Research",
     "License :: OSI Approved :: MIT License",
-    "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
     "Topic :: Scientific/Engineering",
     "Topic :: Scientific/Engineering :: GIS",
     "Topic :: Software Development :: Libraries :: Python Modules"
@@ -42,16 +43,17 @@ Source = "https://github.com/Healy-Hyperspatial/sfeos-tools"
 
 [project.optional-dependencies]
 elasticsearch = [
-    "stac-fastapi-elasticsearch",
+    "stac-fastapi-elasticsearch>=6.6.0",
 ]
 opensearch = [
-    "stac-fastapi-opensearch",
+    "stac-fastapi-opensearch>=6.6.0",
 ]
 dev = [
-    "stac-fastapi-elasticsearch",
-    "stac-fastapi-opensearch",
+    "stac-fastapi-elasticsearch>=6.6.0",
+    "stac-fastapi-opensearch>=6.6.0",
     # Development tools
     "pytest>=6.0",
+    "pytest-asyncio>=0.21.0",
     "pytest-cov>=2.0",
     "black>=21.0",
     "isort>=5.0",
@@ -79,7 +81,11 @@ line_length = 88
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = "test_*.py"
-addopts = "-v --cov=sfeos_tool --cov-report=term-missing"
+addopts = "-v --cov=sfeos_tools --cov-report=term-missing"
+asyncio_mode = "auto"
+filterwarnings = [
+    "ignore::DeprecationWarning:pydantic._internal._config",
+]
 
 [tool.coverage.report]
 exclude_lines = [

sfeos_tools-0.1.1/sfeos_tools/bbox_shape.py

@@ -0,0 +1,114 @@
+"""Bbox shape migration utilities for SFEOS collections."""
+
+import logging
+
+from stac_fastapi.sfeos_helpers.database import add_bbox_shape_to_collection
+from stac_fastapi.sfeos_helpers.mappings import COLLECTIONS_INDEX
+
+logger = logging.getLogger(__name__)
+
+
+async def process_collection_bbox_shape(client, collection_doc, backend):
+    """Process a single collection document to add bbox_shape field.
+
+    Args:
+        client: Elasticsearch/OpenSearch client
+        collection_doc: Collection document from database
+        backend: Backend type ('elasticsearch' or 'opensearch')
+
+    Returns:
+        bool: True if collection was updated, False if no update was needed
+    """
+    collection = collection_doc["_source"]
+    collection_id = collection.get("id", collection_doc["_id"])
+
+    # Use the shared function to add bbox_shape
+    was_added = add_bbox_shape_to_collection(collection)
+
+    if not was_added:
+        return False
+
+    # Update the collection in the database
+    if backend == "elasticsearch":
+        await client.index(
+            index=COLLECTIONS_INDEX,
+            id=collection_id,
+            document=collection,
+            refresh=True,
+        )
+    else:  # opensearch
+        await client.index(
+            index=COLLECTIONS_INDEX,
+            id=collection_id,
+            body=collection,
+            refresh=True,
+        )
+
+    logger.info(f"Collection '{collection_id}': Added bbox_shape field")
+    return True
+
+
+async def run_add_bbox_shape(backend):
+    """Add bbox_shape field to all existing collections.
+
+    Args:
+        backend: Backend type ('elasticsearch' or 'opensearch')
+    """
+    import os
+
+    logger.info(
+        f"Starting migration: Adding bbox_shape to existing collections ({backend})"
+    )
+
+    # Log connection info (showing what will be used by the client)
+    es_host = os.getenv("ES_HOST", "localhost")
+    es_port = os.getenv(
+        "ES_PORT", "9200"
+    )  # Both backends default to 9200 in their config
+    es_use_ssl = os.getenv("ES_USE_SSL", "true")
+    logger.info(f"Connecting to {backend} at {es_host}:{es_port} (SSL: {es_use_ssl})")
+
+    # Create client based on backend
+    if backend == "elasticsearch":
+        from stac_fastapi.elasticsearch.config import AsyncElasticsearchSettings
+
+        settings = AsyncElasticsearchSettings()
+    else:  # opensearch
+        from stac_fastapi.opensearch.config import AsyncOpensearchSettings
+
+        settings = AsyncOpensearchSettings()
+
+    client = settings.create_client
+
+    try:
+        # Get all collections
+        response = await client.search(
+            index=COLLECTIONS_INDEX,
+            body={
+                "query": {"match_all": {}},
+                "size": 10000,
+            },  # Adjust size if you have more collections
+        )
+
+        total_collections = response["hits"]["total"]["value"]
+        logger.info(f"Found {total_collections} collections to process")
+
+        updated_count = 0
+        skipped_count = 0
+
+        for hit in response["hits"]["hits"]:
+            was_updated = await process_collection_bbox_shape(client, hit, backend)
+            if was_updated:
+                updated_count += 1
+            else:
+                skipped_count += 1
+
+        logger.info(
+            f"Migration complete: {updated_count} collections updated, {skipped_count} skipped"
+        )
+
+    except Exception as e:
+        logger.error(f"Migration failed with error: {e}")
+        raise
+    finally:
+        await client.close()

{sfeos_tools-0.1.0 → sfeos_tools-0.1.1}/sfeos_tools/cli.py

@@ -13,121 +13,14 @@ import logging
 import sys
 
 import click
-from stac_fastapi.sfeos_helpers.database import add_bbox_shape_to_collection
-from stac_fastapi.sfeos_helpers.mappings import COLLECTIONS_INDEX
 
+from .bbox_shape import run_add_bbox_shape
 from .reindex import run as unified_reindex_run
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
-async def process_collection_bbox_shape(client, collection_doc, backend):
-    """Process a single collection document to add bbox_shape field.
-
-    Args:
-        client: Elasticsearch/OpenSearch client
-        collection_doc: Collection document from database
-        backend: Backend type ('elasticsearch' or 'opensearch')
-
-    Returns:
-        bool: True if collection was updated, False if no update was needed
-    """
-    collection = collection_doc["_source"]
-    collection_id = collection.get("id", collection_doc["_id"])
-
-    # Use the shared function to add bbox_shape
-    was_added = add_bbox_shape_to_collection(collection)
-
-    if not was_added:
-        return False
-
-    # Update the collection in the database
-    if backend == "elasticsearch":
-        await client.index(
-            index=COLLECTIONS_INDEX,
-            id=collection_id,
-            document=collection,
-            refresh=True,
-        )
-    else:  # opensearch
-        await client.index(
-            index=COLLECTIONS_INDEX,
-            id=collection_id,
-            body=collection,
-            refresh=True,
-        )
-
-    logger.info(f"Collection '{collection_id}': Added bbox_shape field")
-    return True
-
-
-async def run_add_bbox_shape(backend):
-    """Add bbox_shape field to all existing collections.
-
-    Args:
-        backend: Backend type ('elasticsearch' or 'opensearch')
-    """
-    import os
-
-    logger.info(
-        f"Starting migration: Adding bbox_shape to existing collections ({backend})"
-    )
-
-    # Log connection info (showing what will be used by the client)
-    es_host = os.getenv("ES_HOST", "localhost")
-    es_port = os.getenv(
-        "ES_PORT", "9200"
-    )  # Both backends default to 9200 in their config
-    es_use_ssl = os.getenv("ES_USE_SSL", "true")
-    logger.info(f"Connecting to {backend} at {es_host}:{es_port} (SSL: {es_use_ssl})")
-
-    # Create client based on backend
-    if backend == "elasticsearch":
-        from stac_fastapi.elasticsearch.config import AsyncElasticsearchSettings
-
-        settings = AsyncElasticsearchSettings()
-    else:  # opensearch
-        from stac_fastapi.opensearch.config import AsyncOpensearchSettings
-
-        settings = AsyncOpensearchSettings()
-
-    client = settings.create_client
-
-    try:
-        # Get all collections
-        response = await client.search(
-            index=COLLECTIONS_INDEX,
-            body={
-                "query": {"match_all": {}},
-                "size": 10000,
-            },  # Adjust size if you have more collections
-        )
-
-        total_collections = response["hits"]["total"]["value"]
-        logger.info(f"Found {total_collections} collections to process")
-
-        updated_count = 0
-        skipped_count = 0
-
-        for hit in response["hits"]["hits"]:
-            was_updated = await process_collection_bbox_shape(client, hit, backend)
-            if was_updated:
-                updated_count += 1
-            else:
-                skipped_count += 1
-
-        logger.info(
-            f"Migration complete: {updated_count} collections updated, {skipped_count} skipped"
-        )
-
-    except Exception as e:
-        logger.error(f"Migration failed with error: {e}")
-        raise
-    finally:
-        await client.close()
-
-
 @click.group()
 @click.version_option(version="0.1.0", prog_name="sfeos-tools")
 def cli():

{sfeos_tools-0.1.0 → sfeos_tools-0.1.1/sfeos_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfeos-tools
-Version: 0.1.0
+Version: 0.1.1
 Summary: CLI tools for managing stac-fastapi-elasticsearch-opensearch deployments
 Author: CloudFerro S.A.
 Author-email: Jonathan Healy <jon@healy-hypersaptial.dev>
@@ -35,11 +35,12 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: GIS
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
@@ -48,13 +49,14 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: click>=8.0.0
 Provides-Extra: elasticsearch
-Requires-Dist: stac-fastapi-elasticsearch; extra == "elasticsearch"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "elasticsearch"
 Provides-Extra: opensearch
-Requires-Dist: stac-fastapi-opensearch; extra == "opensearch"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "opensearch"
 Provides-Extra: dev
-Requires-Dist: stac-fastapi-elasticsearch; extra == "dev"
-Requires-Dist: stac-fastapi-opensearch; extra == "dev"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "dev"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "dev"
 Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: pytest-cov>=2.0; extra == "dev"
 Requires-Dist: black>=21.0; extra == "dev"
 Requires-Dist: isort>=5.0; extra == "dev"
@@ -66,6 +68,22 @@ Dynamic: license-file
 
 CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch) deployments.
 
+<!-- markdownlint-disable MD033 MD041 -->
+
+
+<p align="left">
+  <img src="https://raw.githubusercontent.com/stac-utils/stac-fastapi-elasticsearch-opensearch/refs/heads/main/assets/sfeos.png" width=1000>
+</p>
+
+<!-- **Jump to:** [Project Introduction](#project-introduction---what-is-sfeos) | [Quick Start](#quick-start) | [Table of Contents](#table-of-contents) -->
+
+  [![Downloads](https://static.pepy.tech/badge/sfeos-tools?color=blue)](https://pepy.tech/project/sfeos-tools)
+  [![GitHub contributors](https://img.shields.io/github/contributors/healy-hyperspatial/sfeos-tools?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/graphs/contributors)
+  [![GitHub stars](https://img.shields.io/github/stars/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/stargazers)
+  [![GitHub forks](https://img.shields.io/github/forks/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/network/members)
+   [![PyPI version](https://img.shields.io/pypi/v/sfeos-tools.svg?color=blue)](https://pypi.org/project/sfeos-tools/)
+  [![STAC](https://img.shields.io/badge/STAC-1.1.0-blue.svg)](https://github.com/radiantearth/stac-spec/tree/v1.1.0)
+
 ## Table of Contents
 
 - [Installation](#installation)

{sfeos_tools-0.1.0 → sfeos_tools-0.1.1}/sfeos_tools.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 sfeos_tools/__init__.py
+sfeos_tools/bbox_shape.py
 sfeos_tools/cli.py
 sfeos_tools/reindex.py
 sfeos_tools.egg-info/PKG-INFO
@@ -9,4 +10,5 @@ sfeos_tools.egg-info/SOURCES.txt
 sfeos_tools.egg-info/dependency_links.txt
 sfeos_tools.egg-info/entry_points.txt
 sfeos_tools.egg-info/requires.txt
-sfeos_tools.egg-info/top_level.txt
+sfeos_tools.egg-info/top_level.txt
+tests/test_bbox_shape.py

sfeos_tools-0.1.1/sfeos_tools.egg-info/requires.txt

@@ -0,0 +1,18 @@
+click>=8.0.0
+
+[dev]
+stac-fastapi-elasticsearch>=6.6.0
+stac-fastapi-opensearch>=6.6.0
+pytest>=6.0
+pytest-asyncio>=0.21.0
+pytest-cov>=2.0
+black>=21.0
+isort>=5.0
+mypy>=0.900
+flake8>=4.0.0
+
+[elasticsearch]
+stac-fastapi-elasticsearch>=6.6.0
+
+[opensearch]
+stac-fastapi-opensearch>=6.6.0

sfeos_tools-0.1.1/tests/test_bbox_shape.py

@@ -0,0 +1,275 @@
+"""Tests for bbox_shape module."""
+
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from sfeos_tools.bbox_shape import process_collection_bbox_shape, run_add_bbox_shape
+
+
+class TestProcessCollectionBboxShape:
+    """Tests for process_collection_bbox_shape function."""
+
+    @pytest.mark.asyncio
+    async def test_process_collection_bbox_shape_elasticsearch_updated(self):
+        """Test processing a collection with Elasticsearch backend when bbox_shape is added."""
+        mock_client = AsyncMock()
+        collection_doc = {
+            "_id": "test-collection",
+            "_source": {
+                "id": "test-collection",
+                "bbox": [[-180, -90, 180, 90]],
+            },
+        }
+
+        with patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            result = await process_collection_bbox_shape(
+                mock_client, collection_doc, "elasticsearch"
+            )
+
+        assert result is True
+        mock_client.index.assert_called_once()
+        call_kwargs = mock_client.index.call_args[1]
+        assert call_kwargs["index"] == "collections"
+        assert call_kwargs["id"] == "test-collection"
+        assert call_kwargs["refresh"] is True
+        assert "document" in call_kwargs
+
+    @pytest.mark.asyncio
+    async def test_process_collection_bbox_shape_opensearch_updated(self):
+        """Test processing a collection with OpenSearch backend when bbox_shape is added."""
+        mock_client = AsyncMock()
+        collection_doc = {
+            "_id": "test-collection",
+            "_source": {
+                "id": "test-collection",
+                "bbox": [[-180, -90, 180, 90]],
+            },
+        }
+
+        with patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            result = await process_collection_bbox_shape(
+                mock_client, collection_doc, "opensearch"
+            )
+
+        assert result is True
+        mock_client.index.assert_called_once()
+        call_kwargs = mock_client.index.call_args[1]
+        assert call_kwargs["index"] == "collections"
+        assert call_kwargs["id"] == "test-collection"
+        assert call_kwargs["refresh"] is True
+        assert "body" in call_kwargs
+
+    @pytest.mark.asyncio
+    async def test_process_collection_bbox_shape_not_updated(self):
+        """Test processing a collection when bbox_shape is not added."""
+        mock_client = AsyncMock()
+        collection_doc = {
+            "_id": "test-collection",
+            "_source": {
+                "id": "test-collection",
+                "bbox": [[-180, -90, 180, 90]],
+            },
+        }
+
+        with patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=False
+        ):
+            result = await process_collection_bbox_shape(
+                mock_client, collection_doc, "elasticsearch"
+            )
+
+        assert result is False
+        mock_client.index.assert_not_called()
+
+    @pytest.mark.asyncio
+    async def test_process_collection_uses_id_from_source(self):
+        """Test that collection ID is extracted from _source when available."""
+        mock_client = AsyncMock()
+        collection_doc = {
+            "_id": "doc-id",
+            "_source": {
+                "id": "collection-id",
+                "bbox": [[-180, -90, 180, 90]],
+            },
+        }
+
+        with patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            await process_collection_bbox_shape(
+                mock_client, collection_doc, "elasticsearch"
+            )
+
+        call_kwargs = mock_client.index.call_args[1]
+        assert call_kwargs["id"] == "collection-id"
+
+    @pytest.mark.asyncio
+    async def test_process_collection_falls_back_to_doc_id(self):
+        """Test that collection ID falls back to _id when not in _source."""
+        mock_client = AsyncMock()
+        collection_doc = {
+            "_id": "doc-id",
+            "_source": {
+                "bbox": [[-180, -90, 180, 90]],
+            },
+        }
+
+        with patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            await process_collection_bbox_shape(
+                mock_client, collection_doc, "elasticsearch"
+            )
+
+        call_kwargs = mock_client.index.call_args[1]
+        assert call_kwargs["id"] == "doc-id"
+
+
+class TestRunAddBboxShape:
+    """Tests for run_add_bbox_shape function."""
+
+    @pytest.mark.asyncio
+    async def test_run_add_bbox_shape_elasticsearch(self):
+        """Test run_add_bbox_shape with Elasticsearch backend."""
+        mock_client = AsyncMock()
+        mock_settings = MagicMock()
+        mock_settings.create_client = mock_client
+
+        mock_client.search.return_value = {
+            "hits": {
+                "total": {"value": 2},
+                "hits": [
+                    {
+                        "_id": "collection-1",
+                        "_source": {
+                            "id": "collection-1",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                    {
+                        "_id": "collection-2",
+                        "_source": {
+                            "id": "collection-2",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                ],
+            }
+        }
+
+        with patch(
+            "stac_fastapi.elasticsearch.config.AsyncElasticsearchSettings",
+            return_value=mock_settings,
+        ), patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            await run_add_bbox_shape("elasticsearch")
+
+        mock_client.search.assert_called_once()
+        assert mock_client.index.call_count == 2
+        mock_client.close.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_run_add_bbox_shape_opensearch(self):
+        """Test run_add_bbox_shape with OpenSearch backend."""
+        mock_client = AsyncMock()
+        mock_settings = MagicMock()
+        mock_settings.create_client = mock_client
+
+        mock_client.search.return_value = {
+            "hits": {
+                "total": {"value": 1},
+                "hits": [
+                    {
+                        "_id": "collection-1",
+                        "_source": {
+                            "id": "collection-1",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                ],
+            }
+        }
+
+        with patch(
+            "stac_fastapi.opensearch.config.AsyncOpensearchSettings",
+            return_value=mock_settings,
+        ), patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection", return_value=True
+        ):
+            await run_add_bbox_shape("opensearch")
+
+        mock_client.search.assert_called_once()
+        mock_client.index.assert_called_once()
+        mock_client.close.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_run_add_bbox_shape_handles_mixed_results(self):
+        """Test run_add_bbox_shape with some collections updated and some skipped."""
+        mock_client = AsyncMock()
+        mock_settings = MagicMock()
+        mock_settings.create_client = mock_client
+
+        mock_client.search.return_value = {
+            "hits": {
+                "total": {"value": 3},
+                "hits": [
+                    {
+                        "_id": "collection-1",
+                        "_source": {
+                            "id": "collection-1",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                    {
+                        "_id": "collection-2",
+                        "_source": {
+                            "id": "collection-2",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                    {
+                        "_id": "collection-3",
+                        "_source": {
+                            "id": "collection-3",
+                            "bbox": [[-180, -90, 180, 90]],
+                        },
+                    },
+                ],
+            }
+        }
+
+        side_effects = [True, False, True]
+
+        with patch(
+            "stac_fastapi.elasticsearch.config.AsyncElasticsearchSettings",
+            return_value=mock_settings,
+        ), patch(
+            "sfeos_tools.bbox_shape.add_bbox_shape_to_collection",
+            side_effect=side_effects,
+        ):
+            await run_add_bbox_shape("elasticsearch")
+
+        assert mock_client.index.call_count == 2
+
+    @pytest.mark.asyncio
+    async def test_run_add_bbox_shape_closes_client_on_error(self):
+        """Test that client is closed even when an error occurs."""
+        mock_client = AsyncMock()
+        mock_settings = MagicMock()
+        mock_settings.create_client = mock_client
+
+        mock_client.search.side_effect = Exception("Connection error")
+
+        with patch(
+            "stac_fastapi.elasticsearch.config.AsyncElasticsearchSettings",
+            return_value=mock_settings,
+        ), pytest.raises(Exception, match="Connection error"):
+            await run_add_bbox_shape("elasticsearch")
+
+        mock_client.close.assert_called_once()

sfeos_tools-0.1.0/sfeos_tools.egg-info/requires.txt

@@ -1,17 +0,0 @@
-click>=8.0.0
-
-[dev]
-stac-fastapi-elasticsearch
-stac-fastapi-opensearch
-pytest>=6.0
-pytest-cov>=2.0
-black>=21.0
-isort>=5.0
-mypy>=0.900
-flake8>=4.0.0
-
-[elasticsearch]
-stac-fastapi-elasticsearch
-
-[opensearch]
-stac-fastapi-opensearch