refine vector index wait strategy

Author: christophstrobl

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/repository/aot/MongoRepositoryContributorTests.java

@@ -32,6 +32,7 @@
 
 import org.bson.BsonString;
 import org.bson.Document;
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -151,18 +152,26 @@ private static void initializeVectorIndex() {
 						.append("queryVector", List.of(1.0, 1.12345, 2.23456, 3.34567, 4.45678)));
 
 		// wait for search index to serve data
-		Awaitility.await().atMost(Duration.ofSeconds(120)).pollInterval(Duration.ofMillis(200)).until(() -> {
-			try (MongoCursor<Document> cursor = userCollection.aggregate(List.of($vectorSearch)).iterator()) {
-				return cursor.hasNext();
-			}
-		});
+		Awaitility.await().atLeast(Duration.ofMillis(50)).atMost(Duration.ofSeconds(120)).ignoreExceptions()
+				.pollInterval(Duration.ofMillis(250)).until(() -> {
+					try (MongoCursor<Document> cursor = userCollection.aggregate(List.of($vectorSearch)).iterator()) {
+						if (cursor.hasNext()) {
+							Document next = cursor.next();
+							return true;
+						}
+						return false;
+					}
+				});
 	}
 
 	@BeforeEach
 	void beforeEach() throws InterruptedException {
+		initUsers();
+	}
 
+	@AfterEach
+	void afterEach() {
 		MongoTestUtils.flushCollection(DB_NAME, "user", client);
-		initUsers();
 	}
 
 	@Test