[Fix] NeuralSparseTwoPhaseProcessorIT created wrong ingest pipeline, …

…fix it to correct API. Signed-off-by: conggguan <[email protected]>
opensearch-project · Jul 8, 2024 · 15aeb02 · 15aeb02
1 parent 363bd18
commit 15aeb02
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...pgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java b/...pgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java
@@ -10,6 +10,7 @@
 
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.util.List;
 
 import static org.opensearch.neuralsearch.util.TestUtils.NODES_BWC_CLUSTER;
 import static org.opensearch.neuralsearch.util.TestUtils.TEXT_EMBEDDING_PROCESSOR;
@@ -29,13 +30,13 @@ public void testNeuralSparseQueryTwoPhaseProcessor_NeuralSearch_E2EFlow() throws
             String modelId = uploadSparseEncodingModel();
             loadModel(modelId);
             neuralSparseQueryBuilder.modelId(modelId);
-            createPipelineProcessor(modelId, NEURAL_SPARSE_INGEST_PIPELINE_NAME);
+            createPipelineForSparseEncodingProcessor(modelId, NEURAL_SPARSE_INGEST_PIPELINE_NAME);
             createIndexWithConfiguration(
                 getIndexNameForTest(),
                 Files.readString(Path.of(classLoader.getResource("processor/IndexMappingMultipleShard.json").toURI())),
                 NEURAL_SPARSE_INGEST_PIPELINE_NAME
             );
-            addDocument(getIndexNameForTest(), "0", TEST_TEXT_FIELD, TEXT_1, null, null);
+            addSparseEncodingDoc(getIndexNameForTest(), "0", List.of(), List.of(), List.of(TEST_TEXT_FIELD), List.of(TEXT_1));
             createNeuralSparseTwoPhaseSearchProcessor(NEURAL_SPARSE_TWO_PHASE_SEARCH_PIPELINE_NAME);
             updateIndexSettings(
                 getIndexNameForTest(),