From 6065df211c0d058a2172934579e479fb5486a134 Mon Sep 17 00:00:00 2001 From: rustagir Date: Wed, 25 Sep 2024 10:41:37 -0400 Subject: [PATCH] test fixes --- examples/src/test/kotlin/AggregatesBuilderTest.kt | 7 ++++--- examples/src/test/kotlin/SearchIndexesTest.kt | 1 + .../AggregatesBuilderTest.snippet.vector-search.kt | 3 +-- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/examples/src/test/kotlin/AggregatesBuilderTest.kt b/examples/src/test/kotlin/AggregatesBuilderTest.kt index 7baabe05..707fe312 100644 --- a/examples/src/test/kotlin/AggregatesBuilderTest.kt +++ b/examples/src/test/kotlin/AggregatesBuilderTest.kt @@ -28,7 +28,8 @@ import com.mongodb.client.model.geojson.Position import com.mongodb.client.model.search.SearchOperator import com.mongodb.client.model.search.SearchOptions import com.mongodb.client.model.search.SearchPath -import com.mongodb.client.model.search.VectorSearchOptions.vectorSearchOptions +import com.mongodb.client.model.search.VectorSearchOptions +import com.mongodb.client.model.search.VectorSearchOptions.exactVectorSearchOptions import com.mongodb.kotlin.client.coroutine.MongoClient import config.getConfig import kotlinx.coroutines.flow.firstOrNull @@ -981,13 +982,13 @@ class AggregatesBuilderTest { SearchPath.fieldPath(MovieAlt::plotEmbedding.name), listOf(-0.0072121937, -0.030757688, -0.012945653), "mflix_movies_embedding_index", - 2.toLong(), 1.toLong(), - vectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016)) + exactVectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016)) ) // :snippet-end: ) ) + val results = resultsFlow.toList() assertEquals(1, resultsFlow.toList().size) assertEquals(1, results.first().get("count", Document::class.java).get("lowerBound", java.lang.Long::class.java)?.toInt()) diff --git a/examples/src/test/kotlin/SearchIndexesTest.kt b/examples/src/test/kotlin/SearchIndexesTest.kt index 47b8b44f..e3448803 100644 --- a/examples/src/test/kotlin/SearchIndexesTest.kt +++ b/examples/src/test/kotlin/SearchIndexesTest.kt @@ -1,5 +1,6 @@ import com.mongodb.client.model.SearchIndexModel +import com.mongodb.client.model.SearchIndexType import com.mongodb.kotlin.client.coroutine.MongoClient import config.getConfig import kotlinx.coroutines.flow.toList diff --git a/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt b/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt index 7be37509..ee4648f2 100644 --- a/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt +++ b/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt @@ -2,7 +2,6 @@ Aggregates.vectorSearch( SearchPath.fieldPath(MovieAlt::plotEmbedding.name), listOf(-0.0072121937, -0.030757688, -0.012945653), "mflix_movies_embedding_index", - 2.toLong(), 1.toLong(), - vectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016)) + exactVectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016)) )