fix unit tests for inference processor

Signed-off-by: yuye-aws <[email protected]>
yuye-aws · Mar 1, 2024 · 6065b6f · 6065b6f
1 parent cd323ab
commit 6065b6f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/src/test/java/org/opensearch/neuralsearch/processor/SparseEncodingProcessorTests.java b/src/test/java/org/opensearch/neuralsearch/processor/SparseEncodingProcessorTests.java
@@ -51,6 +51,9 @@ public class SparseEncodingProcessorTests extends OpenSearchTestCase {
     @Mock
     private Environment env;
 
+    @Mock
+    private ProcessorInputValidator processorInputValidator;
+
     @InjectMocks
     private SparseEncodingProcessorFactory SparseEncodingProcessorFactory;
     private static final String PROCESSOR_TAG = "mockTag";
@@ -97,7 +100,11 @@ public void testExecute_whenInferenceTextListEmpty_SuccessWithoutAnyMap() {
         IngestDocument ingestDocument = new IngestDocument(sourceAndMetadata, new HashMap<>());
         Map<String, Processor.Factory> registry = new HashMap<>();
         MLCommonsClientAccessor accessor = mock(MLCommonsClientAccessor.class);
-        SparseEncodingProcessorFactory sparseEncodingProcessorFactory = new SparseEncodingProcessorFactory(accessor, env);
+        SparseEncodingProcessorFactory sparseEncodingProcessorFactory = new SparseEncodingProcessorFactory(
+            accessor,
+            env,
+            processorInputValidator
+        );
 
         Map<String, Object> config = new HashMap<>();
         config.put(TextEmbeddingProcessor.MODEL_ID_FIELD, "mockModelId");

diff --git a/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorTests.java b/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorTests.java
@@ -51,6 +51,9 @@ public class TextEmbeddingProcessorTests extends OpenSearchTestCase {
     @Mock
     private Environment env;
 
+    @Mock
+    private ProcessorInputValidator processorInputValidator;
+
     @InjectMocks
     private TextEmbeddingProcessorFactory textEmbeddingProcessorFactory;
     private static final String PROCESSOR_TAG = "mockTag";
@@ -127,7 +130,11 @@ public void testExecute_whenInferenceThrowInterruptedException_throwRuntimeExcep
         IngestDocument ingestDocument = new IngestDocument(sourceAndMetadata, new HashMap<>());
         Map<String, Processor.Factory> registry = new HashMap<>();
         MLCommonsClientAccessor accessor = mock(MLCommonsClientAccessor.class);
-        TextEmbeddingProcessorFactory textEmbeddingProcessorFactory = new TextEmbeddingProcessorFactory(accessor, env);
+        TextEmbeddingProcessorFactory textEmbeddingProcessorFactory = new TextEmbeddingProcessorFactory(
+            accessor,
+            env,
+            processorInputValidator
+        );
 
         Map<String, Object> config = new HashMap<>();
         config.put(TextEmbeddingProcessor.MODEL_ID_FIELD, "mockModelId");
@@ -145,7 +152,11 @@ public void testExecute_whenInferenceTextListEmpty_SuccessWithoutEmbedding() {
         IngestDocument ingestDocument = new IngestDocument(sourceAndMetadata, new HashMap<>());
         Map<String, Processor.Factory> registry = new HashMap<>();
         MLCommonsClientAccessor accessor = mock(MLCommonsClientAccessor.class);
-        TextEmbeddingProcessorFactory textEmbeddingProcessorFactory = new TextEmbeddingProcessorFactory(accessor, env);
+        TextEmbeddingProcessorFactory textEmbeddingProcessorFactory = new TextEmbeddingProcessorFactory(
+            accessor,
+            env,
+            processorInputValidator
+        );
 
         Map<String, Object> config = new HashMap<>();
         config.put(TextEmbeddingProcessor.MODEL_ID_FIELD, "mockModelId");