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,87 @@ 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(monkeypatch):
+    # Reset the global before the test
+    redis_utils.redis_pool = None
+
+    redis_mock = AsyncMock()
+    redis_pool_mock = AsyncMock()
+
+    # Patch aioredis classes
+    monkeypatch.setattr(
+        "stac_fastapi.core.redis_utils.aioredis.Redis",
+        lambda **kwargs: redis_mock,
+    )
+    monkeypatch.setattr(
+        "stac_fastapi.core.redis_utils.aioredis.ConnectionPool",
+        lambda **kwargs: redis_pool_mock,
+    )
+
+    settings = RedisSettings(REDIS_HOST="localhost", REDIS_PORT=6379)
+    redis = await connect_redis(settings)
+
+    assert redis is redis_mock
+
+
+@pytest.mark.asyncio
+async def test_connect_redis_sentinel(monkeypatch):
+    # Reset the global before the test
+    redis_utils.redis_pool = None
+
+    master_mock = AsyncMock()
+
+    # Sentinel itself is synchronous
+    sentinel_mock = MagicMock()
+    sentinel_mock.master_for.return_value = master_mock
+
+    # Patch Sentinel constructor
+    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)
+
+    # Should return our injected master_mock
+    assert redis is master_mock
+
+    # Verify correct args passed
+    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")