Switch processor IT to use Lucene (opensearch-project#48)

Updated processor test to use Lucene engine. Refactored ingest document to refresh after ingestion and confirm doc gets indexed. Signed-off-by: John Mazanec <[email protected]>
zhichao-aws · Nov 1, 2022 · 0b8c0d1 · 0b8c0d1
1 parent e30285b
commit 0b8c0d1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java
@@ -32,6 +32,7 @@ public void testTextEmbeddingProcessor() throws Exception {
         createPipelineProcessor(modelId, PIPELINE_NAME);
         createTextEmbeddingIndex();
         ingestDocument();
+        assertEquals(1, getDocCount(INDEX_NAME));
     }
 
     private String uploadTextEmbeddingModel() throws Exception {
@@ -64,7 +65,7 @@ private void ingestDocument() throws Exception {
         Response response = makeRequest(
             client(),
             "POST",
-            INDEX_NAME + "/_doc",
+            INDEX_NAME + "/_doc?refresh",
             null,
             toHttpEntity(ingestDocument),
             ImmutableList.of(new BasicHeader(HttpHeaders.USER_AGENT, "Kibana"))

diff --git a/src/test/resources/processor/IndexMappings.json b/src/test/resources/processor/IndexMappings.json
@@ -17,7 +17,7 @@
         "method": {
           "name": "hnsw",
           "space_type": "l2",
-          "engine": "nmslib",
+          "engine": "lucene",
           "parameters": {
             "ef_construction": 128,
             "m": 24
@@ -30,7 +30,7 @@
         "method": {
           "name": "hnsw",
           "space_type": "l2",
-          "engine": "nmslib",
+          "engine": "lucene",
           "parameters": {
             "ef_construction": 128,
             "m": 24
@@ -43,7 +43,7 @@
         "method": {
           "name": "hnsw",
           "space_type": "l2",
-          "engine": "nmslib",
+          "engine": "lucene",
           "parameters": {
             "ef_construction": 128,
             "m": 24
@@ -59,7 +59,7 @@
             "method": {
               "name": "hnsw",
               "space_type": "l2",
-              "engine": "nmslib",
+              "engine": "lucene",
               "parameters": {
                 "ef_construction": 128,
                 "m": 24