fix uts

Author: jimczi

server/src/test/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldTypeTests.java

@@ -26,6 +26,7 @@
 import org.elasticsearch.index.mapper.vectors.DenseVectorFieldMapper.VectorSimilarity;
 import org.elasticsearch.search.DocValueFormat;
 import org.elasticsearch.search.vectors.DenseVectorQuery;
+import org.elasticsearch.search.vectors.DiversifyingParentBlockQuery;
 import org.elasticsearch.search.vectors.ESKnnByteVectorQuery;
 import org.elasticsearch.search.vectors.ESKnnFloatVectorQuery;
 import org.elasticsearch.search.vectors.RescoreKnnVectorQuery;
@@ -240,7 +241,7 @@ public void testCreateNestedKnnQuery() {
                 query = rescoreKnnVectorQuery.innerQuery();
             }
             if (field.getIndexOptions().isFlat()) {
-                assertThat(query, instanceOf(ToParentBlockJoinQuery.class));
+                assertThat(query, instanceOf(DiversifyingParentBlockQuery.class));
             } else {
                 assertThat(query, instanceOf(DiversifyingChildrenFloatKnnVectorQuery.class));
             }

server/src/test/java/org/elasticsearch/search/vectors/RescoreKnnVectorQueryTests.java

@@ -139,7 +139,7 @@ private void checkProfiling(int k, int numDocs, float[] queryVector, IndexReader
             VectorSimilarityFunction.COSINE,
             k,
             k,
-            null
+            innerQuery
         );
         IndexSearcher searcher = newSearcher(reader, true, false);
         searcher.search(rescoreKnnVectorQuery, numDocs);