Fixing Integ Test case

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Sep 26, 2023 · badced1 · badced1
1 parent b13eb80
commit badced1
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryProcessorIT.java
@@ -48,7 +48,7 @@ public void tearDown() {
     }
 
     public void testNeuralQueryProcessor() throws Exception {
-        initializeIndexIfNotExist(index);
+        initializeIndexIfNotExist();
         String modelId = getDeployedModelId();
         createSearchRequestProcessor(modelId, search_pipeline);
         createPipelineProcessor(modelId, ingest_pipeline);
@@ -61,12 +61,10 @@ public void testNeuralQueryProcessor() throws Exception {
 
         assertFalse(response.isEmpty());
 
-        assertEquals(modelId, neuralQueryBuilder.modelId());
-
     }
 
-    private void initializeIndexIfNotExist(String indexName) throws IOException {
-        if (index.equals(indexName) && !indexExists(index)) {
+    private void initializeIndexIfNotExist() throws IOException {
+        if (index.equals(NeuralQueryProcessorIT.index) && !indexExists(index)) {
             prepareKnnIndex(
                 index,
                 Collections.singletonList(new KNNFieldConfig(TEST_KNN_VECTOR_FIELD_NAME_1, TEST_DIMENSION, TEST_SPACE_TYPE))