Skip to content

Commit 029e8a1

Browse files
committed
test: fix useAutoConfiguredJdbcChatMemoryRepository test after connection leak fix
Add schema initialization property after commit 560540f properly closed connections Signed-off-by: Soby Chacko <[email protected]>
1 parent 6f61fee commit 029e8a1

File tree

1 file changed

+16
-15
lines changed

1 file changed

+16
-15
lines changed

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

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -73,28 +73,29 @@ void initializeSchemaEmbeddedDefault() {
7373

7474
@Test
7575
void useAutoConfiguredJdbcChatMemoryRepository() {
76-
this.contextRunner.run(context -> {
77-
var chatMemoryRepository = context.getBean(JdbcChatMemoryRepository.class);
78-
var conversationId = UUID.randomUUID().toString();
79-
var userMessage = new UserMessage("Message from the user");
76+
this.contextRunner.withPropertyValues("spring.ai.chat.memory.repository.jdbc.initialize-schema=always")
77+
.run(context -> {
78+
var chatMemoryRepository = context.getBean(JdbcChatMemoryRepository.class);
79+
var conversationId = UUID.randomUUID().toString();
80+
var userMessage = new UserMessage("Message from the user");
8081

81-
chatMemoryRepository.saveAll(conversationId, List.of(userMessage));
82+
chatMemoryRepository.saveAll(conversationId, List.of(userMessage));
8283

83-
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(1);
84-
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(List.of(userMessage));
84+
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(1);
85+
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(List.of(userMessage));
8586

86-
chatMemoryRepository.deleteByConversationId(conversationId);
87+
chatMemoryRepository.deleteByConversationId(conversationId);
8788

88-
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEmpty();
89+
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEmpty();
8990

90-
var multipleMessages = List.<Message>of(new UserMessage("Message from the user 1"),
91-
new AssistantMessage("Message from the assistant 1"));
91+
var multipleMessages = List.<Message>of(new UserMessage("Message from the user 1"),
92+
new AssistantMessage("Message from the assistant 1"));
9293

93-
chatMemoryRepository.saveAll(conversationId, multipleMessages);
94+
chatMemoryRepository.saveAll(conversationId, multipleMessages);
9495

95-
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(multipleMessages.size());
96-
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(multipleMessages);
97-
});
96+
assertThat(chatMemoryRepository.findByConversationId(conversationId)).hasSize(multipleMessages.size());
97+
assertThat(chatMemoryRepository.findByConversationId(conversationId)).isEqualTo(multipleMessages);
98+
});
9899
}
99100

100101
@Test

0 commit comments

Comments
 (0)