Use MongoDBAtlasLocalContainer from Testcontainers 1.20.2

spring-ai-spring-boot-autoconfigure/pom.xml

@@ -494,6 +494,13 @@
 			<scope>test</scope>
 		</dependency>
 
+		<dependency>
+			<groupId>org.testcontainers</groupId>
+			<artifactId>mongodb</artifactId>
+			<version>1.20.2</version>
+			<scope>test</scope>
+		</dependency>
+
 		<dependency>
 			<groupId>org.testcontainers</groupId>
 			<artifactId>qdrant</artifactId>

spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/vectorstore/mongo/MongoDBAtlasVectorStoreAutoConfigurationIT.java

@@ -18,13 +18,10 @@
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.springframework.ai.autoconfigure.vectorstore.observation.ObservationTestUtil.assertObservationRegistry;
 
-import java.time.Duration;
 import java.util.Collections;
 import java.util.List;
-import java.util.Map;
 import java.util.stream.Collectors;
 
-import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
 import org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration;
@@ -33,7 +30,6 @@
 import org.springframework.ai.observation.conventions.VectorStoreProvider;
 import org.springframework.ai.vectorstore.SearchRequest;
 import org.springframework.ai.vectorstore.VectorStore;
-import org.springframework.ai.vectorstore.filter.Filter;
 import org.springframework.ai.vectorstore.filter.FilterExpressionBuilder;
 import org.springframework.ai.vectorstore.observation.VectorStoreObservationContext;
 import org.springframework.boot.autoconfigure.AutoConfigurations;
@@ -44,12 +40,11 @@
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.data.mongodb.core.MongoTemplate;
-import org.testcontainers.containers.GenericContainer;
-import org.testcontainers.containers.wait.strategy.Wait;
 import org.testcontainers.junit.jupiter.Container;
 import org.testcontainers.junit.jupiter.Testcontainers;
 
 import io.micrometer.observation.tck.TestObservationRegistry;
+import org.testcontainers.mongodb.MongoDBAtlasLocalContainer;
 
 /**
  * @author Eddú Meléndez
@@ -62,12 +57,7 @@
 class MongoDBAtlasVectorStoreAutoConfigurationIT {
 
 	@Container
-	static GenericContainer<?> mongo = new GenericContainer<>("mongodb/atlas:v1.24.0").withPrivilegedMode(true)
-		.withCommand("/bin/bash", "-c",
-				"atlas deployments setup local-test --type local --port 27778 --bindIpAll --username root --password root --force && tail -f /dev/null")
-		.withExposedPorts(27778)
-		.waitingFor(Wait.forLogMessage(".*Deployment created!.*\\n", 1))
-		.withStartupTimeout(Duration.ofMinutes(5));
+	static MongoDBAtlasLocalContainer mongo = new MongoDBAtlasLocalContainer("mongodb/mongodb-atlas-local:7.0.9");
 
 	List<Document> documents = List.of(
 			new Document("Spring AI rocks!! Spring AI rocks!! Spring AI rocks!! Spring AI rocks!! Spring AI rocks!!",
@@ -94,9 +84,7 @@ class MongoDBAtlasVectorStoreAutoConfigurationIT {
 				// "spring.ai.vectorstore.mongodb.path-name=testembedding",
 				"spring.ai.vectorstore.mongodb.index-name=text_index",
 				"spring.ai.openai.api-key=" + System.getenv("OPENAI_API_KEY"),
-				String.format(
-						"spring.data.mongodb.uri=" + String.format("mongodb://root:root@%s:%s/?directConnection=true",
-								mongo.getHost(), mongo.getMappedPort(27778))));
+				String.format("spring.data.mongodb.uri=" + mongo.getConnectionString()));
 
 	@Test
 	public void addAndSearch() {

vector-stores/spring-ai-mongodb-atlas-store/pom.xml

@@ -53,6 +53,11 @@
             <groupId>org.testcontainers</groupId>
             <artifactId>junit-jupiter</artifactId>
             <scope>test</scope>
+        </dependency><dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>mongodb</artifactId>
+			<version>1.20.2</version>
+            <scope>test</scope>
         </dependency>
 
         <dependency>

vector-stores/spring-ai-mongodb-atlas-store/src/test/java/org/springframework/ai/vectorstore/MongoDBAtlasVectorStoreIT.java

@@ -36,6 +36,7 @@
 
 import org.testcontainers.junit.jupiter.Container;
 import org.testcontainers.junit.jupiter.Testcontainers;
+import org.testcontainers.mongodb.MongoDBAtlasLocalContainer;
 
 import java.util.Arrays;
 import java.util.Collections;
@@ -49,12 +50,14 @@
 /**
  * @author Chris Smith
  * @author Soby Chacko
+ * @author Eddú Meléndez
  */
 @Testcontainers
 class MongoDBAtlasVectorStoreIT {
 
 	@Container
-	private static MongoDBAtlasContainer container = new MongoDBAtlasContainer();
+	private static MongoDBAtlasLocalContainer container = new MongoDBAtlasLocalContainer(
+			"mongodb/mongodb-atlas-local:7.0.9");
 
 	private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
 		.withUserConfiguration(TestApplication.class)

vector-stores/spring-ai-mongodb-atlas-store/src/test/java/org/springframework/ai/vectorstore/MongoDbVectorStoreObservationIT.java

@@ -56,18 +56,21 @@
 import io.micrometer.observation.ObservationRegistry;
 import io.micrometer.observation.tck.TestObservationRegistry;
 import io.micrometer.observation.tck.TestObservationRegistryAssert;
+import org.testcontainers.mongodb.MongoDBAtlasLocalContainer;
 
 /**
  * @author Christian Tzolov
  * @author Soby Chacko
  * @author Thomas Vitale
+ * @author Eddú Meléndez
  */
 @Testcontainers
 @EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
 public class MongoDbVectorStoreObservationIT {
 
 	@Container
-	private static MongoDBAtlasContainer container = new MongoDBAtlasContainer();
+	private static MongoDBAtlasLocalContainer container = new MongoDBAtlasLocalContainer(
+			"mongodb/mongodb-atlas-local:7.0.9");
 
 	private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
 		.withUserConfiguration(Config.class)