fix integration test

Signed-off-by: stroller <[email protected]>

Author: jiafu1115

memory/repository/spring-ai-model-chat-memory-repository-elasticsearch/src/test/java/org/springframework/ai/chat/memory/repository/elasticsearch/ElasticSearchChatMemoryRepositoryIT.java

@@ -19,6 +19,7 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.UUID;
+import java.util.concurrent.TimeUnit;
 
 import co.elastic.clients.elasticsearch.ElasticsearchClient;
 import co.elastic.clients.json.jackson.JacksonJsonpMapper;
@@ -91,6 +92,7 @@ void add_shouldInsertSingleMessage(String content, MessageType messageType) {
 			};
 
 			chatMemory.saveAll(conversationId, List.of(message));
+			sleepForSearchable();
 			assertThat(chatMemory.findConversationIds()).isNotEmpty();
 			assertThat(chatMemory.findConversationIds()).contains(conversationId);
 
@@ -101,6 +103,10 @@ void add_shouldInsertSingleMessage(String content, MessageType messageType) {
 		});
 	}
 
+	private static void sleepForSearchable() throws InterruptedException {
+		TimeUnit.SECONDS.sleep(2);
+	}
+
 	@Test
 	void shouldSaveAndRetrieveMultipleMessages() {
 		this.contextRunner.run(context -> {
@@ -114,6 +120,8 @@ void shouldSaveAndRetrieveMultipleMessages() {
 
 			chatMemory.saveAll(conversationId, messages);
 
+			sleepForSearchable();
+
 			List<Message> retrievedMessages = chatMemory.findByConversationId(conversationId);
 			assertThat(retrievedMessages).hasSize(3);
 
@@ -140,6 +148,8 @@ void shouldReplaceExistingMessages() {
 					new AssistantMessage("Initial assistant message"));
 			chatMemory.saveAll(conversationId, initialMessages);
 
+			sleepForSearchable();
+
 			// Verify initial save
 			List<Message> retrievedMessages = chatMemory.findByConversationId(conversationId);
 			assertThat(retrievedMessages).hasSize(2);
@@ -149,6 +159,8 @@ void shouldReplaceExistingMessages() {
 					new UserMessage("New user message"));
 			chatMemory.saveAll(conversationId, newMessages);
 
+			sleepForSearchable();
+
 			// Verify replacement
 			retrievedMessages = chatMemory.findByConversationId(conversationId);
 			assertThat(retrievedMessages).hasSize(2);
@@ -168,12 +180,16 @@ void shouldDeleteConversation() {
 					new AssistantMessage("Assistant message"));
 			chatMemory.saveAll(conversationId, messages);
 
+			sleepForSearchable();
+
 			// Verify messages exist
 			assertThat(chatMemory.findByConversationId(conversationId)).hasSize(2);
 
 			// Delete conversation
 			chatMemory.deleteByConversationId(conversationId);
 
+			sleepForSearchable();
+
 			// Verify messages are deleted
 			assertThat(chatMemory.findByConversationId(conversationId)).isEmpty();
 		});
@@ -190,6 +206,8 @@ void shouldFindAllConversationIds() {
 			chatMemory.saveAll(conversationId1, List.of(new UserMessage("Message 1")));
 			chatMemory.saveAll(conversationId2, List.of(new UserMessage("Message 2")));
 
+			sleepForSearchable();
+
 			// Verify both conversation IDs are found
 			List<String> conversationIds = chatMemory.findConversationIds();
 			assertThat(conversationIds).contains(conversationId1, conversationId2);