remove null from mock get result of field

Signed-off-by: bowenlan-amzn <[email protected]>
bowenlan-amzn · Aug 2, 2024 · b18b44e · b18b44e
1 parent a82a896
commit b18b44e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/src/test/java/org/opensearch/index/query/TermsQueryBuilderTests.java b/server/src/test/java/org/opensearch/index/query/TermsQueryBuilderTests.java
@@ -251,7 +251,13 @@ public GetResponse executeGet(GetRequest getRequest) {
             throw new OpenSearchException("boom", ex);
         }
         Map<String, DocumentField> documentField = new HashMap<>();
-        documentField.put(termsPath, new DocumentField(termsPath, randomTerms));
+        List<Object> nonNullTerms = new ArrayList<>();
+        for (Object obj : randomTerms) {
+            if (obj != null) {
+                nonNullTerms.add(obj);
+            }
+        }
+        documentField.put(termsPath, new DocumentField(termsPath, nonNullTerms));
         return new GetResponse(
             new GetResult(getRequest.index(), getRequest.id(), 0, 1, 0, true, new BytesArray(json), documentField, null)
         );