fix unit tests

Author: Samiul-TheSoccerFan

x-pack/plugin/inference/src/test/java/org/elasticsearch/index/query/SemanticKnnVectorQueryRewriteInterceptorTests.java

@@ -65,7 +65,7 @@ public void testKnnQueryWithVectorBuilderIsInterceptedAndRewritten() throws IOEx
     }
 
     public void testKnnQueryWithVectorBuilderIsInterceptedAndRewrittenWithBoostAndQueryName() throws IOException {
-        float BOOST = 2.0f;
+        float BOOST = 5.0f;
         String QUERY_NAME = "knn_query";
 
         Map<String, InferenceFieldMetadata> inferenceFields = Map.of(
@@ -81,10 +81,12 @@ public void testKnnQueryWithVectorBuilderIsInterceptedAndRewrittenWithBoostAndQu
         testRewrittenInferenceQuery(context, original);
         QueryBuilder rewritten = original.rewrite(context);
         InterceptedQueryBuilderWrapper intercepted = (InterceptedQueryBuilderWrapper) rewritten;
+        assertEquals(BOOST, intercepted.boost(), 0.0f);
+        assertEquals(QUERY_NAME, intercepted.queryName());
         NestedQueryBuilder nestedQueryBuilder = (NestedQueryBuilder) intercepted.queryBuilder;
         KnnVectorQueryBuilder knnVectorQueryBuilder = (KnnVectorQueryBuilder) nestedQueryBuilder.query();
-        assertEquals(BOOST, knnVectorQueryBuilder.boost(), 0.0f);
-        assertEquals(QUERY_NAME, knnVectorQueryBuilder.queryName());
+        assertEquals(BOOST, knnVectorQueryBuilder.boost(), 5.0f);
+        assertNull(knnVectorQueryBuilder.queryName());
     }
 
     public void testKnnWithQueryBuilderWithoutInferenceIdIsInterceptedAndRewritten() throws IOException {

x-pack/plugin/inference/src/test/java/org/elasticsearch/index/query/SemanticMatchQueryRewriteInterceptorTests.java

@@ -37,7 +37,7 @@ public class SemanticMatchQueryRewriteInterceptorTests extends ESTestCase {
     private static final String FIELD_NAME = "fieldName";
     private static final String VALUE = "value";
     private static final String QUERY_NAME = "match_query";
-    private static final float BOOST = 2.0f;
+    private static final float BOOST = 5.0f;
 
     @Before
     public void setup() {

x-pack/plugin/inference/src/test/java/org/elasticsearch/index/query/SemanticSparseVectorQueryRewriteInterceptorTests.java

@@ -111,7 +111,7 @@ public void testSparseVectorQueryOnNonInferenceFieldRemainsUnchanged() throws IO
     }
 
     public void testBoostAndQueryNameOnSparseVectorQueryRewrite() throws IOException {
-        float BOOST = 2.0f;
+        float BOOST = 5.0f;
         String QUERY_NAME = "sparse_vector_query";
 
         Map<String, InferenceFieldMetadata> inferenceFields = Map.of(
@@ -128,7 +128,7 @@ public void testBoostAndQueryNameOnSparseVectorQueryRewrite() throws IOException
             rewritten instanceof InterceptedQueryBuilderWrapper
         );
         InterceptedQueryBuilderWrapper intercepted = (InterceptedQueryBuilderWrapper) rewritten;
-        assertEquals(BOOST, intercepted.boost(), 1.0f);
+        assertEquals(BOOST, intercepted.boost(), 0.0f);
         assertEquals(QUERY_NAME, intercepted.queryName());
         assertTrue(intercepted.queryBuilder instanceof NestedQueryBuilder);
         NestedQueryBuilder nestedQueryBuilder = (NestedQueryBuilder) intercepted.queryBuilder;
@@ -139,7 +139,8 @@ public void testBoostAndQueryNameOnSparseVectorQueryRewrite() throws IOException
         assertEquals(SemanticTextField.getEmbeddingsFieldName(FIELD_NAME), sparseVectorQueryBuilder.getFieldName());
         assertEquals(INFERENCE_ID, sparseVectorQueryBuilder.getInferenceId());
         assertEquals(QUERY, sparseVectorQueryBuilder.getQuery());
-        assertEquals(BOOST, sparseVectorQueryBuilder.boost(), 1.0f);
+        assertEquals(BOOST, sparseVectorQueryBuilder.boost(), 5.0f);
+        assertNull(sparseVectorQueryBuilder.queryName());
     }
 
     private QueryRewriteContext createQueryRewriteContext(Map<String, InferenceFieldMetadata> inferenceFields) {