chore: add integ test for creating/updating memories with indexed keys

Author: nborges-aws

tests_integ/memory/test_memory_client.py

@@ -18,6 +18,7 @@
 import pytest
 
 from bedrock_agentcore.memory import MemoryClient
+from bedrock_agentcore.memory.models import IndexedKey, MetadataValueType
 
 from .helpers import poll_until
 
@@ -103,6 +104,34 @@ def test_stream_delivery_create_and_update(self):
         )
         assert response["memory"]["streamDeliveryResources"] == updated_config
 
+    def test_indexed_keys_create_and_update(self):
+        """Create memory with indexed_keys, then add another via update_memory passthrough."""
+        initial_keys = [
+            IndexedKey.build("priority", MetadataValueType.NUMBER),
+            IndexedKey.build("agent_type", MetadataValueType.STRING),
+        ]
+
+        memory = self.client.create_memory_and_wait(
+            name=f"{self.test_prefix}_indexed",
+            strategies=[],
+            indexed_keys=initial_keys,
+        )
+
+        memory_id = memory["id"]
+        self.__class__.memory_ids.append(memory_id)
+
+        assert memory.get("indexedKeys") == initial_keys
+
+        added_key = IndexedKey.build("region", MetadataValueType.STRING)
+        response = self.client.update_memory(
+            memory_id=memory_id,
+            client_token=str(uuid.uuid4()),
+            add_indexed_keys=[added_key],
+        )
+
+        updated_keys = response["memory"]["indexedKeys"]
+        assert {k["key"] for k in updated_keys} == {"priority", "agent_type", "region"}
+
     # --- Data Plane Tests (order 1-10) ---
 
     @pytest.mark.order(1)