fix tests

Author: Rafael Pierre

src/agents/memory/providers/redis.py

@@ -144,6 +144,36 @@ async def add_item(self, item: TResponseInputItem) -> None:
 
         await pipeline.execute()
 
+    async def add_items(self, items: list[TResponseInputItem]) -> None:
+        """Add multiple items to the session's conversation history.
+
+        Args:
+            items: List of response input items to add
+        """
+        if not items:
+            return
+
+        client = await self._get_redis_client()
+
+        # Ensure session exists first
+        await self._ensure_session_exists(client)
+
+        # Serialize all items and add them to the messages list in one rpush call
+        serialized_items = [json.dumps(item) for item in items]
+        pipeline = client.pipeline()
+        pipeline.rpush(self.messages_key, *serialized_items)
+
+        # Update session timestamp
+        current_time = time.time()
+        pipeline.hset(self.session_key, "updated_at", str(current_time))
+
+        # Set expiration on both keys if TTL is configured
+        if self.ttl:
+            pipeline.expire(self.session_key, self.ttl)
+            pipeline.expire(self.messages_key, self.ttl)
+
+        await pipeline.execute()
+
     async def pop_item(self) -> TResponseInputItem | None:
         """Remove and return the most recent item from the session.
 

tests/test_redis_session.py

@@ -188,14 +188,10 @@ async def test_add_items(
         mock_time.return_value = 1234567890.0
         mock_from_url.return_value = mock_redis
 
-        # Create a proper async context manager mock
-        mock_pipeline = AsyncMock()
-        mock_context_manager = AsyncMock()
-        mock_context_manager.__aenter__.return_value = mock_pipeline
-        mock_context_manager.__aexit__.return_value = None
-
-        # Make pipeline() return the context manager directly (not a coroutine)
-        mock_redis.pipeline = MagicMock(return_value=mock_context_manager)
+        # Create a mock pipeline that's returned directly (not as context manager)
+        mock_pipeline = MagicMock()
+        mock_pipeline.execute = AsyncMock()
+        mock_redis.pipeline = MagicMock(return_value=mock_pipeline)
 
         # Mock _ensure_session_exists method
         redis_session._ensure_session_exists = AsyncMock()