Examples with SearchProfileShardResults org.opensearch.search.profile.SearchProfileShardResults used on opensource projects

Example 6 with SearchProfileShardResults

use of org.opensearch.search.profile.SearchProfileShardResults in project OpenSearch by opensearch-project.

the class SearchResponseMergerTests method testMergeProfileResults.

public void testMergeProfileResults() throws InterruptedException {
    SearchTimeProvider searchTimeProvider = new SearchTimeProvider(0, 0, () -> 0);
    SearchResponseMerger merger = new SearchResponseMerger(0, 0, SearchContext.TRACK_TOTAL_HITS_ACCURATE, searchTimeProvider, emptyReduceContextBuilder());
    Map<String, ProfileShardResult> expectedProfile = new HashMap<>();
    for (int i = 0; i < numResponses; i++) {
        SearchProfileShardResults profile = SearchProfileShardResultsTests.createTestItem();
        expectedProfile.putAll(profile.getShardResults());
        SearchHits searchHits = new SearchHits(new SearchHit[0], new TotalHits(0, TotalHits.Relation.EQUAL_TO), Float.NaN);
        InternalSearchResponse internalSearchResponse = new InternalSearchResponse(searchHits, null, null, profile, false, null, 1);
        SearchResponse searchResponse = new SearchResponse(internalSearchResponse, null, 1, 1, 0, 100L, ShardSearchFailure.EMPTY_ARRAY, SearchResponse.Clusters.EMPTY);
        addResponse(merger, searchResponse);
    }
    awaitResponsesAdded();
    assertEquals(numResponses, merger.numResponses());
    SearchResponse.Clusters clusters = SearchResponseTests.randomClusters();
    SearchResponse mergedResponse = merger.getMergedResponse(clusters);
    assertSame(clusters, mergedResponse.getClusters());
    assertEquals(numResponses, mergedResponse.getTotalShards());
    assertEquals(numResponses, mergedResponse.getSuccessfulShards());
    assertEquals(0, mergedResponse.getSkippedShards());
    assertEquals(0, mergedResponse.getFailedShards());
    assertEquals(0, mergedResponse.getShardFailures().length);
    assertEquals(expectedProfile, mergedResponse.getProfileResults());
}