5252import static org .opensearch .test .framework .TestSecurityConfig .AuthcDomain .AUTHC_HTTPBASIC_INTERNAL ;
5353import static org .opensearch .test .framework .TestSecurityConfig .Role .ALL_ACCESS ;
5454import static org .opensearch .test .framework .cluster .SearchRequestFactory .averageAggregationRequest ;
55+ import static org .opensearch .test .framework .cluster .SearchRequestFactory .searchRequestWithSort ;
5556import static org .opensearch .test .framework .matcher .SearchResponseMatchers .containAggregationWithNameAndType ;
5657import static org .opensearch .test .framework .matcher .SearchResponseMatchers .isSuccessfulSearchResponse ;
5758import static org .opensearch .test .framework .matcher .SearchResponseMatchers .numberOfTotalHitsIsEqualTo ;
@@ -281,7 +282,7 @@ public static void createTestData() {
281282 public void testShouldSearchAll () throws IOException {
282283
283284 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_ALL_USER )) {
284- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
285+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
285286 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
286287
287288 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -293,15 +294,15 @@ public void testShouldSearchAll() throws IOException {
293294 assertThat (searchResponse , searchHitContainsFieldWithValue (4 , FIELD_ARTIST , ARTIST_YES ));
294295 assertThat (searchResponse , searchHitContainsFieldWithValue (5 , FIELD_ARTIST , ARTIST_UNKNOWN ));
295296
296- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
297+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
297298 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
298299
299300 assertThat (searchResponse , isSuccessfulSearchResponse ());
300301 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (4 ));
301302 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_NO ));
302303 }
303304 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_FIRST_AND_SECOND_USER )) {
304- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
305+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
305306 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
306307
307308 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -313,7 +314,7 @@ public void testShouldSearchAll() throws IOException {
313314 assertThat (searchResponse , searchHitContainsFieldWithValue (4 , FIELD_ARTIST , ARTIST_YES ));
314315 assertThat (searchResponse , searchHitContainsFieldWithValue (5 , FIELD_ARTIST , ARTIST_UNKNOWN ));
315316
316- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
317+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
317318 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
318319
319320 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -326,14 +327,14 @@ public void testShouldSearchAll() throws IOException {
326327 public void testShouldSearchI1_S2I2_S3 () throws IOException {
327328
328329 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_WHERE_FIELD_ARTIST_MATCHES_ARTIST_STRING )) {
329- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
330+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
330331 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
331332
332333 assertThat (searchResponse , isSuccessfulSearchResponse ());
333334 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (1 ));
334335 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_STRING ));
335336
336- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
337+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
337338 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
338339
339340 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -349,15 +350,15 @@ public void testShouldSearchI1_S3I1_S6I2_S2() throws IOException {
349350 READ_WHERE_FIELD_ARTIST_MATCHES_ARTIST_TWINS_OR_FIELD_STARS_GREATER_THAN_FIVE
350351 )
351352 ) {
352- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
353+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
353354 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
354355
355356 assertThat (searchResponse , isSuccessfulSearchResponse ());
356357 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (2 ));
357358 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_TWINS ));
358359 assertThat (searchResponse , searchHitContainsFieldWithValue (1 , FIELD_ARTIST , ARTIST_UNKNOWN ));
359360
360- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
361+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
361362 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
362363
363364 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -373,15 +374,15 @@ public void testShouldSearchI1_S1I1_S3I2_S2I2_S4() throws IOException {
373374 READ_WHERE_FIELD_ARTIST_MATCHES_ARTIST_TWINS_OR_MATCHES_ARTIST_FIRST
374375 )
375376 ) {
376- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
377+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
377378 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
378379
379380 assertThat (searchResponse , isSuccessfulSearchResponse ());
380381 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (2 ));
381382 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_TWINS ));
382383 assertThat (searchResponse , searchHitContainsFieldWithValue (1 , FIELD_ARTIST , ARTIST_FIRST ));
383384
384- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
385+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
385386 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
386387
387388 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -394,15 +395,15 @@ public void testShouldSearchI1_S1I1_S3I2_S2I2_S4() throws IOException {
394395 public void testShouldSearchStarsLessThanThree () throws IOException {
395396
396397 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_WHERE_STARS_LESS_THAN_THREE )) {
397- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_NAME );
398+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_NAME );
398399 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
399400
400401 assertThat (searchResponse , isSuccessfulSearchResponse ());
401402 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (2 ));
402403 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_FIRST ));
403404 assertThat (searchResponse , searchHitContainsFieldWithValue (1 , FIELD_ARTIST , ARTIST_STRING ));
404405
405- searchRequest = new SearchRequest (SECOND_INDEX_NAME );
406+ searchRequest = searchRequestWithSort (SECOND_INDEX_NAME );
406407 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
407408
408409 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -417,7 +418,7 @@ public void testSearchForAllDocumentsWithIndexPattern() throws IOException {
417418
418419 // DLS
419420 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_ALL_USER )) {
420- SearchRequest searchRequest = new SearchRequest ("*" .concat (FIRST_INDEX_NAME ));
421+ SearchRequest searchRequest = searchRequestWithSort ("*" .concat (FIRST_INDEX_NAME ));
421422 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
422423
423424 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -429,7 +430,7 @@ public void testSearchForAllDocumentsWithIndexPattern() throws IOException {
429430 assertThat (searchResponse , searchHitContainsFieldWithValue (4 , FIELD_ARTIST , ARTIST_YES ));
430431 assertThat (searchResponse , searchHitContainsFieldWithValue (5 , FIELD_ARTIST , ARTIST_UNKNOWN ));
431432
432- searchRequest = new SearchRequest ("*" .concat (SECOND_INDEX_NAME ));
433+ searchRequest = searchRequestWithSort ("*" .concat (SECOND_INDEX_NAME ));
433434 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
434435
435436 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -442,7 +443,7 @@ public void testSearchForAllDocumentsWithIndexPattern() throws IOException {
442443 public void testSearchForAllDocumentsWithAlias () throws IOException {
443444
444445 try (RestHighLevelClient restHighLevelClient = cluster .getRestHighLevelClient (READ_ALL_USER )) {
445- SearchRequest searchRequest = new SearchRequest (FIRST_INDEX_ALIAS );
446+ SearchRequest searchRequest = searchRequestWithSort (FIRST_INDEX_ALIAS );
446447 SearchResponse searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
447448
448449 assertThat (searchResponse , isSuccessfulSearchResponse ());
@@ -454,12 +455,15 @@ public void testSearchForAllDocumentsWithAlias() throws IOException {
454455 assertThat (searchResponse , searchHitContainsFieldWithValue (4 , FIELD_ARTIST , ARTIST_YES ));
455456 assertThat (searchResponse , searchHitContainsFieldWithValue (5 , FIELD_ARTIST , ARTIST_UNKNOWN ));
456457
457- searchRequest = new SearchRequest ("*" .concat (SECOND_INDEX_NAME ));
458+ searchRequest = searchRequestWithSort ("*" .concat (SECOND_INDEX_NAME ));
458459 searchResponse = restHighLevelClient .search (searchRequest , DEFAULT );
459460
460461 assertThat (searchResponse , isSuccessfulSearchResponse ());
461462 assertThat (searchResponse , numberOfTotalHitsIsEqualTo (4 ));
462463 assertThat (searchResponse , searchHitContainsFieldWithValue (0 , FIELD_ARTIST , ARTIST_NO ));
464+ assertThat (searchResponse , searchHitContainsFieldWithValue (1 , FIELD_ARTIST , ARTIST_TWINS ));
465+ assertThat (searchResponse , searchHitContainsFieldWithValue (2 , FIELD_ARTIST , ARTIST_STRING ));
466+ assertThat (searchResponse , searchHitContainsFieldWithValue (3 , FIELD_ARTIST , ARTIST_FIRST ));
463467 }
464468 }
465469
0 commit comments