tests: adding redis tests

Author: Yuri Zmytrakov

stac_fastapi/elasticsearch/setup.py

@@ -11,6 +11,7 @@
     "elasticsearch[async]~=8.18.0",
     "uvicorn~=0.23.0",
     "starlette>=0.35.0,<0.36.0",
+    "redis==6.4.0",
 ]
 
 extra_reqs = {

stac_fastapi/opensearch/setup.py

@@ -12,6 +12,7 @@
     "opensearch-py[async]~=2.8.0",
     "uvicorn~=0.23.0",
     "starlette>=0.35.0,<0.36.0",
+    "redis==6.4.0",
 ]
 
 extra_reqs = {

stac_fastapi/sfeos_helpers/setup.py

@@ -7,6 +7,7 @@
 
 install_requires = [
     "stac-fastapi.core==6.4.0",
+    "redis==6.4.0",
 ]
 
 setup(

stac_fastapi/tests/api/test_api.py

@@ -3,10 +3,19 @@
 import uuid
 from copy import deepcopy
 from datetime import datetime, timedelta
-from unittest.mock import patch
+from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 
+import stac_fastapi.core.redis_utils as redis_utils
+from stac_fastapi.core.redis_utils import (
+    RedisSentinelSettings,
+    RedisSettings,
+    connect_redis,
+    connect_redis_sentinel,
+    get_prev_link,
+    save_self_link,
+)
 from stac_fastapi.types.errors import ConflictError
 
 from ..conftest import create_collection, create_item
@@ -1623,3 +1632,105 @@ async def test_use_datetime_false(app_client, load_test_data, txn_client, monkey
 
     assert "test-item-datetime-only" not in found_ids
     assert "test-item-start-end-only" in found_ids
+
+
+@pytest.mark.asyncio
+async def test_connect_redis():
+    test_settings = RedisSettings(
+        REDIS_HOST="test-redis-host",
+        REDIS_PORT=6380,
+        REDIS_DB=5,
+        REDIS_MAX_CONNECTIONS=20,
+        REDIS_RETRY_TIMEOUT=False,
+        REDIS_DECODE_RESPONSES=False,
+        REDIS_CLIENT_NAME="custom-client",
+        REDIS_HEALTH_CHECK_INTERVAL=50,
+    )
+
+    with patch(
+        "stac_fastapi.core.redis_utils.aioredis.ConnectionPool"
+    ) as mock_pool_class, patch(
+        "stac_fastapi.core.redis_utils.aioredis.Redis"
+    ) as mock_redis_class:
+
+        mock_pool_instance = AsyncMock()
+        mock_redis_instance = AsyncMock()
+        mock_pool_class.return_value = mock_pool_instance
+        mock_redis_class.return_value = mock_redis_instance
+
+        # from stac_fastapi.core.redis_utils import redis_pool
+
+        # redis_pool = None
+
+        result = await connect_redis(test_settings)
+
+        mock_pool_class.assert_called_once_with(
+            host="test-redis-host",
+            port=6380,
+            db=5,
+            max_connections=20,
+            decode_responses=False,
+            retry_on_timeout=False,
+            health_check_interval=50,
+        )
+
+        mock_redis_class.assert_called_once_with(
+            connection_pool=mock_pool_instance, client_name="custom-client"
+        )
+
+        assert result == mock_redis_instance
+
+
+@pytest.mark.asyncio
+async def test_connect_redis_sentinel(monkeypatch):
+    redis_utils.redis_pool = None
+
+    master_mock = AsyncMock()
+
+    sentinel_mock = MagicMock()
+    sentinel_mock.master_for.return_value = master_mock
+
+    monkeypatch.setattr(
+        "stac_fastapi.core.redis_utils.Sentinel",
+        lambda *a, **k: sentinel_mock,
+    )
+
+    settings = RedisSentinelSettings(
+        REDIS_SENTINEL_HOSTS="localhost",
+        REDIS_SENTINEL_PORTS="26379",
+        REDIS_SENTINEL_MASTER_NAME="master",
+    )
+
+    redis = await connect_redis_sentinel(settings)
+
+    assert redis is master_mock
+
+    sentinel_mock.master_for.assert_called_once_with(
+        service_name="master",
+        db=0,
+        decode_responses=True,
+        retry_on_timeout=True,
+        client_name="stac-fastapi-app",
+        max_connections=10,
+        health_check_interval=30,
+    )
+
+
+@pytest.mark.asyncio
+async def test_save_and_get_prev_link(monkeypatch):
+    fake_redis = AsyncMock()
+
+    await save_self_link(fake_redis, "token123", "http://example.com/page2")
+    fake_redis.setex.assert_awaited_once_with(
+        "nav:self:token123", 1800, "http://example.com/page2"
+    )
+
+    fake_redis.get.return_value = "http://example.com/page2"
+    result = await get_prev_link(fake_redis, "token123")
+    assert result == "http://example.com/page2"
+
+    result_none = await get_prev_link(fake_redis, None)
+    assert result_none is None
+
+    await get_prev_link(fake_redis, "token456")
+    fake_redis.get.assert_awaited_with("nav:self:token456")