From 790b6a24a7f74c082cfb9ec6f6356f9296dc30f5 Mon Sep 17 00:00:00 2001
From: Ian Menendez <ianfmenendezd@gmail.com>
Date: Fri, 13 Sep 2024 23:54:53 -0300
Subject: [PATCH] fix: test

Signed-off-by: Ian Menendez <ianfmenendezd@gmail.com>
---
 .../processor/TextChunkingProcessorIT.java           | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/test/java/org/opensearch/neuralsearch/processor/TextChunkingProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/TextChunkingProcessorIT.java
index 088c4d039..9d72cb778 100644
--- a/src/test/java/org/opensearch/neuralsearch/processor/TextChunkingProcessorIT.java
+++ b/src/test/java/org/opensearch/neuralsearch/processor/TextChunkingProcessorIT.java
@@ -188,7 +188,17 @@ public void testTextChunkingProcessor_withIgnoreMissing() {
             createTextChunkingIndex(INDEX_NAME, IGNORE_MISSING_PIPELINE_NAME);
             ingestDocument(TEST_MISSING_FIELD_DOCUMENT);
 
-            validateIndexIngestResults(INDEX_NAME, OUTPUT_FIELD, null);
+            assertEquals(1, getDocCount(INDEX_NAME));
+            MatchAllQueryBuilder query = new MatchAllQueryBuilder();
+            Map<String, Object> searchResults = search(INDEX_NAME, query, 10);
+            assertNotNull(searchResults);
+            Map<String, Object> document = getFirstInnerHit(searchResults);
+            assertNotNull(document);
+            Object documentSource = document.get("_source");
+            assert (documentSource instanceof Map);
+            @SuppressWarnings("unchecked")
+            Map<String, Object> documentSourceMap = (Map<String, Object>) documentSource;
+            assert !(documentSourceMap).containsKey(OUTPUT_FIELD);
         } finally {
             wipeOfTestResources(INDEX_NAME, IGNORE_MISSING_PIPELINE_NAME, null, null);
         }