added tests

Signed-off-by: Andre van de Ven <andrebvandeven@gmail.com>

Author: Andre van de Ven

server/src/main/java/org/opensearch/search/profile/fetch/FetchProfileShardResult.java

@@ -71,7 +71,4 @@ public static FetchProfileShardResult fromXContent(XContentParser parser) throws
         }
         return new FetchProfileShardResult(results);
     }
-
-
-
 }

server/src/test/java/org/opensearch/search/profile/SearchProfileShardResultsTests.java

@@ -38,6 +38,7 @@
 import org.opensearch.core.xcontent.XContentParser;
 import org.opensearch.search.profile.aggregation.AggregationProfileShardResult;
 import org.opensearch.search.profile.aggregation.AggregationProfileShardResultTests;
+import org.opensearch.search.profile.fetch.FetchProfileShardResult;
 import org.opensearch.search.profile.query.QueryProfileShardResult;
 import org.opensearch.search.profile.query.QueryProfileShardResultTests;
 import org.opensearch.test.OpenSearchTestCase;
@@ -69,10 +70,16 @@ public static SearchProfileShardResults createTestItem() {
                 queryProfileResults.add(QueryProfileShardResultTests.createTestItem());
             }
             AggregationProfileShardResult aggProfileShardResult = AggregationProfileShardResultTests.createTestItem(1);
+            List<ProfileResult> fetchResults = new ArrayList<>();
+            int fetchItems = rarely() ? 0 : randomIntBetween(1, 2);
+            for (int f = 0; f < fetchItems; f++) {
+                fetchResults.add(ProfileResultTests.createTestItem(1, false));
+            }
+            FetchProfileShardResult fetchProfileShardResult = new FetchProfileShardResult(fetchResults);
             NetworkTime networkTime = new NetworkTime(inboundTime, outboundTime);
             searchProfileResults.put(
                 randomAlphaOfLengthBetween(5, 10),
-                new ProfileShardResult(queryProfileResults, aggProfileShardResult, networkTime)
+                new ProfileShardResult(queryProfileResults, aggProfileShardResult, fetchProfileShardResult, networkTime)
             );
         }
         return new SearchProfileShardResults(searchProfileResults);

server/src/test/java/org/opensearch/search/profile/fetch/FetchProfileBreakdownTests.java

@@ -0,0 +1,44 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package org.opensearch.search.profile.fetch;
+
+import org.opensearch.search.profile.AbstractProfileBreakdown;
+import org.opensearch.search.profile.Timer;
+import org.opensearch.test.OpenSearchTestCase;
+
+import java.util.Map;
+
+import static org.hamcrest.Matchers.equalTo;
+import static org.hamcrest.Matchers.greaterThan;
+
+public class FetchProfileBreakdownTests extends OpenSearchTestCase {
+
+    public void testBreakdownMap() {
+        FetchProfileBreakdown breakdown = new FetchProfileBreakdown();
+        for (FetchTimingType type : FetchTimingType.values()) {
+            Timer t = breakdown.getTimer(type);
+            t.start();
+            t.stop();
+        }
+        Map<String, Long> map = breakdown.toBreakdownMap();
+        assertEquals(FetchTimingType.values().length * 3, map.size());
+        for (FetchTimingType type : FetchTimingType.values()) {
+            String key = type.toString();
+            assertThat(map.get(key), greaterThan(0L));
+            assertThat(map.get(key + AbstractProfileBreakdown.TIMING_TYPE_COUNT_SUFFIX), equalTo(1L));
+            assertThat(map.get(key + AbstractProfileBreakdown.TIMING_TYPE_START_TIME_SUFFIX), greaterThan(0L));
+        }
+        long total = breakdown.toNodeTime();
+        long sum = 0;
+        for (FetchTimingType type : FetchTimingType.values()) {
+            sum += map.get(type.toString());
+        }
+        assertEquals(sum, total);
+    }
+}

server/src/test/java/org/opensearch/search/profile/fetch/FetchProfileShardResultTests.java

@@ -0,0 +1,53 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+package org.opensearch.search.profile.fetch;
+
+import org.opensearch.common.xcontent.XContentType;
+import org.opensearch.core.common.bytes.BytesReference;
+import org.opensearch.core.xcontent.ToXContent;
+import org.opensearch.core.xcontent.XContentParser;
+import org.opensearch.core.xcontent.XContentParserUtils;
+import org.opensearch.search.profile.ProfileResult;
+import org.opensearch.search.profile.ProfileResultTests;
+import org.opensearch.test.OpenSearchTestCase;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.opensearch.core.xcontent.XContentHelper.toXContent;
+import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertToXContentEquivalent;
+
+public class FetchProfileShardResultTests extends OpenSearchTestCase {
+
+    public static FetchProfileShardResult createTestItem() {
+        int size = randomIntBetween(0, 5);
+        List<ProfileResult> results = new ArrayList<>(size);
+        for (int i = 0; i < size; i++) {
+            results.add(ProfileResultTests.createTestItem(1, false));
+        }
+        return new FetchProfileShardResult(results);
+    }
+
+    public void testFromXContent() throws IOException {
+        FetchProfileShardResult profileResult = createTestItem();
+        XContentType xContentType = randomFrom(XContentType.values());
+        boolean humanReadable = randomBoolean();
+        BytesReference originalBytes = toShuffledXContent(profileResult, xContentType, ToXContent.EMPTY_PARAMS, humanReadable);
+        FetchProfileShardResult parsed;
+        try (XContentParser parser = createParser(xContentType.xContent(), originalBytes)) {
+            XContentParserUtils.ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.nextToken(), parser);
+            XContentParserUtils.ensureFieldName(parser, parser.nextToken(), FetchProfileShardResult.FETCH);
+            XContentParserUtils.ensureExpectedToken(XContentParser.Token.START_ARRAY, parser.nextToken(), parser);
+            parsed = FetchProfileShardResult.fromXContent(parser);
+            assertEquals(XContentParser.Token.END_OBJECT, parser.nextToken());
+            assertNull(parser.nextToken());
+        }
+        assertToXContentEquivalent(originalBytes, toXContent(parsed, xContentType, humanReadable), xContentType);
+    }
+}

server/src/test/java/org/opensearch/search/profile/fetch/FetchProfilerTests.java

@@ -0,0 +1,85 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+/*
+ * Licensed to Elasticsearch under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Modifications Copyright OpenSearch Contributors. See
+ * GitHub history for details.
+ */
+package org.opensearch.search.profile.fetch;
+
+import org.opensearch.search.profile.AbstractProfileBreakdown;
+import org.opensearch.search.profile.ProfileResult;
+import org.opensearch.search.profile.Timer;
+import org.opensearch.test.OpenSearchTestCase;
+
+import java.util.List;
+import java.util.Map;
+
+import static org.hamcrest.Matchers.equalTo;
+import static org.hamcrest.Matchers.greaterThan;
+
+public class FetchProfilerTests extends OpenSearchTestCase {
+
+    public void testBasicProfiling() {
+        FetchProfiler profiler = new FetchProfiler();
+        FetchProfileBreakdown breakdown = profiler.getQueryBreakdown("fetch");
+        for (FetchTimingType type : FetchTimingType.values()) {
+            Timer t = breakdown.getTimer(type);
+            t.start();
+            t.stop();
+        }
+        profiler.pollLastElement();
+        List<ProfileResult> results = profiler.getTree();
+        assertEquals(1, results.size());
+        ProfileResult profileResult = results.get(0);
+        assertEquals("fetch", profileResult.getQueryName());
+        Map<String, Long> map = profileResult.getTimeBreakdown();
+        assertEquals(FetchTimingType.values().length * 3, map.size());
+        long sum = 0;
+        for (FetchTimingType type : FetchTimingType.values()) {
+            String key = type.toString();
+            assertThat(map.get(key), greaterThan(0L));
+            assertThat(map.get(key + AbstractProfileBreakdown.TIMING_TYPE_COUNT_SUFFIX), equalTo(1L));
+            assertThat(map.get(key + AbstractProfileBreakdown.TIMING_TYPE_START_TIME_SUFFIX), greaterThan(0L));
+            sum += map.get(key);
+        }
+        assertEquals(sum, profileResult.getTime());
+    }
+
+    public void testTimerAggregation() {
+        FetchProfileBreakdown breakdown = new FetchProfileBreakdown();
+        Timer timer = breakdown.getTimer(FetchTimingType.FETCH_FIELDS);
+        timer.start();
+        timer.stop();
+        timer.start();
+        timer.stop();
+        Map<String, Long> map = breakdown.toBreakdownMap();
+        assertThat(map.get(FetchTimingType.FETCH_FIELDS.toString()), greaterThan(0L));
+        assertThat(map.get(FetchTimingType.FETCH_FIELDS + AbstractProfileBreakdown.TIMING_TYPE_COUNT_SUFFIX), equalTo(2L));
+        assertThat(breakdown.toNodeTime(), equalTo(map.get(FetchTimingType.FETCH_FIELDS.toString())));
+    }
+}