Only set request to default if null

Signed-off-by: Craig Perkins <[email protected]>
cwperks · Dec 15, 2023 · d1e25f3 · d1e25f3
1 parent 340162d
commit d1e25f3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/...les/lang-mustache/src/main/java/org/opensearch/script/mustache/SearchTemplateRequest.java b/...les/lang-mustache/src/main/java/org/opensearch/script/mustache/SearchTemplateRequest.java
@@ -120,7 +120,7 @@ public boolean isSimulate() {
     }
 
     public void setSimulate(boolean simulate) {
-        if (simulate) {
+        if (simulate && request == null) {
             this.request = SIMULATED_REQUEST;
         }
         this.simulate = simulate;

diff --git a/...ang-mustache/src/test/java/org/opensearch/script/mustache/SearchTemplateRequestTests.java b/...ang-mustache/src/test/java/org/opensearch/script/mustache/SearchTemplateRequestTests.java
@@ -114,10 +114,17 @@ public static SearchTemplateRequest createRandomRequest() {
 
     public void testSimulatedSearchTemplateRequest() {
         SearchTemplateRequest request = createRandomRequest();
+        request.setRequest(null);
         request.setSimulate(true);
 
         assertEquals(0, request.indices().length);
         assertEquals(SearchRequest.DEFAULT_INDICES_OPTIONS, request.indicesOptions());
         assertEquals(2, request.indices("index1", "index2").indices().length);
+
+        SearchTemplateRequest randomRequest = createRandomRequest();
+        int expectedIndicesLength = randomRequest.indices().length;
+        request.setSimulate(true);
+
+        assertEquals(expectedIndicesLength, randomRequest.indices().length);
     }
 }