feat: implement refresh test

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

Author: juntae6942

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/test/java/org/springframework/ai/chat/memory/repository/jdbc/AbstractJdbcChatMemoryRepositoryIT.java

@@ -186,6 +186,31 @@ void testMessageOrder() {
 				"4-Fourth message");
 	}
 
+	@Test
+	void refreshConversation() {
+		var conversationId = UUID.randomUUID().toString();
+		List<Message> initialMessages = List.of(new UserMessage("Hello"), new AssistantMessage("Hi there"),
+				new UserMessage("How are you?"));
+		this.chatMemoryRepository.saveAll(conversationId, initialMessages);
+
+		assertThat(this.chatMemoryRepository.findByConversationId(conversationId)).hasSize(3);
+
+		// Define changes
+		List<Message> toDelete = List.of(new UserMessage("How are you?"));
+		List<Message> toAdd = List.of(new AssistantMessage("I am fine, thank you."));
+
+		// Apply changes
+		this.chatMemoryRepository.refresh(conversationId, toDelete, toAdd);
+
+		// Verify final state
+		List<Message> finalMessages = this.chatMemoryRepository.findByConversationId(conversationId);
+		assertThat(finalMessages).hasSize(3);
+		assertThat(finalMessages).contains(new UserMessage("Hello"));
+		assertThat(finalMessages).contains(new AssistantMessage("Hi there"));
+		assertThat(finalMessages).contains(new AssistantMessage("I am fine, thank you."));
+		assertThat(finalMessages).doesNotContain(new UserMessage("How are you?"));
+	}
+
 	/**
 	 * Base configuration for all integration tests.
 	 */

memory/repository/spring-ai-model-chat-memory-repository-neo4j/src/test/java/org/springframework/ai/chat/memory/repository/neo4j/Neo4jChatMemoryRepositoryIT.java

@@ -399,6 +399,45 @@ void saveAndFindMessagesWithEmptyContentOrMetadata() {
 																					// messageType
 	}
 
+	@Test
+	void refreshConversation() {
+		var conversationId = UUID.randomUUID().toString();
+
+		// 1. Save initial messages
+		var initialMessages = List.of(new UserMessage("Hello"), new AssistantMessage("Hi there"),
+				new UserMessage("How are you?"));
+		this.chatMemoryRepository.saveAll(conversationId, initialMessages);
+
+		// Retrieve to get metadata (especially the generated message IDs)
+		List<Message> savedMessages = this.chatMemoryRepository.findByConversationId(conversationId);
+		assertThat(savedMessages).hasSize(3);
+
+		// 2. Define changes
+		var messageToDelete = savedMessages.stream().filter(m -> m.getText().equals("How are you?")).findFirst().get();
+		var toDelete = List.of(messageToDelete);
+		var toAdd = List.of(new AssistantMessage("I am fine, thank you."));
+
+		// 3. Apply changes
+		this.chatMemoryRepository.refresh(conversationId, toDelete, toAdd);
+
+		// 4. Verify final state
+		List<Message> finalMessages = this.chatMemoryRepository.findByConversationId(conversationId);
+		assertThat(finalMessages).hasSize(3);
+
+		List<String> finalContents = finalMessages.stream().map(Message::getText).toList();
+		assertThat(finalContents).contains("Hello", "Hi there", "I am fine, thank you.");
+		assertThat(finalContents).doesNotContain("How are you?");
+
+		// Verify directly in the database
+		try (Session session = this.driver.session()) {
+			var result = session.run(
+					"MATCH (s:%s {id:$conversationId})-[:HAS_MESSAGE]->(m:%s) RETURN count(m) as count"
+						.formatted(this.config.getSessionLabel(), this.config.getMessageLabel()),
+					Map.of("conversationId", conversationId));
+			assertThat(result.single().get("count").asLong()).isEqualTo(3);
+		}
+	}
+
 	private Message createMessageByType(String content, MessageType messageType) {
 		return switch (messageType) {
 			case ASSISTANT -> new AssistantMessage(content);

spring-ai-model/src/test/java/org/springframework/ai/chat/memory/InMemoryChatMemoryRepositoryTests.java

@@ -153,4 +153,25 @@ void messagesWithNullElementsNotAllowed() {
 			.hasMessageContaining("messages cannot contain null elements");
 	}
 
+	@Test
+	void refreshConversation() {
+		String conversationId = UUID.randomUUID().toString();
+		List<Message> initialMessages = List.of(new UserMessage("Hello"), new AssistantMessage("Hi"));
+		this.chatMemoryRepository.saveAll(conversationId, initialMessages);
+
+		assertThat(this.chatMemoryRepository.findByConversationId(conversationId)).hasSize(2);
+
+		List<Message> toDelete = List.of(new UserMessage("Hello"));
+		List<Message> toAdd = List.of(new UserMessage("How are you?"), new AssistantMessage("I'm fine, thanks!"));
+
+		this.chatMemoryRepository.refresh(conversationId, toDelete, toAdd);
+
+		List<Message> updatedMessages = this.chatMemoryRepository.findByConversationId(conversationId);
+		assertThat(updatedMessages).hasSize(3);
+		assertThat(updatedMessages).contains(new AssistantMessage("Hi"));
+		assertThat(updatedMessages).contains(new UserMessage("How are you?"));
+		assertThat(updatedMessages).contains(new AssistantMessage("I'm fine, thanks!"));
+		assertThat(updatedMessages).doesNotContain(new UserMessage("Hello"));
+	}
+
 }