fixing tests

Author: Yuri Zmytrakov

stac_fastapi/core/stac_fastapi/core/redis_utils.py

@@ -2,6 +2,7 @@
 
 import json
 import logging
+from contextlib import asynccontextmanager
 from typing import List, Optional, Tuple
 
 from pydantic import field_validator
@@ -237,31 +238,44 @@ async def get_prev_link(redis: aioredis.Redis, token: Optional[str]) -> Optional
     return await redis.get(f"nav:self:{token}")
 
 
-async def redis_pagination_links(
-    current_url: str, token: str, next_token: str, links: list
-) -> None:
-    """Handle Redis pagination."""
+@asynccontextmanager
+async def redis_connection():
+    """Async context manager for Redis connections."""
     redis = await connect_redis()
     if not redis:
         logger.warning("Redis connection failed.")
+        yield None
         return
-
     try:
-        if next_token:
-            await save_self_link(redis, next_token, current_url)
-
-        prev_link = await get_prev_link(redis, token)
-        if prev_link:
-            links.insert(
-                0,
-                {
-                    "rel": "prev",
-                    "type": "application/json",
-                    "method": "GET",
-                    "href": prev_link,
-                },
-            )
-    except Exception as e:
-        logger.warning(f"Redis pagination operation failed: {e}")
+        yield redis
     finally:
         await redis.close()
+
+
+async def redis_pagination_links(
+    current_url: str, token: str, next_token: str, links: list
+) -> None:
+    """Handle Redis pagination."""
+    async with redis_connection() as redis:
+        # redis = await connect_redis()
+        if not redis:
+            logger.warning("Redis connection failed.")
+            return
+
+        try:
+            if next_token:
+                await save_self_link(redis, next_token, current_url)
+
+            prev_link = await get_prev_link(redis, token)
+            if prev_link:
+                links.insert(
+                    0,
+                    {
+                        "rel": "prev",
+                        "type": "application/json",
+                        "method": "GET",
+                        "href": prev_link,
+                    },
+                )
+        except Exception as e:
+            logger.warning(f"Redis pagination operation failed: {e}")