Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: fix useAutoConfiguredJdbcChatMemoryRepository test after connection leak fix

Browse Source 
Add schema initialization property after commit 560540f properly closed connections

Signed-off-by: Soby Chacko <soby.chacko@broadcom.com>
This commit is contained in:
Soby Chacko
2025-05-21 19:34:15 -04:00
parent 6f61fee774
commit 029e8a10af
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
auto-configurations/models/chat/memory/repository/spring-ai-autoconfigure-model-chat-memory-repository-jdbc/src/test/java/org/springframework/ai/model/chat/memory/repository/jdbc/autoconfigure/JdbcChatMemoryRepositoryPostgresqlAutoConfigurationIT.java
View File

@@ -73,28 +73,29 @@ class JdbcChatMemoryRepositoryPostgresqlAutoConfigurationIT {
@Test
void useAutoConfiguredJdbcChatMemoryRepository() {
this.contextRunner.run(context -> {
var chatMemoryRepository = context.getBean(JdbcChatMemoryRepository.class);
var conversationId = UUID.randomUUID().toString();
var userMessage = new UserMessage("Message from the user");
this.contextRunner.withPropertyValues("spring.ai.chat.memory.repository.jdbc.initialize-schema=always")
.run(context -> {
var chatMemoryRepository = context.getBean(JdbcChatMemoryRepository.class);
var conversationId = UUID.randomUUID().toString();
var userMessage = new UserMessage("Message from the user");
chatMemoryRepository.saveAll(conversationId, List.of(userMessage));
chatMemoryRepository.saveAll(conversationId, List.of(userMessage));
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(1);
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(List.of(userMessage));
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(1);
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(List.of(userMessage));
chatMemoryRepository.deleteByConversationId(conversationId);
chatMemoryRepository.deleteByConversationId(conversationId);
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEmpty();
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEmpty();
var multipleMessages = List.<Message>of(new UserMessage("Message from the user 1"),
new AssistantMessage("Message from the assistant 1"));
var multipleMessages = List.<Message>of(new UserMessage("Message from the user 1"),
new AssistantMessage("Message from the assistant 1"));
chatMemoryRepository.saveAll(conversationId, multipleMessages);
chatMemoryRepository.saveAll(conversationId, multipleMessages);
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(multipleMessages.size());
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(multipleMessages);
});
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(multipleMessages.size());
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(multipleMessages);
});
}
@Test