blob: b294cad1802c8189c6fe681f1d74d3e4f55054cb [file] [log] [blame]
Yangsterf2bee6f2017-11-29 12:01:05 -08001// Copyright (C) 2017 The Android Open Source Project
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15#include "src/metrics/DurationMetricProducer.h"
Yangster-macb8144812018-01-04 10:56:23 -080016#include "src/stats_log_util.h"
Yangsterf2bee6f2017-11-29 12:01:05 -080017#include "metrics_test_helper.h"
18#include "src/condition/ConditionWizard.h"
19
20#include <gmock/gmock.h>
21#include <gtest/gtest.h>
22#include <stdio.h>
23#include <set>
24#include <unordered_map>
25#include <vector>
26
27using namespace android::os::statsd;
28using namespace testing;
29using android::sp;
30using std::set;
31using std::unordered_map;
32using std::vector;
33
34#ifdef __ANDROID__
35
36namespace android {
37namespace os {
38namespace statsd {
39
Yangster-mac94e197c2018-01-02 16:03:03 -080040const ConfigKey kConfigKey(0, 12345);
Yangsterf2bee6f2017-11-29 12:01:05 -080041
Chenjie Yue1361ed2018-07-23 17:33:09 -070042TEST(DurationMetricTrackerTest, TestFirstBucket) {
43 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
44 DurationMetric metric;
45 metric.set_id(1);
46 metric.set_bucket(ONE_MINUTE);
47 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
48
49 FieldMatcher dimensions;
50 DurationMetricProducer durationProducer(
51 kConfigKey, metric, -1 /*no condition*/, 1 /* start index */, 2 /* stop index */,
52 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, 5, 600 * NS_PER_SEC + NS_PER_SEC/2);
53
54 EXPECT_EQ(600500000000, durationProducer.mCurrentBucketStartTimeNs);
55 EXPECT_EQ(10, durationProducer.mCurrentBucketNum);
56 EXPECT_EQ(660000000005, durationProducer.getCurrentBucketEndTimeNs());
57}
58
Yangsterf2bee6f2017-11-29 12:01:05 -080059TEST(DurationMetricTrackerTest, TestNoCondition) {
60 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
Yangster-macb142cc82018-03-30 15:22:08 -070061 int64_t bucketStartTimeNs = 10000000000;
Yangster-macb8144812018-01-04 10:56:23 -080062 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
Yangsterf2bee6f2017-11-29 12:01:05 -080063
64 DurationMetric metric;
Yangster-mac94e197c2018-01-02 16:03:03 -080065 metric.set_id(1);
Yangster-macb8144812018-01-04 10:56:23 -080066 metric.set_bucket(ONE_MINUTE);
Yangsterf2bee6f2017-11-29 12:01:05 -080067 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
68
69 int tagId = 1;
70 LogEvent event1(tagId, bucketStartTimeNs + 1);
Yangster-mac330af582018-02-08 15:24:38 -080071 event1.init();
Yangsterf2bee6f2017-11-29 12:01:05 -080072 LogEvent event2(tagId, bucketStartTimeNs + bucketSizeNs + 2);
Yangster-mac330af582018-02-08 15:24:38 -080073 event2.init();
Yangsterf2bee6f2017-11-29 12:01:05 -080074
Yangster-mac20877162017-12-22 17:19:39 -080075 FieldMatcher dimensions;
Yangsterf2bee6f2017-11-29 12:01:05 -080076 DurationMetricProducer durationProducer(
77 kConfigKey, metric, -1 /*no condition*/, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -070078 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
Yangsterf2bee6f2017-11-29 12:01:05 -080079
Chenjie Yua7259ab2017-12-10 08:31:05 -080080 durationProducer.onMatchedLogEvent(1 /* start index*/, event1);
81 durationProducer.onMatchedLogEvent(2 /* stop index*/, event2);
Yangsterf2bee6f2017-11-29 12:01:05 -080082 durationProducer.flushIfNeededLocked(bucketStartTimeNs + 2 * bucketSizeNs + 1);
83 EXPECT_EQ(1UL, durationProducer.mPastBuckets.size());
Yangster-mac93694462018-01-22 20:49:31 -080084 EXPECT_TRUE(durationProducer.mPastBuckets.find(DEFAULT_METRIC_DIMENSION_KEY) !=
Yangsterf2bee6f2017-11-29 12:01:05 -080085 durationProducer.mPastBuckets.end());
Yangster-mac93694462018-01-22 20:49:31 -080086 const auto& buckets = durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
Yangsterf2bee6f2017-11-29 12:01:05 -080087 EXPECT_EQ(2UL, buckets.size());
88 EXPECT_EQ(bucketStartTimeNs, buckets[0].mBucketStartNs);
89 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[0].mBucketEndNs);
Yangster-macb142cc82018-03-30 15:22:08 -070090 EXPECT_EQ(bucketSizeNs - 1LL, buckets[0].mDuration);
Yangsterf2bee6f2017-11-29 12:01:05 -080091 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[1].mBucketStartNs);
92 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[1].mBucketEndNs);
Yangster-macb142cc82018-03-30 15:22:08 -070093 EXPECT_EQ(2LL, buckets[1].mDuration);
Yangsterf2bee6f2017-11-29 12:01:05 -080094}
95
96TEST(DurationMetricTrackerTest, TestNonSlicedCondition) {
97 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
Yangster-macb142cc82018-03-30 15:22:08 -070098 int64_t bucketStartTimeNs = 10000000000;
Yangster-macb8144812018-01-04 10:56:23 -080099 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
Yangsterf2bee6f2017-11-29 12:01:05 -0800100
101 DurationMetric metric;
Yangster-mac94e197c2018-01-02 16:03:03 -0800102 metric.set_id(1);
Yangster-macb8144812018-01-04 10:56:23 -0800103 metric.set_bucket(ONE_MINUTE);
Yangsterf2bee6f2017-11-29 12:01:05 -0800104 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
105
106 int tagId = 1;
107 LogEvent event1(tagId, bucketStartTimeNs + 1);
Yangster-mac330af582018-02-08 15:24:38 -0800108 event1.init();
Yangsterf2bee6f2017-11-29 12:01:05 -0800109 LogEvent event2(tagId, bucketStartTimeNs + 2);
Yangster-mac330af582018-02-08 15:24:38 -0800110 event2.init();
Yangsterf2bee6f2017-11-29 12:01:05 -0800111 LogEvent event3(tagId, bucketStartTimeNs + bucketSizeNs + 1);
Yangster-mac330af582018-02-08 15:24:38 -0800112 event3.init();
Yangsterf2bee6f2017-11-29 12:01:05 -0800113 LogEvent event4(tagId, bucketStartTimeNs + bucketSizeNs + 3);
Yangster-mac330af582018-02-08 15:24:38 -0800114 event4.init();
Yangsterf2bee6f2017-11-29 12:01:05 -0800115
Yangster-mac20877162017-12-22 17:19:39 -0800116 FieldMatcher dimensions;
Yangsterf2bee6f2017-11-29 12:01:05 -0800117 DurationMetricProducer durationProducer(
Yao Chenf60e0ba2017-11-29 15:06:41 -0800118 kConfigKey, metric, 0 /* condition index */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700119 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
Olivier Gaillarde63d9e02019-02-12 14:43:59 +0000120 durationProducer.mCondition = ConditionState::kFalse;
yro59cc24d2018-02-13 20:17:32 -0800121
Yangsterf2bee6f2017-11-29 12:01:05 -0800122 EXPECT_FALSE(durationProducer.mCondition);
123 EXPECT_FALSE(durationProducer.isConditionSliced());
124
Chenjie Yua7259ab2017-12-10 08:31:05 -0800125 durationProducer.onMatchedLogEvent(1 /* start index*/, event1);
126 durationProducer.onMatchedLogEvent(2 /* stop index*/, event2);
Yangsterf2bee6f2017-11-29 12:01:05 -0800127 durationProducer.flushIfNeededLocked(bucketStartTimeNs + bucketSizeNs + 1);
Yao Chenf60e0ba2017-11-29 15:06:41 -0800128 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
Yangsterf2bee6f2017-11-29 12:01:05 -0800129
Chenjie Yua7259ab2017-12-10 08:31:05 -0800130 durationProducer.onMatchedLogEvent(1 /* start index*/, event3);
Yangsterf2bee6f2017-11-29 12:01:05 -0800131 durationProducer.onConditionChanged(true /* condition */, bucketStartTimeNs + bucketSizeNs + 2);
Chenjie Yua7259ab2017-12-10 08:31:05 -0800132 durationProducer.onMatchedLogEvent(2 /* stop index*/, event4);
Yangsterf2bee6f2017-11-29 12:01:05 -0800133 durationProducer.flushIfNeededLocked(bucketStartTimeNs + 2 * bucketSizeNs + 1);
134 EXPECT_EQ(1UL, durationProducer.mPastBuckets.size());
Yangster-mac93694462018-01-22 20:49:31 -0800135 EXPECT_TRUE(durationProducer.mPastBuckets.find(DEFAULT_METRIC_DIMENSION_KEY) !=
Yangsterf2bee6f2017-11-29 12:01:05 -0800136 durationProducer.mPastBuckets.end());
Yangster-mac93694462018-01-22 20:49:31 -0800137 const auto& buckets2 = durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
Yangsterf2bee6f2017-11-29 12:01:05 -0800138 EXPECT_EQ(1UL, buckets2.size());
139 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets2[0].mBucketStartNs);
140 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets2[0].mBucketEndNs);
Yangster-macb142cc82018-03-30 15:22:08 -0700141 EXPECT_EQ(1LL, buckets2[0].mDuration);
Yangsterf2bee6f2017-11-29 12:01:05 -0800142}
143
Olivier Gaillarde63d9e02019-02-12 14:43:59 +0000144TEST(DurationMetricTrackerTest, TestNonSlicedConditionUnknownState) {
145 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
146 int64_t bucketStartTimeNs = 10000000000;
147 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
148
149 DurationMetric metric;
150 metric.set_id(1);
151 metric.set_bucket(ONE_MINUTE);
152 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
153
154 int tagId = 1;
155 LogEvent event1(tagId, bucketStartTimeNs + 1);
156 event1.init();
157 LogEvent event2(tagId, bucketStartTimeNs + 2);
158 event2.init();
159 LogEvent event3(tagId, bucketStartTimeNs + bucketSizeNs + 1);
160 event3.init();
161 LogEvent event4(tagId, bucketStartTimeNs + bucketSizeNs + 3);
162 event4.init();
163
164 FieldMatcher dimensions;
165 DurationMetricProducer durationProducer(
166 kConfigKey, metric, 0 /* condition index */, 1 /* start index */, 2 /* stop index */,
167 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
168
169 EXPECT_EQ(ConditionState::kUnknown, durationProducer.mCondition);
170 EXPECT_FALSE(durationProducer.isConditionSliced());
171
172 durationProducer.onMatchedLogEvent(1 /* start index*/, event1);
173 durationProducer.onMatchedLogEvent(2 /* stop index*/, event2);
174 durationProducer.flushIfNeededLocked(bucketStartTimeNs + bucketSizeNs + 1);
175 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
176
177 durationProducer.onMatchedLogEvent(1 /* start index*/, event3);
178 durationProducer.onConditionChanged(true /* condition */, bucketStartTimeNs + bucketSizeNs + 2);
179 durationProducer.onMatchedLogEvent(2 /* stop index*/, event4);
180 durationProducer.flushIfNeededLocked(bucketStartTimeNs + 2 * bucketSizeNs + 1);
181 EXPECT_EQ(1UL, durationProducer.mPastBuckets.size());
182 const auto& buckets2 = durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
183 EXPECT_EQ(1UL, buckets2.size());
184 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets2[0].mBucketStartNs);
185 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets2[0].mBucketEndNs);
186 EXPECT_EQ(1LL, buckets2[0].mDuration);
187}
188
David Chen27785a82018-01-19 17:06:45 -0800189TEST(DurationMetricTrackerTest, TestSumDurationWithUpgrade) {
190 /**
191 * The duration starts from the first bucket, through the two partial buckets (10-70sec),
192 * another bucket, and ends at the beginning of the next full bucket.
193 * Expected buckets:
194 * - [10,25]: 14 secs
195 * - [25,70]: All 45 secs
196 * - [70,130]: All 60 secs
197 * - [130, 210]: Only 5 secs (event ended at 135sec)
198 */
Yangster-macb142cc82018-03-30 15:22:08 -0700199 int64_t bucketStartTimeNs = 10000000000;
200 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
201 int64_t eventUpgradeTimeNs = bucketStartTimeNs + 15 * NS_PER_SEC;
202 int64_t startTimeNs = bucketStartTimeNs + 1 * NS_PER_SEC;
203 int64_t endTimeNs = startTimeNs + 125 * NS_PER_SEC;
David Chen27785a82018-01-19 17:06:45 -0800204
205 int tagId = 1;
206
207 DurationMetric metric;
208 metric.set_id(1);
209 metric.set_bucket(ONE_MINUTE);
210 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
211 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
212 FieldMatcher dimensions;
213 DurationMetricProducer durationProducer(
214 kConfigKey, metric, -1 /* no condition */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700215 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
David Chen27785a82018-01-19 17:06:45 -0800216
Yangster-mac330af582018-02-08 15:24:38 -0800217 LogEvent start_event(tagId, startTimeNs);
218 start_event.init();
219 durationProducer.onMatchedLogEvent(1 /* start index*/, start_event);
David Chen27785a82018-01-19 17:06:45 -0800220 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
221 EXPECT_EQ(bucketStartTimeNs, durationProducer.mCurrentBucketStartTimeNs);
222
223 durationProducer.notifyAppUpgrade(eventUpgradeTimeNs, "ANY.APP", 1, 1);
224 EXPECT_EQ(1UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
225 std::vector<DurationBucket> buckets =
226 durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
227 EXPECT_EQ(bucketStartTimeNs, buckets[0].mBucketStartNs);
228 EXPECT_EQ(eventUpgradeTimeNs, buckets[0].mBucketEndNs);
229 EXPECT_EQ(eventUpgradeTimeNs - startTimeNs, buckets[0].mDuration);
230 EXPECT_EQ(eventUpgradeTimeNs, durationProducer.mCurrentBucketStartTimeNs);
231
232 // We skip ahead one bucket, so we fill in the first two partial buckets and one full bucket.
Yangster-mac330af582018-02-08 15:24:38 -0800233 LogEvent end_event(tagId, endTimeNs);
234 end_event.init();
235 durationProducer.onMatchedLogEvent(2 /* stop index*/, end_event);
David Chen27785a82018-01-19 17:06:45 -0800236 buckets = durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
237 EXPECT_EQ(3UL, buckets.size());
238 EXPECT_EQ(eventUpgradeTimeNs, buckets[1].mBucketStartNs);
239 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[1].mBucketEndNs);
240 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs - eventUpgradeTimeNs, buckets[1].mDuration);
241 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[2].mBucketStartNs);
242 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[2].mBucketEndNs);
243 EXPECT_EQ(bucketSizeNs, buckets[2].mDuration);
244}
245
246TEST(DurationMetricTrackerTest, TestSumDurationWithUpgradeInFollowingBucket) {
247 /**
248 * Expected buckets (start at 11s, upgrade at 75s, end at 135s):
249 * - [10,70]: 59 secs
250 * - [70,75]: 5 sec
251 * - [75,130]: 55 secs
252 */
Yangster-macb142cc82018-03-30 15:22:08 -0700253 int64_t bucketStartTimeNs = 10000000000;
254 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
255 int64_t eventUpgradeTimeNs = bucketStartTimeNs + 65 * NS_PER_SEC;
256 int64_t startTimeNs = bucketStartTimeNs + 1 * NS_PER_SEC;
257 int64_t endTimeNs = startTimeNs + 125 * NS_PER_SEC;
David Chen27785a82018-01-19 17:06:45 -0800258
259 int tagId = 1;
260
261 DurationMetric metric;
262 metric.set_id(1);
263 metric.set_bucket(ONE_MINUTE);
264 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
265 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
266 FieldMatcher dimensions;
267 DurationMetricProducer durationProducer(
268 kConfigKey, metric, -1 /* no condition */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700269 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
David Chen27785a82018-01-19 17:06:45 -0800270
Yangster-mac330af582018-02-08 15:24:38 -0800271 LogEvent start_event(tagId, startTimeNs);
272 start_event.init();
273 durationProducer.onMatchedLogEvent(1 /* start index*/, start_event);
David Chen27785a82018-01-19 17:06:45 -0800274 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
275 EXPECT_EQ(bucketStartTimeNs, durationProducer.mCurrentBucketStartTimeNs);
276
277 durationProducer.notifyAppUpgrade(eventUpgradeTimeNs, "ANY.APP", 1, 1);
278 EXPECT_EQ(2UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
279 std::vector<DurationBucket> buckets =
280 durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
281 EXPECT_EQ(bucketStartTimeNs, buckets[0].mBucketStartNs);
282 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[0].mBucketEndNs);
283 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs - startTimeNs, buckets[0].mDuration);
284 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[1].mBucketStartNs);
285 EXPECT_EQ(eventUpgradeTimeNs, buckets[1].mBucketEndNs);
286 EXPECT_EQ(eventUpgradeTimeNs - (bucketStartTimeNs + bucketSizeNs), buckets[1].mDuration);
287 EXPECT_EQ(eventUpgradeTimeNs, durationProducer.mCurrentBucketStartTimeNs);
288
289 // We skip ahead one bucket, so we fill in the first two partial buckets and one full bucket.
Yangster-mac330af582018-02-08 15:24:38 -0800290 LogEvent end_event(tagId, endTimeNs);
291 end_event.init();
292 durationProducer.onMatchedLogEvent(2 /* stop index*/, end_event);
David Chen27785a82018-01-19 17:06:45 -0800293 buckets = durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
294 EXPECT_EQ(3UL, buckets.size());
295 EXPECT_EQ(eventUpgradeTimeNs, buckets[2].mBucketStartNs);
296 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[2].mBucketEndNs);
297 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs - eventUpgradeTimeNs, buckets[2].mDuration);
298}
299
300TEST(DurationMetricTrackerTest, TestSumDurationAnomalyWithUpgrade) {
Yangster-mac932ecec2018-02-01 10:23:52 -0800301 sp<AlarmMonitor> alarmMonitor;
Yangster-macb142cc82018-03-30 15:22:08 -0700302 int64_t bucketStartTimeNs = 10000000000;
303 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
304 int64_t eventUpgradeTimeNs = bucketStartTimeNs + 15 * NS_PER_SEC;
305 int64_t startTimeNs = bucketStartTimeNs + 1;
306 int64_t endTimeNs = startTimeNs + 65 * NS_PER_SEC;
David Chen27785a82018-01-19 17:06:45 -0800307
308 int tagId = 1;
309
310 // Setup metric with alert.
311 DurationMetric metric;
312 metric.set_id(1);
313 metric.set_bucket(ONE_MINUTE);
314 metric.set_aggregation_type(DurationMetric_AggregationType_SUM);
315 Alert alert;
316 alert.set_num_buckets(3);
317 alert.set_trigger_if_sum_gt(2);
318
319 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
320 FieldMatcher dimensions;
321 DurationMetricProducer durationProducer(
322 kConfigKey, metric, -1 /* no condition */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700323 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
yro59cc24d2018-02-13 20:17:32 -0800324
Yangster-mac932ecec2018-02-01 10:23:52 -0800325 sp<AnomalyTracker> anomalyTracker = durationProducer.addAnomalyTracker(alert, alarmMonitor);
David Chen27785a82018-01-19 17:06:45 -0800326 EXPECT_TRUE(anomalyTracker != nullptr);
327
Yangster-mac330af582018-02-08 15:24:38 -0800328 LogEvent start_event(tagId, startTimeNs);
329 start_event.init();
330 durationProducer.onMatchedLogEvent(1 /* start index*/, start_event);
David Chen27785a82018-01-19 17:06:45 -0800331 durationProducer.notifyAppUpgrade(eventUpgradeTimeNs, "ANY.APP", 1, 1);
332 // We skip ahead one bucket, so we fill in the first two partial buckets and one full bucket.
Yangster-mac330af582018-02-08 15:24:38 -0800333 LogEvent end_event(tagId, endTimeNs);
334 end_event.init();
335 durationProducer.onMatchedLogEvent(2 /* stop index*/, end_event);
David Chen27785a82018-01-19 17:06:45 -0800336
337 EXPECT_EQ(bucketStartTimeNs + bucketSizeNs - startTimeNs,
Yangster-macb142cc82018-03-30 15:22:08 -0700338 anomalyTracker->getSumOverPastBuckets(DEFAULT_METRIC_DIMENSION_KEY));
David Chen27785a82018-01-19 17:06:45 -0800339}
340
341TEST(DurationMetricTrackerTest, TestMaxDurationWithUpgrade) {
Yangster-macb142cc82018-03-30 15:22:08 -0700342 int64_t bucketStartTimeNs = 10000000000;
343 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
344 int64_t eventUpgradeTimeNs = bucketStartTimeNs + 15 * NS_PER_SEC;
345 int64_t startTimeNs = bucketStartTimeNs + 1;
346 int64_t endTimeNs = startTimeNs + 125 * NS_PER_SEC;
David Chen27785a82018-01-19 17:06:45 -0800347
348 int tagId = 1;
349
350 DurationMetric metric;
351 metric.set_id(1);
352 metric.set_bucket(ONE_MINUTE);
353 metric.set_aggregation_type(DurationMetric_AggregationType_MAX_SPARSE);
354 LogEvent event1(tagId, startTimeNs);
355 event1.write("111"); // uid
356 event1.init();
357 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
358 FieldMatcher dimensions;
359 DurationMetricProducer durationProducer(
360 kConfigKey, metric, -1 /* no condition */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700361 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
David Chen27785a82018-01-19 17:06:45 -0800362
Yangster-mac330af582018-02-08 15:24:38 -0800363 LogEvent start_event(tagId, startTimeNs);
364 start_event.init();
365 durationProducer.onMatchedLogEvent(1 /* start index*/, start_event);
David Chen27785a82018-01-19 17:06:45 -0800366 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
367 EXPECT_EQ(bucketStartTimeNs, durationProducer.mCurrentBucketStartTimeNs);
368
369 durationProducer.notifyAppUpgrade(eventUpgradeTimeNs, "ANY.APP", 1, 1);
370 EXPECT_EQ(0UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
371 EXPECT_EQ(eventUpgradeTimeNs, durationProducer.mCurrentBucketStartTimeNs);
372
373 // We skip ahead one bucket, so we fill in the first two partial buckets and one full bucket.
Yangster-mac330af582018-02-08 15:24:38 -0800374 LogEvent end_event(tagId, endTimeNs);
375 end_event.init();
376 durationProducer.onMatchedLogEvent(2 /* stop index*/, end_event);
David Chen27785a82018-01-19 17:06:45 -0800377 EXPECT_EQ(0UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
378
379 durationProducer.flushIfNeededLocked(bucketStartTimeNs + 3 * bucketSizeNs + 1);
380 std::vector<DurationBucket> buckets =
381 durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
382 EXPECT_EQ(1UL, buckets.size());
383 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[0].mBucketStartNs);
384 EXPECT_EQ(bucketStartTimeNs + 3 * bucketSizeNs, buckets[0].mBucketEndNs);
385 EXPECT_EQ(endTimeNs - startTimeNs, buckets[0].mDuration);
386}
387
388TEST(DurationMetricTrackerTest, TestMaxDurationWithUpgradeInNextBucket) {
Yangster-macb142cc82018-03-30 15:22:08 -0700389 int64_t bucketStartTimeNs = 10000000000;
390 int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(ONE_MINUTE) * 1000000LL;
391 int64_t eventUpgradeTimeNs = bucketStartTimeNs + 65 * NS_PER_SEC;
392 int64_t startTimeNs = bucketStartTimeNs + 1;
393 int64_t endTimeNs = startTimeNs + 115 * NS_PER_SEC;
David Chen27785a82018-01-19 17:06:45 -0800394
395 int tagId = 1;
396
397 DurationMetric metric;
398 metric.set_id(1);
399 metric.set_bucket(ONE_MINUTE);
400 metric.set_aggregation_type(DurationMetric_AggregationType_MAX_SPARSE);
401 LogEvent event1(tagId, startTimeNs);
402 event1.write("111"); // uid
403 event1.init();
404 sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
405 FieldMatcher dimensions;
406 DurationMetricProducer durationProducer(
407 kConfigKey, metric, -1 /* no condition */, 1 /* start index */, 2 /* stop index */,
Chenjie Yue1361ed2018-07-23 17:33:09 -0700408 3 /* stop_all index */, false /*nesting*/, wizard, dimensions, bucketStartTimeNs, bucketStartTimeNs);
David Chen27785a82018-01-19 17:06:45 -0800409
Yangster-mac330af582018-02-08 15:24:38 -0800410 LogEvent start_event(tagId, startTimeNs);
411 start_event.init();
412 durationProducer.onMatchedLogEvent(1 /* start index*/, start_event);
David Chen27785a82018-01-19 17:06:45 -0800413 EXPECT_EQ(0UL, durationProducer.mPastBuckets.size());
414 EXPECT_EQ(bucketStartTimeNs, durationProducer.mCurrentBucketStartTimeNs);
415
416 durationProducer.notifyAppUpgrade(eventUpgradeTimeNs, "ANY.APP", 1, 1);
417 EXPECT_EQ(0UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
418 EXPECT_EQ(eventUpgradeTimeNs, durationProducer.mCurrentBucketStartTimeNs);
419
420 // Stop occurs in the same partial bucket as created for the app upgrade.
Yangster-mac330af582018-02-08 15:24:38 -0800421 LogEvent end_event(tagId, endTimeNs);
422 end_event.init();
423 durationProducer.onMatchedLogEvent(2 /* stop index*/, end_event);
David Chen27785a82018-01-19 17:06:45 -0800424 EXPECT_EQ(0UL, durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY].size());
425 EXPECT_EQ(eventUpgradeTimeNs, durationProducer.mCurrentBucketStartTimeNs);
426
427 durationProducer.flushIfNeededLocked(bucketStartTimeNs + 2 * bucketSizeNs + 1);
428 std::vector<DurationBucket> buckets =
429 durationProducer.mPastBuckets[DEFAULT_METRIC_DIMENSION_KEY];
430 EXPECT_EQ(1UL, buckets.size());
431 EXPECT_EQ(eventUpgradeTimeNs, buckets[0].mBucketStartNs);
432 EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[0].mBucketEndNs);
433 EXPECT_EQ(endTimeNs - startTimeNs, buckets[0].mDuration);
434}
435
Yangsterf2bee6f2017-11-29 12:01:05 -0800436} // namespace statsd
437} // namespace os
438} // namespace android
439#else
440GTEST_LOG_(INFO) << "This test does nothing.\n";
441#endif