Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 1 | // 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 | |
Ruchir Rastogi | e449b0c | 2020-02-10 17:40:09 -0800 | [diff] [blame] | 15 | #include <android/binder_interface_utils.h> |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 16 | #include <gtest/gtest.h> |
| 17 | |
| 18 | #include "src/StatsLogProcessor.h" |
| 19 | #include "src/stats_log_util.h" |
| 20 | #include "tests/statsd_test_util.h" |
| 21 | |
| 22 | #include <vector> |
| 23 | |
Ruchir Rastogi | e449b0c | 2020-02-10 17:40:09 -0800 | [diff] [blame] | 24 | using ::ndk::SharedRefBase; |
| 25 | |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 26 | namespace android { |
| 27 | namespace os { |
| 28 | namespace statsd { |
| 29 | |
| 30 | #ifdef __ANDROID__ |
| 31 | |
| 32 | namespace { |
| 33 | |
Tej Singh | 597c716 | 2019-04-17 16:41:45 -0700 | [diff] [blame] | 34 | const int64_t metricId = 123456; |
| 35 | |
| 36 | StatsdConfig CreateStatsdConfig(bool useCondition = true) { |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 37 | StatsdConfig config; |
| 38 | config.add_allowed_log_source("AID_ROOT"); // LogEvent defaults to UID of root. |
Tej Singh | 3be093b | 2020-03-04 20:08:38 -0800 | [diff] [blame] | 39 | config.add_default_pull_packages("AID_ROOT"); // Fake puller is registered with root. |
Tej Singh | d7b5118 | 2019-01-29 17:02:12 -0800 | [diff] [blame] | 40 | auto pulledAtomMatcher = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 41 | CreateSimpleAtomMatcher("TestMatcher", util::SUBSYSTEM_SLEEP_STATE); |
Tej Singh | d7b5118 | 2019-01-29 17:02:12 -0800 | [diff] [blame] | 42 | *config.add_atom_matcher() = pulledAtomMatcher; |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 43 | *config.add_atom_matcher() = CreateScreenTurnedOnAtomMatcher(); |
| 44 | *config.add_atom_matcher() = CreateScreenTurnedOffAtomMatcher(); |
| 45 | |
| 46 | auto screenIsOffPredicate = CreateScreenIsOffPredicate(); |
| 47 | *config.add_predicate() = screenIsOffPredicate; |
| 48 | |
| 49 | auto valueMetric = config.add_value_metric(); |
Tej Singh | 597c716 | 2019-04-17 16:41:45 -0700 | [diff] [blame] | 50 | valueMetric->set_id(metricId); |
Tej Singh | d7b5118 | 2019-01-29 17:02:12 -0800 | [diff] [blame] | 51 | valueMetric->set_what(pulledAtomMatcher.id()); |
Tej Singh | 597c716 | 2019-04-17 16:41:45 -0700 | [diff] [blame] | 52 | if (useCondition) { |
| 53 | valueMetric->set_condition(screenIsOffPredicate.id()); |
| 54 | } |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 55 | *valueMetric->mutable_value_field() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 56 | CreateDimensions(util::SUBSYSTEM_SLEEP_STATE, {4 /* time sleeping field */}); |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 57 | *valueMetric->mutable_dimensions_in_what() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 58 | CreateDimensions(util::SUBSYSTEM_SLEEP_STATE, {1 /* subsystem name */}); |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 59 | valueMetric->set_bucket(FIVE_MINUTES); |
Chenjie Yu | 021e253 | 2018-05-16 12:23:07 -0700 | [diff] [blame] | 60 | valueMetric->set_use_absolute_value_on_reset(true); |
Chenjie Yu | c715b9e | 2018-10-19 07:52:12 -0700 | [diff] [blame] | 61 | valueMetric->set_skip_zero_diff_output(false); |
Chenjie Yu | 0bd73db | 2018-12-16 07:37:04 -0800 | [diff] [blame] | 62 | valueMetric->set_max_pull_delay_sec(INT_MAX); |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 63 | return config; |
| 64 | } |
| 65 | |
| 66 | } // namespace |
| 67 | |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 68 | TEST(ValueMetricE2eTest, TestPulledEvents) { |
| 69 | auto config = CreateStatsdConfig(); |
| 70 | int64_t baseTimeNs = getElapsedRealtimeNs(); |
| 71 | int64_t configAddedTimeNs = 10 * 60 * NS_PER_SEC + baseTimeNs; |
| 72 | int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(config.value_metric(0).bucket()) * 1000000; |
| 73 | |
| 74 | ConfigKey cfgKey; |
| 75 | auto processor = CreateStatsLogProcessor(baseTimeNs, configAddedTimeNs, config, cfgKey, |
| 76 | SharedRefBase::make<FakeSubsystemSleepCallback>(), |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 77 | util::SUBSYSTEM_SLEEP_STATE); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 78 | EXPECT_EQ(processor->mMetricsManagers.size(), 1u); |
| 79 | EXPECT_TRUE(processor->mMetricsManagers.begin()->second->isConfigValid()); |
| 80 | processor->mPullerManager->ForceClearPullerCache(); |
| 81 | |
| 82 | int startBucketNum = processor->mMetricsManagers.begin() |
| 83 | ->second->mAllMetricProducers[0] |
| 84 | ->getCurrentBucketNum(); |
| 85 | EXPECT_GT(startBucketNum, (int64_t)0); |
| 86 | |
| 87 | // When creating the config, the value metric producer should register the alarm at the |
| 88 | // end of the current bucket. |
| 89 | EXPECT_EQ((size_t)1, processor->mPullerManager->mReceivers.size()); |
| 90 | EXPECT_EQ(bucketSizeNs, |
| 91 | processor->mPullerManager->mReceivers.begin()->second.front().intervalNs); |
| 92 | int64_t& expectedPullTimeNs = |
| 93 | processor->mPullerManager->mReceivers.begin()->second.front().nextPullTimeNs; |
| 94 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + bucketSizeNs, expectedPullTimeNs); |
| 95 | |
| 96 | auto screenOffEvent = |
| 97 | CreateScreenStateChangedEvent(configAddedTimeNs + 55, android::view::DISPLAY_STATE_OFF); |
| 98 | processor->OnLogEvent(screenOffEvent.get()); |
| 99 | |
| 100 | auto screenOnEvent = |
| 101 | CreateScreenStateChangedEvent(configAddedTimeNs + 65, android::view::DISPLAY_STATE_ON); |
| 102 | processor->OnLogEvent(screenOnEvent.get()); |
| 103 | |
| 104 | screenOffEvent = |
| 105 | CreateScreenStateChangedEvent(configAddedTimeNs + 75, android::view::DISPLAY_STATE_OFF); |
| 106 | processor->OnLogEvent(screenOffEvent.get()); |
| 107 | |
| 108 | // Pulling alarm arrives on time and reset the sequential pulling alarm. |
| 109 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 110 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 2 * bucketSizeNs, expectedPullTimeNs); |
| 111 | |
| 112 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 113 | |
| 114 | screenOnEvent = CreateScreenStateChangedEvent(configAddedTimeNs + 2 * bucketSizeNs + 15, |
| 115 | android::view::DISPLAY_STATE_ON); |
| 116 | processor->OnLogEvent(screenOnEvent.get()); |
| 117 | |
| 118 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 119 | |
| 120 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 121 | |
| 122 | screenOffEvent = CreateScreenStateChangedEvent(configAddedTimeNs + 4 * bucketSizeNs + 11, |
| 123 | android::view::DISPLAY_STATE_OFF); |
| 124 | processor->OnLogEvent(screenOffEvent.get()); |
| 125 | |
| 126 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 127 | |
| 128 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 129 | |
| 130 | ConfigMetricsReportList reports; |
| 131 | vector<uint8_t> buffer; |
| 132 | processor->onDumpReport(cfgKey, configAddedTimeNs + 7 * bucketSizeNs + 10, false, true, |
| 133 | ADB_DUMP, FAST, &buffer); |
| 134 | EXPECT_TRUE(buffer.size() > 0); |
| 135 | EXPECT_TRUE(reports.ParseFromArray(&buffer[0], buffer.size())); |
| 136 | backfillDimensionPath(&reports); |
| 137 | backfillStringInReport(&reports); |
| 138 | backfillStartEndTimestamp(&reports); |
| 139 | EXPECT_EQ(1, reports.reports_size()); |
| 140 | EXPECT_EQ(1, reports.reports(0).metrics_size()); |
| 141 | StatsLogReport::ValueMetricDataWrapper valueMetrics; |
| 142 | sortMetricDataByDimensionsValue(reports.reports(0).metrics(0).value_metrics(), &valueMetrics); |
| 143 | EXPECT_GT((int)valueMetrics.data_size(), 1); |
| 144 | |
| 145 | auto data = valueMetrics.data(0); |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 146 | EXPECT_EQ(util::SUBSYSTEM_SLEEP_STATE, data.dimensions_in_what().field()); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 147 | EXPECT_EQ(1, data.dimensions_in_what().value_tuple().dimensions_value_size()); |
| 148 | EXPECT_EQ(1 /* subsystem name field */, |
| 149 | data.dimensions_in_what().value_tuple().dimensions_value(0).field()); |
| 150 | EXPECT_FALSE(data.dimensions_in_what().value_tuple().dimensions_value(0).value_str().empty()); |
| 151 | // We have 4 buckets, the first one was incomplete since the condition was unknown. |
| 152 | EXPECT_EQ(4, data.bucket_info_size()); |
| 153 | |
| 154 | EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, data.bucket_info(0).start_bucket_elapsed_nanos()); |
| 155 | EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(0).end_bucket_elapsed_nanos()); |
| 156 | EXPECT_EQ(1, data.bucket_info(0).values_size()); |
| 157 | |
| 158 | EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, data.bucket_info(1).start_bucket_elapsed_nanos()); |
| 159 | EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(1).end_bucket_elapsed_nanos()); |
| 160 | EXPECT_EQ(1, data.bucket_info(1).values_size()); |
| 161 | |
| 162 | EXPECT_EQ(baseTimeNs + 6 * bucketSizeNs, data.bucket_info(2).start_bucket_elapsed_nanos()); |
| 163 | EXPECT_EQ(baseTimeNs + 7 * bucketSizeNs, data.bucket_info(2).end_bucket_elapsed_nanos()); |
| 164 | EXPECT_EQ(1, data.bucket_info(2).values_size()); |
| 165 | |
| 166 | EXPECT_EQ(baseTimeNs + 7 * bucketSizeNs, data.bucket_info(3).start_bucket_elapsed_nanos()); |
| 167 | EXPECT_EQ(baseTimeNs + 8 * bucketSizeNs, data.bucket_info(3).end_bucket_elapsed_nanos()); |
| 168 | EXPECT_EQ(1, data.bucket_info(3).values_size()); |
| 169 | } |
| 170 | |
| 171 | TEST(ValueMetricE2eTest, TestPulledEvents_LateAlarm) { |
| 172 | auto config = CreateStatsdConfig(); |
| 173 | int64_t baseTimeNs = getElapsedRealtimeNs(); |
| 174 | // 10 mins == 2 bucket durations. |
| 175 | int64_t configAddedTimeNs = 10 * 60 * NS_PER_SEC + baseTimeNs; |
| 176 | int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(config.value_metric(0).bucket()) * 1000000; |
| 177 | |
| 178 | ConfigKey cfgKey; |
| 179 | auto processor = CreateStatsLogProcessor(baseTimeNs, configAddedTimeNs, config, cfgKey, |
| 180 | SharedRefBase::make<FakeSubsystemSleepCallback>(), |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 181 | util::SUBSYSTEM_SLEEP_STATE); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 182 | EXPECT_EQ(processor->mMetricsManagers.size(), 1u); |
| 183 | EXPECT_TRUE(processor->mMetricsManagers.begin()->second->isConfigValid()); |
| 184 | processor->mPullerManager->ForceClearPullerCache(); |
| 185 | |
| 186 | int startBucketNum = processor->mMetricsManagers.begin() |
| 187 | ->second->mAllMetricProducers[0] |
| 188 | ->getCurrentBucketNum(); |
| 189 | EXPECT_GT(startBucketNum, (int64_t)0); |
| 190 | |
| 191 | // When creating the config, the value metric producer should register the alarm at the |
| 192 | // end of the current bucket. |
| 193 | EXPECT_EQ((size_t)1, processor->mPullerManager->mReceivers.size()); |
| 194 | EXPECT_EQ(bucketSizeNs, |
| 195 | processor->mPullerManager->mReceivers.begin()->second.front().intervalNs); |
| 196 | int64_t& expectedPullTimeNs = |
| 197 | processor->mPullerManager->mReceivers.begin()->second.front().nextPullTimeNs; |
| 198 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + bucketSizeNs, expectedPullTimeNs); |
| 199 | |
| 200 | // Screen off/on/off events. |
| 201 | auto screenOffEvent = |
| 202 | CreateScreenStateChangedEvent(configAddedTimeNs + 55, android::view::DISPLAY_STATE_OFF); |
| 203 | processor->OnLogEvent(screenOffEvent.get()); |
| 204 | |
| 205 | auto screenOnEvent = |
| 206 | CreateScreenStateChangedEvent(configAddedTimeNs + 65, android::view::DISPLAY_STATE_ON); |
| 207 | processor->OnLogEvent(screenOnEvent.get()); |
| 208 | |
| 209 | screenOffEvent = |
| 210 | CreateScreenStateChangedEvent(configAddedTimeNs + 75, android::view::DISPLAY_STATE_OFF); |
| 211 | processor->OnLogEvent(screenOffEvent.get()); |
| 212 | |
| 213 | // Pulling alarm arrives late by 2 buckets and 1 ns. 2 buckets late is too far away in the |
| 214 | // future, data will be skipped. |
| 215 | processor->informPullAlarmFired(expectedPullTimeNs + 2 * bucketSizeNs + 1); |
| 216 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 4 * bucketSizeNs, expectedPullTimeNs); |
| 217 | |
| 218 | // This screen state change will start a new bucket. |
| 219 | screenOnEvent = CreateScreenStateChangedEvent(configAddedTimeNs + 4 * bucketSizeNs + 65, |
| 220 | android::view::DISPLAY_STATE_ON); |
| 221 | processor->OnLogEvent(screenOnEvent.get()); |
| 222 | |
| 223 | // The alarm is delayed but we already created a bucket thanks to the screen state condition. |
| 224 | // This bucket does not have to be skipped since the alarm arrives in time for the next bucket. |
| 225 | processor->informPullAlarmFired(expectedPullTimeNs + bucketSizeNs + 21); |
| 226 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 6 * bucketSizeNs, expectedPullTimeNs); |
| 227 | |
| 228 | screenOffEvent = CreateScreenStateChangedEvent(configAddedTimeNs + 6 * bucketSizeNs + 31, |
| 229 | android::view::DISPLAY_STATE_OFF); |
| 230 | processor->OnLogEvent(screenOffEvent.get()); |
| 231 | |
| 232 | processor->informPullAlarmFired(expectedPullTimeNs + bucketSizeNs + 21); |
| 233 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 8 * bucketSizeNs, expectedPullTimeNs); |
| 234 | |
| 235 | processor->informPullAlarmFired(expectedPullTimeNs + 1); |
| 236 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 9 * bucketSizeNs, expectedPullTimeNs); |
| 237 | |
| 238 | ConfigMetricsReportList reports; |
| 239 | vector<uint8_t> buffer; |
| 240 | processor->onDumpReport(cfgKey, configAddedTimeNs + 9 * bucketSizeNs + 10, false, true, |
| 241 | ADB_DUMP, FAST, &buffer); |
| 242 | EXPECT_TRUE(buffer.size() > 0); |
| 243 | EXPECT_TRUE(reports.ParseFromArray(&buffer[0], buffer.size())); |
| 244 | backfillDimensionPath(&reports); |
| 245 | backfillStringInReport(&reports); |
| 246 | backfillStartEndTimestamp(&reports); |
| 247 | EXPECT_EQ(1, reports.reports_size()); |
| 248 | EXPECT_EQ(1, reports.reports(0).metrics_size()); |
| 249 | StatsLogReport::ValueMetricDataWrapper valueMetrics; |
| 250 | sortMetricDataByDimensionsValue(reports.reports(0).metrics(0).value_metrics(), &valueMetrics); |
| 251 | EXPECT_GT((int)valueMetrics.data_size(), 1); |
| 252 | |
| 253 | auto data = valueMetrics.data(0); |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 254 | EXPECT_EQ(util::SUBSYSTEM_SLEEP_STATE, data.dimensions_in_what().field()); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 255 | EXPECT_EQ(1, data.dimensions_in_what().value_tuple().dimensions_value_size()); |
| 256 | EXPECT_EQ(1 /* subsystem name field */, |
| 257 | data.dimensions_in_what().value_tuple().dimensions_value(0).field()); |
| 258 | EXPECT_FALSE(data.dimensions_in_what().value_tuple().dimensions_value(0).value_str().empty()); |
| 259 | EXPECT_EQ(3, data.bucket_info_size()); |
| 260 | |
| 261 | EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, data.bucket_info(0).start_bucket_elapsed_nanos()); |
| 262 | EXPECT_EQ(baseTimeNs + 6 * bucketSizeNs, data.bucket_info(0).end_bucket_elapsed_nanos()); |
| 263 | EXPECT_EQ(1, data.bucket_info(0).values_size()); |
| 264 | |
| 265 | EXPECT_EQ(baseTimeNs + 8 * bucketSizeNs, data.bucket_info(1).start_bucket_elapsed_nanos()); |
| 266 | EXPECT_EQ(baseTimeNs + 9 * bucketSizeNs, data.bucket_info(1).end_bucket_elapsed_nanos()); |
| 267 | EXPECT_EQ(1, data.bucket_info(1).values_size()); |
| 268 | |
| 269 | EXPECT_EQ(baseTimeNs + 9 * bucketSizeNs, data.bucket_info(2).start_bucket_elapsed_nanos()); |
| 270 | EXPECT_EQ(baseTimeNs + 10 * bucketSizeNs, data.bucket_info(2).end_bucket_elapsed_nanos()); |
| 271 | EXPECT_EQ(1, data.bucket_info(2).values_size()); |
| 272 | } |
| 273 | |
| 274 | TEST(ValueMetricE2eTest, TestPulledEvents_WithActivation) { |
| 275 | auto config = CreateStatsdConfig(false); |
| 276 | int64_t baseTimeNs = getElapsedRealtimeNs(); |
| 277 | int64_t configAddedTimeNs = 10 * 60 * NS_PER_SEC + baseTimeNs; |
| 278 | int64_t bucketSizeNs = TimeUnitToBucketSizeInMillis(config.value_metric(0).bucket()) * 1000000; |
| 279 | |
| 280 | auto batterySaverStartMatcher = CreateBatterySaverModeStartAtomMatcher(); |
| 281 | *config.add_atom_matcher() = batterySaverStartMatcher; |
| 282 | const int64_t ttlNs = 2 * bucketSizeNs; // Two buckets. |
| 283 | auto metric_activation = config.add_metric_activation(); |
| 284 | metric_activation->set_metric_id(metricId); |
| 285 | metric_activation->set_activation_type(ACTIVATE_IMMEDIATELY); |
| 286 | auto event_activation = metric_activation->add_event_activation(); |
| 287 | event_activation->set_atom_matcher_id(batterySaverStartMatcher.id()); |
| 288 | event_activation->set_ttl_seconds(ttlNs / 1000000000); |
| 289 | |
| 290 | ConfigKey cfgKey; |
| 291 | auto processor = CreateStatsLogProcessor(baseTimeNs, configAddedTimeNs, config, cfgKey, |
| 292 | SharedRefBase::make<FakeSubsystemSleepCallback>(), |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 293 | util::SUBSYSTEM_SLEEP_STATE); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 294 | EXPECT_EQ(processor->mMetricsManagers.size(), 1u); |
| 295 | EXPECT_TRUE(processor->mMetricsManagers.begin()->second->isConfigValid()); |
| 296 | processor->mPullerManager->ForceClearPullerCache(); |
| 297 | |
| 298 | int startBucketNum = processor->mMetricsManagers.begin() |
| 299 | ->second->mAllMetricProducers[0] |
| 300 | ->getCurrentBucketNum(); |
| 301 | EXPECT_GT(startBucketNum, (int64_t)0); |
| 302 | EXPECT_FALSE(processor->mMetricsManagers.begin()->second->mAllMetricProducers[0]->isActive()); |
| 303 | |
| 304 | // When creating the config, the value metric producer should register the alarm at the |
| 305 | // end of the current bucket. |
| 306 | EXPECT_EQ((size_t)1, processor->mPullerManager->mReceivers.size()); |
| 307 | EXPECT_EQ(bucketSizeNs, |
| 308 | processor->mPullerManager->mReceivers.begin()->second.front().intervalNs); |
| 309 | int64_t& expectedPullTimeNs = |
| 310 | processor->mPullerManager->mReceivers.begin()->second.front().nextPullTimeNs; |
| 311 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + bucketSizeNs, expectedPullTimeNs); |
| 312 | |
| 313 | // Pulling alarm arrives on time and reset the sequential pulling alarm. |
| 314 | processor->informPullAlarmFired(expectedPullTimeNs + 1); // 15 mins + 1 ns. |
| 315 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 2 * bucketSizeNs, expectedPullTimeNs); |
| 316 | EXPECT_FALSE(processor->mMetricsManagers.begin()->second->mAllMetricProducers[0]->isActive()); |
| 317 | |
| 318 | // Activate the metric. A pull occurs here |
| 319 | const int64_t activationNs = configAddedTimeNs + bucketSizeNs + (2 * 1000 * 1000); // 2 millis. |
| 320 | auto batterySaverOnEvent = CreateBatterySaverOnEvent(activationNs); |
| 321 | processor->OnLogEvent(batterySaverOnEvent.get()); // 15 mins + 2 ms. |
| 322 | EXPECT_TRUE(processor->mMetricsManagers.begin()->second->mAllMetricProducers[0]->isActive()); |
| 323 | |
| 324 | processor->informPullAlarmFired(expectedPullTimeNs + 1); // 20 mins + 1 ns. |
| 325 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 3 * bucketSizeNs, expectedPullTimeNs); |
| 326 | |
| 327 | processor->informPullAlarmFired(expectedPullTimeNs + 2); // 25 mins + 2 ns. |
| 328 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 4 * bucketSizeNs, expectedPullTimeNs); |
| 329 | |
| 330 | // Create random event to deactivate metric. |
| 331 | auto deactivationEvent = CreateScreenBrightnessChangedEvent(activationNs + ttlNs + 1, 50); |
| 332 | processor->OnLogEvent(deactivationEvent.get()); |
| 333 | EXPECT_FALSE(processor->mMetricsManagers.begin()->second->mAllMetricProducers[0]->isActive()); |
| 334 | |
| 335 | processor->informPullAlarmFired(expectedPullTimeNs + 3); |
| 336 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 5 * bucketSizeNs, expectedPullTimeNs); |
| 337 | |
| 338 | processor->informPullAlarmFired(expectedPullTimeNs + 4); |
| 339 | EXPECT_EQ(baseTimeNs + startBucketNum * bucketSizeNs + 6 * bucketSizeNs, expectedPullTimeNs); |
| 340 | |
| 341 | ConfigMetricsReportList reports; |
| 342 | vector<uint8_t> buffer; |
| 343 | processor->onDumpReport(cfgKey, configAddedTimeNs + 7 * bucketSizeNs + 10, false, true, |
| 344 | ADB_DUMP, FAST, &buffer); |
| 345 | EXPECT_TRUE(buffer.size() > 0); |
| 346 | EXPECT_TRUE(reports.ParseFromArray(&buffer[0], buffer.size())); |
| 347 | backfillDimensionPath(&reports); |
| 348 | backfillStringInReport(&reports); |
| 349 | backfillStartEndTimestamp(&reports); |
| 350 | EXPECT_EQ(1, reports.reports_size()); |
| 351 | EXPECT_EQ(1, reports.reports(0).metrics_size()); |
| 352 | StatsLogReport::ValueMetricDataWrapper valueMetrics; |
| 353 | sortMetricDataByDimensionsValue(reports.reports(0).metrics(0).value_metrics(), &valueMetrics); |
| 354 | EXPECT_GT((int)valueMetrics.data_size(), 0); |
| 355 | |
| 356 | auto data = valueMetrics.data(0); |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 357 | EXPECT_EQ(util::SUBSYSTEM_SLEEP_STATE, data.dimensions_in_what().field()); |
tsaichristine | 7747d37 | 2020-02-28 17:36:59 -0800 | [diff] [blame] | 358 | EXPECT_EQ(1, data.dimensions_in_what().value_tuple().dimensions_value_size()); |
| 359 | EXPECT_EQ(1 /* subsystem name field */, |
| 360 | data.dimensions_in_what().value_tuple().dimensions_value(0).field()); |
| 361 | EXPECT_FALSE(data.dimensions_in_what().value_tuple().dimensions_value(0).value_str().empty()); |
| 362 | // We have 2 full buckets, the two surrounding the activation are dropped. |
| 363 | EXPECT_EQ(2, data.bucket_info_size()); |
| 364 | |
| 365 | auto bucketInfo = data.bucket_info(0); |
| 366 | EXPECT_EQ(baseTimeNs + 3 * bucketSizeNs, bucketInfo.start_bucket_elapsed_nanos()); |
| 367 | EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, bucketInfo.end_bucket_elapsed_nanos()); |
| 368 | EXPECT_EQ(1, bucketInfo.values_size()); |
| 369 | |
| 370 | bucketInfo = data.bucket_info(1); |
| 371 | EXPECT_EQ(baseTimeNs + 4 * bucketSizeNs, bucketInfo.start_bucket_elapsed_nanos()); |
| 372 | EXPECT_EQ(baseTimeNs + 5 * bucketSizeNs, bucketInfo.end_bucket_elapsed_nanos()); |
| 373 | EXPECT_EQ(1, bucketInfo.values_size()); |
| 374 | } |
Tej Singh | 597c716 | 2019-04-17 16:41:45 -0700 | [diff] [blame] | 375 | |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 376 | /** |
| 377 | * Test initialization of a simple value metric that is sliced by a state. |
| 378 | * |
| 379 | * ValueCpuUserTimePerScreenState |
| 380 | */ |
| 381 | TEST(ValueMetricE2eTest, TestInitWithSlicedState) { |
| 382 | // Create config. |
| 383 | StatsdConfig config; |
| 384 | config.add_allowed_log_source("AID_ROOT"); // LogEvent defaults to UID of root. |
| 385 | |
| 386 | auto pulledAtomMatcher = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 387 | CreateSimpleAtomMatcher("TestMatcher", util::SUBSYSTEM_SLEEP_STATE); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 388 | *config.add_atom_matcher() = pulledAtomMatcher; |
| 389 | |
| 390 | auto screenState = CreateScreenState(); |
| 391 | *config.add_state() = screenState; |
| 392 | |
| 393 | // Create value metric that slices by screen state without a map. |
| 394 | int64_t metricId = 123456; |
| 395 | auto valueMetric = config.add_value_metric(); |
| 396 | valueMetric->set_id(metricId); |
| 397 | valueMetric->set_bucket(TimeUnit::FIVE_MINUTES); |
| 398 | valueMetric->set_what(pulledAtomMatcher.id()); |
| 399 | *valueMetric->mutable_value_field() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 400 | CreateDimensions(util::CPU_TIME_PER_UID, {2 /* user_time_micros */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 401 | valueMetric->add_slice_by_state(screenState.id()); |
| 402 | valueMetric->set_max_pull_delay_sec(INT_MAX); |
| 403 | |
| 404 | // Initialize StatsLogProcessor. |
| 405 | const uint64_t bucketStartTimeNs = 10000000000; // 0:10 |
| 406 | const uint64_t bucketSizeNs = |
| 407 | TimeUnitToBucketSizeInMillis(config.value_metric(0).bucket()) * 1000000LL; |
| 408 | int uid = 12345; |
| 409 | int64_t cfgId = 98765; |
| 410 | ConfigKey cfgKey(uid, cfgId); |
| 411 | |
| 412 | auto processor = CreateStatsLogProcessor(bucketStartTimeNs, bucketStartTimeNs, config, cfgKey); |
| 413 | |
| 414 | // Check that StateTrackers were initialized correctly. |
| 415 | EXPECT_EQ(1, StateManager::getInstance().getStateTrackersCount()); |
| 416 | EXPECT_EQ(1, StateManager::getInstance().getListenersCount(SCREEN_STATE_ATOM_ID)); |
| 417 | |
| 418 | // Check that ValueMetricProducer was initialized correctly. |
| 419 | EXPECT_EQ(1U, processor->mMetricsManagers.size()); |
| 420 | sp<MetricsManager> metricsManager = processor->mMetricsManagers.begin()->second; |
| 421 | EXPECT_TRUE(metricsManager->isConfigValid()); |
| 422 | EXPECT_EQ(1, metricsManager->mAllMetricProducers.size()); |
| 423 | sp<MetricProducer> metricProducer = metricsManager->mAllMetricProducers[0]; |
| 424 | EXPECT_EQ(1, metricProducer->mSlicedStateAtoms.size()); |
| 425 | EXPECT_EQ(SCREEN_STATE_ATOM_ID, metricProducer->mSlicedStateAtoms.at(0)); |
| 426 | EXPECT_EQ(0, metricProducer->mStateGroupMap.size()); |
| 427 | } |
| 428 | |
| 429 | /** |
| 430 | * Test initialization of a value metric that is sliced by state and has |
| 431 | * dimensions_in_what. |
| 432 | * |
| 433 | * ValueCpuUserTimePerUidPerUidProcessState |
| 434 | */ |
| 435 | TEST(ValueMetricE2eTest, TestInitWithSlicedState_WithDimensions) { |
| 436 | // Create config. |
| 437 | StatsdConfig config; |
| 438 | config.add_allowed_log_source("AID_ROOT"); // LogEvent defaults to UID of root. |
| 439 | |
| 440 | auto cpuTimePerUidMatcher = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 441 | CreateSimpleAtomMatcher("CpuTimePerUidMatcher", util::CPU_TIME_PER_UID); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 442 | *config.add_atom_matcher() = cpuTimePerUidMatcher; |
| 443 | |
| 444 | auto uidProcessState = CreateUidProcessState(); |
| 445 | *config.add_state() = uidProcessState; |
| 446 | |
| 447 | // Create value metric that slices by screen state with a complete map. |
| 448 | int64_t metricId = 123456; |
| 449 | auto valueMetric = config.add_value_metric(); |
| 450 | valueMetric->set_id(metricId); |
| 451 | valueMetric->set_bucket(TimeUnit::FIVE_MINUTES); |
| 452 | valueMetric->set_what(cpuTimePerUidMatcher.id()); |
| 453 | *valueMetric->mutable_value_field() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 454 | CreateDimensions(util::CPU_TIME_PER_UID, {2 /* user_time_micros */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 455 | *valueMetric->mutable_dimensions_in_what() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 456 | CreateDimensions(util::CPU_TIME_PER_UID, {1 /* uid */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 457 | valueMetric->add_slice_by_state(uidProcessState.id()); |
| 458 | MetricStateLink* stateLink = valueMetric->add_state_link(); |
| 459 | stateLink->set_state_atom_id(UID_PROCESS_STATE_ATOM_ID); |
| 460 | auto fieldsInWhat = stateLink->mutable_fields_in_what(); |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 461 | *fieldsInWhat = CreateDimensions(util::CPU_TIME_PER_UID, {1 /* uid */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 462 | auto fieldsInState = stateLink->mutable_fields_in_state(); |
| 463 | *fieldsInState = CreateDimensions(UID_PROCESS_STATE_ATOM_ID, {1 /* uid */}); |
| 464 | valueMetric->set_max_pull_delay_sec(INT_MAX); |
| 465 | |
| 466 | // Initialize StatsLogProcessor. |
| 467 | const uint64_t bucketStartTimeNs = 10000000000; // 0:10 |
| 468 | int uid = 12345; |
| 469 | int64_t cfgId = 98765; |
| 470 | ConfigKey cfgKey(uid, cfgId); |
| 471 | |
| 472 | auto processor = CreateStatsLogProcessor(bucketStartTimeNs, bucketStartTimeNs, config, cfgKey); |
| 473 | |
| 474 | // Check that StateTrackers were initialized correctly. |
| 475 | EXPECT_EQ(1, StateManager::getInstance().getStateTrackersCount()); |
| 476 | EXPECT_EQ(1, StateManager::getInstance().getListenersCount(UID_PROCESS_STATE_ATOM_ID)); |
| 477 | |
| 478 | // Check that ValueMetricProducer was initialized correctly. |
| 479 | EXPECT_EQ(1U, processor->mMetricsManagers.size()); |
| 480 | sp<MetricsManager> metricsManager = processor->mMetricsManagers.begin()->second; |
| 481 | EXPECT_TRUE(metricsManager->isConfigValid()); |
| 482 | EXPECT_EQ(1, metricsManager->mAllMetricProducers.size()); |
| 483 | sp<MetricProducer> metricProducer = metricsManager->mAllMetricProducers[0]; |
| 484 | EXPECT_EQ(1, metricProducer->mSlicedStateAtoms.size()); |
| 485 | EXPECT_EQ(UID_PROCESS_STATE_ATOM_ID, metricProducer->mSlicedStateAtoms.at(0)); |
| 486 | EXPECT_EQ(0, metricProducer->mStateGroupMap.size()); |
| 487 | } |
| 488 | |
| 489 | /** |
| 490 | * Test initialization of a value metric that is sliced by state and has |
| 491 | * dimensions_in_what. |
| 492 | * |
| 493 | * ValueCpuUserTimePerUidPerUidProcessState |
| 494 | */ |
| 495 | TEST(ValueMetricE2eTest, TestInitWithSlicedState_WithIncorrectDimensions) { |
| 496 | // Create config. |
| 497 | StatsdConfig config; |
| 498 | config.add_allowed_log_source("AID_ROOT"); // LogEvent defaults to UID of root. |
| 499 | |
| 500 | auto cpuTimePerUidMatcher = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 501 | CreateSimpleAtomMatcher("CpuTimePerUidMatcher", util::CPU_TIME_PER_UID); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 502 | *config.add_atom_matcher() = cpuTimePerUidMatcher; |
| 503 | |
| 504 | auto uidProcessState = CreateUidProcessState(); |
| 505 | *config.add_state() = uidProcessState; |
| 506 | |
| 507 | // Create value metric that slices by screen state with a complete map. |
| 508 | int64_t metricId = 123456; |
| 509 | auto valueMetric = config.add_value_metric(); |
| 510 | valueMetric->set_id(metricId); |
| 511 | valueMetric->set_bucket(TimeUnit::FIVE_MINUTES); |
| 512 | valueMetric->set_what(cpuTimePerUidMatcher.id()); |
| 513 | *valueMetric->mutable_value_field() = |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 514 | CreateDimensions(util::CPU_TIME_PER_UID, {2 /* user_time_micros */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 515 | valueMetric->add_slice_by_state(uidProcessState.id()); |
| 516 | MetricStateLink* stateLink = valueMetric->add_state_link(); |
| 517 | stateLink->set_state_atom_id(UID_PROCESS_STATE_ATOM_ID); |
| 518 | auto fieldsInWhat = stateLink->mutable_fields_in_what(); |
Jeffrey Huang | 3eb84d4 | 2020-03-17 10:31:22 -0700 | [diff] [blame] | 519 | *fieldsInWhat = CreateDimensions(util::CPU_TIME_PER_UID, {1 /* uid */}); |
tsaichristine | c876b49 | 2019-12-10 13:47:05 -0800 | [diff] [blame] | 520 | auto fieldsInState = stateLink->mutable_fields_in_state(); |
| 521 | *fieldsInState = CreateDimensions(UID_PROCESS_STATE_ATOM_ID, {1 /* uid */}); |
| 522 | valueMetric->set_max_pull_delay_sec(INT_MAX); |
| 523 | |
| 524 | // Initialize StatsLogProcessor. |
| 525 | const uint64_t bucketStartTimeNs = 10000000000; // 0:10 |
| 526 | int uid = 12345; |
| 527 | int64_t cfgId = 98765; |
| 528 | ConfigKey cfgKey(uid, cfgId); |
| 529 | auto processor = CreateStatsLogProcessor(bucketStartTimeNs, bucketStartTimeNs, config, cfgKey); |
| 530 | |
| 531 | // No StateTrackers are initialized. |
| 532 | EXPECT_EQ(0, StateManager::getInstance().getStateTrackersCount()); |
| 533 | |
| 534 | // Config initialization fails. |
| 535 | EXPECT_EQ(0, processor->mMetricsManagers.size()); |
| 536 | } |
| 537 | |
Yangster-mac | 15f6bbc | 2018-04-08 11:52:26 -0700 | [diff] [blame] | 538 | #else |
| 539 | GTEST_LOG_(INFO) << "This test does nothing.\n"; |
| 540 | #endif |
| 541 | |
| 542 | } // namespace statsd |
| 543 | } // namespace os |
| 544 | } // namespace android |