Skip to content

Commit ce7381c

Browse files
committed
Fix MongoChatMemoryAutoConfigurationIT
- Import MongoDB autoconfigurations via @EnableAutoConfiguration instead of manual import - Make SpringBootTest non-web environment Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
1 parent d7a6435 commit ce7381c

File tree

1 file changed

+9
-3
lines changed
  • auto-configurations/models/chat/memory/repository/spring-ai-autoconfigure-model-chat-memory-repository-mongodb/src/test/java/org/springframework/ai/model/chat/memory/repository/mongo/autoconfigure

1 file changed

+9
-3
lines changed

auto-configurations/models/chat/memory/repository/spring-ai-autoconfigure-model-chat-memory-repository-mongodb/src/test/java/org/springframework/ai/model/chat/memory/repository/mongo/autoconfigure/MongoChatMemoryAutoConfigurationIT.java

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,16 +27,16 @@
2727
import org.springframework.ai.chat.memory.repository.mongo.Conversation;
2828
import org.springframework.ai.chat.messages.SystemMessage;
2929
import org.springframework.beans.factory.annotation.Autowired;
30-
import org.springframework.boot.mongodb.autoconfigure.MongoAutoConfiguration;
30+
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
3131
import org.springframework.boot.test.context.SpringBootTest;
3232
import org.springframework.boot.testcontainers.service.connection.ServiceConnection;
33+
import org.springframework.context.annotation.Configuration;
3334
import org.springframework.data.mongodb.core.MongoTemplate;
3435
import org.springframework.test.context.TestPropertySource;
3536

3637
import static org.assertj.core.api.Assertions.assertThat;
3738

38-
@SpringBootTest(classes = { MongoAutoConfiguration.class, MongoChatMemoryAutoConfiguration.class,
39-
MongoChatMemoryIndexCreatorAutoConfiguration.class })
39+
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
4040
@TestPropertySource(properties = { "spring.ai.chat.memory.repository.mongo.create-indices=true" })
4141
class MongoChatMemoryAutoConfigurationIT {
4242

@@ -77,4 +77,10 @@ void indicesShouldBeCreated() {
7777
assertThat(this.mongoTemplate.indexOps(Conversation.class).getIndexInfo().size()).isEqualTo(2);
7878
}
7979

80+
@Configuration
81+
@EnableAutoConfiguration
82+
static class TestConfiguration {
83+
84+
}
85+
8086
}

0 commit comments

Comments
 (0)