commit fbe3035f4c92a6c1f904dfa65b38035111542c65
author Treehugger Robot <treehugger-gerrit@google.com> Wed Dec 15 04:05:35 2021 +0000
committer Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Wed Dec 15 04:05:35 2021 +0000
tree 460bbe454a4c14e2d70b544fcdc690d69df778da
parent 457a10f05ef65edc7ceb5575d72226424b1e0189
parent 291820727f6ce6e5f530b32a6bbd6967831f29d0

Merge "Backfill atom tests for new event metric format" am: 9706e6072f am: 291820727f

Original change: https://android-review.googlesource.com/c/platform/external/libtextclassifier/+/1922001

Change-Id: If86ad1c9cefcd16a97f4e0109d15024859f95bd5

java/tests/instrumentation/src/com/android/textclassifier/common/statsd/StatsdTestUtils.java

diff --git a/java/tests/instrumentation/src/com/android/textclassifier/common/statsd/StatsdTestUtils.java b/java/tests/instrumentation/src/com/android/textclassifier/common/statsd/StatsdTestUtils.java
index 1bcd7b7..ffd2ee4 100644
--- a/java/tests/instrumentation/src/com/android/textclassifier/common/statsd/StatsdTestUtils.java
+++ b/java/tests/instrumentation/src/com/android/textclassifier/common/statsd/StatsdTestUtils.java
@@ -28,6 +28,7 @@
 import com.android.internal.os.StatsdConfigProto.SimpleAtomMatcher;
 import com.android.internal.os.StatsdConfigProto.StatsdConfig;
 import com.android.os.AtomsProto.Atom;
+import com.android.os.StatsLog.AggregatedAtomInfo;
 import com.android.os.StatsLog.ConfigMetricsReport;
 import com.android.os.StatsLog.ConfigMetricsReportList;
 import com.android.os.StatsLog.EventMetricData;
@@ -85,6 +86,8 @@
     return ImmutableList.copyOf(
         metricsList.stream()
             .flatMap(statsLogReport -> statsLogReport.getEventMetrics().getDataList().stream())
+            .flatMap(eventMetricData -> backfillAggregatedAtomsinEventMetric(
+                    eventMetricData).stream())
             .sorted(Comparator.comparing(EventMetricData::getElapsedTimestampNanos))
             .map(EventMetricData::getAtom)
             .collect(Collectors.toList()));
@@ -131,4 +134,20 @@
             /*input=*/ null);
     return ConfigMetricsReportList.parser().parseFrom(new ByteArrayInputStream(output));
   }
+
+  private static ImmutableList<EventMetricData> backfillAggregatedAtomsinEventMetric(
+    EventMetricData metricData) {
+    if (metricData.hasAtom()) {
+      return ImmutableList.of(metricData);
+    }
+    ImmutableList.Builder<EventMetricData> data = ImmutableList.builder();
+    AggregatedAtomInfo atomInfo = metricData.getAggregatedAtomInfo();
+    for (long timestamp : atomInfo.getElapsedTimestampNanosList()) {
+      EventMetricData.Builder newMetricData = EventMetricData.newBuilder();
+      newMetricData.setAtom(atomInfo.getAtom());
+      newMetricData.setElapsedTimestampNanos(timestamp);
+      data.add(newMetricData.build());
+    }
+    return data.build();
+  }
 }