GITHUB-12181: fix false-positive TestKnnFloatVectorQuery.testDocAndScoreQueryBasics() failure (#12182)

cpoerschke · web-flow · commit 3bd06b1cb94f · 2023-03-06T15:29:36.000Z
diff --git a/lucene/core/src/test/org/apache/lucene/search/TestKnnFloatVectorQuery.java b/lucene/core/src/test/org/apache/lucene/search/TestKnnFloatVectorQuery.java
@@ -201,11 +201,12 @@ public void testDocAndScoreQueryBasics() throws IOException {
           scores[i] = scoreDocs[i].score;
           maxScore = Math.max(maxScore, scores[i]);
         }
-        int[] segments = AbstractKnnVectorQuery.findSegmentStarts(reader, docs);
+        IndexReader indexReader = searcher.getIndexReader();
+        int[] segments = AbstractKnnVectorQuery.findSegmentStarts(indexReader, docs);
 
         AbstractKnnVectorQuery.DocAndScoreQuery query =
             new AbstractKnnVectorQuery.DocAndScoreQuery(
-                docs, scores, maxScore, segments, reader.getContext().id());
+                docs, scores, maxScore, segments, indexReader.getContext().id());
         final Weight w = query.createWeight(searcher, ScoreMode.TOP_SCORES, 1.0f);
         TopDocs topDocs = searcher.search(query, 100);
         assertEquals(scoreDocs.length, topDocs.totalHits.value);
