Skip to content

Commit b33ef49

Browse files
committed
Added test for SearchTemplateRequestBuilder
Signed-off-by: Owais <[email protected]>
1 parent 0bdc3bd commit b33ef49

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

modules/lang-mustache/src/internalClusterTest/java/org/opensearch/script/mustache/SearchTemplateIT.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -437,9 +437,19 @@ public void testMultiSearchTemplateWithSearchPipeline() throws Exception {
437437
SearchTemplateRequest request2 = SearchTemplateRequest.fromXContent(createParser(JsonXContent.jsonXContent, template2));
438438
request2.setRequest(new SearchRequest("my-nlp-index1"));
439439

440+
Map<String, Object> templateParams = new HashMap<>();
441+
templateParams.put("play_name", "hello");
442+
443+
// Create second template with SearchTemplateRequestBuilder
444+
SearchTemplateResponse response2 = new SearchTemplateRequestBuilder(client()).setRequest(new SearchRequest("my-nlp-index1"))
445+
.setScript("search_template_2")
446+
.setScriptType(ScriptType.STORED)
447+
.setScriptParams(templateParams)
448+
.setSearchPipeline("my_pipeline1")
449+
.get();
450+
440451
// Execute both requests
441452
SearchTemplateResponse response1 = client().execute(SearchTemplateAction.INSTANCE, request1).get();
442-
SearchTemplateResponse response2 = client().execute(SearchTemplateAction.INSTANCE, request2).get();
443453

444454
assertNotNull(response1.getResponse());
445455
assertNotNull(response2.getResponse());

0 commit comments

Comments
 (0)