Yao Chen | caf339d | 2017-10-06 16:01:10 -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 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 15 | #include <gtest/gtest.h> |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 16 | |
| 17 | #include "src/condition/ConditionTracker.h" |
| 18 | #include "src/matchers/LogMatchingTracker.h" |
| 19 | #include "src/metrics/CountMetricProducer.h" |
Yangster | 1d4d686 | 2017-10-31 12:58:51 -0700 | [diff] [blame] | 20 | #include "src/metrics/GaugeMetricProducer.h" |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 21 | #include "src/metrics/MetricProducer.h" |
Chenjie Yu | b3dda41 | 2017-10-24 13:41:59 -0700 | [diff] [blame] | 22 | #include "src/metrics/ValueMetricProducer.h" |
Joe Onorato | 9fc9edf | 2017-10-15 20:08:52 -0700 | [diff] [blame] | 23 | #include "src/metrics/metrics_manager_util.h" |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 24 | #include "statsd_test_util.h" |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 25 | |
| 26 | #include "frameworks/base/cmds/statsd/src/statsd_config.pb.h" |
| 27 | |
| 28 | #include <stdio.h> |
| 29 | #include <set> |
| 30 | #include <unordered_map> |
| 31 | #include <vector> |
| 32 | |
| 33 | using namespace android::os::statsd; |
| 34 | using android::sp; |
| 35 | using std::set; |
| 36 | using std::unordered_map; |
| 37 | using std::vector; |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 38 | using android::os::statsd::Predicate; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 39 | |
| 40 | #ifdef __ANDROID__ |
| 41 | |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 42 | const ConfigKey kConfigKey(0, 12345); |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 43 | const long kAlertId = 3; |
Yao Chen | b356151 | 2017-11-21 18:07:17 -0800 | [diff] [blame] | 44 | |
Chenjie Yu | 85ed838 | 2017-12-14 16:48:54 -0800 | [diff] [blame] | 45 | const long timeBaseSec = 1000; |
| 46 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 47 | StatsdConfig buildGoodConfig() { |
| 48 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 49 | config.set_id(12345); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 50 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 51 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 52 | eventMatcher->set_id(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 53 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 54 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 55 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 56 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 57 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 58 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 59 | 2 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_ON*/); |
| 60 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 61 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 62 | eventMatcher->set_id(StringToId("SCREEN_IS_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 63 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 64 | simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 65 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 66 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 67 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 68 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 69 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_OFF*/); |
| 70 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 71 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 72 | eventMatcher->set_id(StringToId("SCREEN_ON_OR_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 73 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 74 | AtomMatcher_Combination* combination = eventMatcher->mutable_combination(); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 75 | combination->set_operation(LogicalOperation::OR); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 76 | combination->add_matcher(StringToId("SCREEN_IS_ON")); |
| 77 | combination->add_matcher(StringToId("SCREEN_IS_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 78 | |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 79 | CountMetric* metric = config.add_count_metric(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 80 | metric->set_id(3); |
| 81 | metric->set_what(StringToId("SCREEN_IS_ON")); |
Yangster-mac | b814481 | 2018-01-04 10:56:23 -0800 | [diff] [blame] | 82 | metric->set_bucket(ONE_MINUTE); |
Yangster-mac | 468ff04 | 2018-01-17 12:26:34 -0800 | [diff] [blame] | 83 | metric->mutable_dimensions_in_what()->set_field(2 /*SCREEN_STATE_CHANGE*/); |
| 84 | metric->mutable_dimensions_in_what()->add_child()->set_field(1); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 85 | |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 86 | config.add_no_report_metric(3); |
| 87 | |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 88 | auto alert = config.add_alert(); |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 89 | alert->set_id(kAlertId); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 90 | alert->set_metric_id(3); |
Yangster-mac | a7fb12d | 2018-01-03 17:17:20 -0800 | [diff] [blame] | 91 | alert->set_num_buckets(10); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 92 | alert->set_refractory_period_secs(100); |
| 93 | alert->set_trigger_if_sum_gt(100); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 94 | return config; |
| 95 | } |
| 96 | |
| 97 | StatsdConfig buildCircleMatchers() { |
| 98 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 99 | config.set_id(12345); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 100 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 101 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 102 | eventMatcher->set_id(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 103 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 104 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 105 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 106 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 107 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 108 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 109 | 2 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_ON*/); |
| 110 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 111 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 112 | eventMatcher->set_id(StringToId("SCREEN_ON_OR_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 113 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 114 | AtomMatcher_Combination* combination = eventMatcher->mutable_combination(); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 115 | combination->set_operation(LogicalOperation::OR); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 116 | combination->add_matcher(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 117 | // Circle dependency |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 118 | combination->add_matcher(StringToId("SCREEN_ON_OR_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 119 | |
| 120 | return config; |
| 121 | } |
| 122 | |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 123 | StatsdConfig buildAlertWithUnknownMetric() { |
| 124 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 125 | config.set_id(12345); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 126 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 127 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 128 | eventMatcher->set_id(StringToId("SCREEN_IS_ON")); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 129 | |
| 130 | CountMetric* metric = config.add_count_metric(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 131 | metric->set_id(3); |
| 132 | metric->set_what(StringToId("SCREEN_IS_ON")); |
Yangster-mac | b814481 | 2018-01-04 10:56:23 -0800 | [diff] [blame] | 133 | metric->set_bucket(ONE_MINUTE); |
Yangster-mac | 468ff04 | 2018-01-17 12:26:34 -0800 | [diff] [blame] | 134 | metric->mutable_dimensions_in_what()->set_field(2 /*SCREEN_STATE_CHANGE*/); |
| 135 | metric->mutable_dimensions_in_what()->add_child()->set_field(1); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 136 | |
| 137 | auto alert = config.add_alert(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 138 | alert->set_id(3); |
| 139 | alert->set_metric_id(2); |
Yangster-mac | a7fb12d | 2018-01-03 17:17:20 -0800 | [diff] [blame] | 140 | alert->set_num_buckets(10); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 141 | alert->set_refractory_period_secs(100); |
| 142 | alert->set_trigger_if_sum_gt(100); |
| 143 | return config; |
| 144 | } |
| 145 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 146 | StatsdConfig buildMissingMatchers() { |
| 147 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 148 | config.set_id(12345); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 149 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 150 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 151 | eventMatcher->set_id(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 152 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 153 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 154 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 155 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 156 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 157 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 158 | 2 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_ON*/); |
| 159 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 160 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 161 | eventMatcher->set_id(StringToId("SCREEN_ON_OR_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 162 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 163 | AtomMatcher_Combination* combination = eventMatcher->mutable_combination(); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 164 | combination->set_operation(LogicalOperation::OR); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 165 | combination->add_matcher(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 166 | // undefined matcher |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 167 | combination->add_matcher(StringToId("ABC")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 168 | |
| 169 | return config; |
| 170 | } |
| 171 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 172 | StatsdConfig buildMissingPredicate() { |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 173 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 174 | config.set_id(12345); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 175 | |
| 176 | CountMetric* metric = config.add_count_metric(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 177 | metric->set_id(3); |
| 178 | metric->set_what(StringToId("SCREEN_EVENT")); |
Yangster-mac | b814481 | 2018-01-04 10:56:23 -0800 | [diff] [blame] | 179 | metric->set_bucket(ONE_MINUTE); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 180 | metric->set_condition(StringToId("SOME_CONDITION")); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 181 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 182 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 183 | eventMatcher->set_id(StringToId("SCREEN_EVENT")); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 184 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 185 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 186 | simpleAtomMatcher->set_atom_id(2); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 187 | |
| 188 | return config; |
| 189 | } |
| 190 | |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 191 | StatsdConfig buildDimensionMetricsWithMultiTags() { |
| 192 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 193 | config.set_id(12345); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 194 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 195 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 196 | eventMatcher->set_id(StringToId("BATTERY_VERY_LOW")); |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 197 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 198 | simpleAtomMatcher->set_atom_id(2); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 199 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 200 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 201 | eventMatcher->set_id(StringToId("BATTERY_VERY_VERY_LOW")); |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 202 | simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 203 | simpleAtomMatcher->set_atom_id(3); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 204 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 205 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 206 | eventMatcher->set_id(StringToId("BATTERY_LOW")); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 207 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 208 | AtomMatcher_Combination* combination = eventMatcher->mutable_combination(); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 209 | combination->set_operation(LogicalOperation::OR); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 210 | combination->add_matcher(StringToId("BATTERY_VERY_LOW")); |
| 211 | combination->add_matcher(StringToId("BATTERY_VERY_VERY_LOW")); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 212 | |
| 213 | // Count process state changes, slice by uid, while SCREEN_IS_OFF |
| 214 | CountMetric* metric = config.add_count_metric(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 215 | metric->set_id(3); |
| 216 | metric->set_what(StringToId("BATTERY_LOW")); |
Yangster-mac | b814481 | 2018-01-04 10:56:23 -0800 | [diff] [blame] | 217 | metric->set_bucket(ONE_MINUTE); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 218 | // This case is interesting. We want to dimension across two atoms. |
Yangster-mac | 468ff04 | 2018-01-17 12:26:34 -0800 | [diff] [blame] | 219 | metric->mutable_dimensions_in_what()->add_child()->set_field(1); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 220 | |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 221 | auto alert = config.add_alert(); |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 222 | alert->set_id(kAlertId); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 223 | alert->set_metric_id(3); |
Yangster-mac | a7fb12d | 2018-01-03 17:17:20 -0800 | [diff] [blame] | 224 | alert->set_num_buckets(10); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 225 | alert->set_refractory_period_secs(100); |
| 226 | alert->set_trigger_if_sum_gt(100); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 227 | return config; |
| 228 | } |
| 229 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 230 | StatsdConfig buildCirclePredicates() { |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 231 | StatsdConfig config; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 232 | config.set_id(12345); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 233 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 234 | AtomMatcher* eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 235 | eventMatcher->set_id(StringToId("SCREEN_IS_ON")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 236 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 237 | SimpleAtomMatcher* simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 238 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 239 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 240 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 241 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 242 | 2 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_ON*/); |
| 243 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 244 | eventMatcher = config.add_atom_matcher(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 245 | eventMatcher->set_id(StringToId("SCREEN_IS_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 246 | |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 247 | simpleAtomMatcher = eventMatcher->mutable_simple_atom_matcher(); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 248 | simpleAtomMatcher->set_atom_id(2 /*SCREEN_STATE_CHANGE*/); |
| 249 | simpleAtomMatcher->add_field_value_matcher()->set_field( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 250 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE*/); |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 251 | simpleAtomMatcher->mutable_field_value_matcher(0)->set_eq_int( |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 252 | 1 /*SCREEN_STATE_CHANGE__DISPLAY_STATE__STATE_OFF*/); |
| 253 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 254 | auto condition = config.add_predicate(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 255 | condition->set_id(StringToId("SCREEN_IS_ON")); |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 256 | SimplePredicate* simplePredicate = condition->mutable_simple_predicate(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 257 | simplePredicate->set_start(StringToId("SCREEN_IS_ON")); |
| 258 | simplePredicate->set_stop(StringToId("SCREEN_IS_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 259 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 260 | condition = config.add_predicate(); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 261 | condition->set_id(StringToId("SCREEN_IS_EITHER_ON_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 262 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 263 | Predicate_Combination* combination = condition->mutable_combination(); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 264 | combination->set_operation(LogicalOperation::OR); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 265 | combination->add_predicate(StringToId("SCREEN_IS_ON")); |
| 266 | combination->add_predicate(StringToId("SCREEN_IS_EITHER_ON_OFF")); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 267 | |
| 268 | return config; |
| 269 | } |
| 270 | |
| 271 | TEST(MetricsManagerTest, TestGoodConfig) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 272 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 273 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 274 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 275 | sp<AlarmMonitor> periodicAlarmMonitor; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 276 | StatsdConfig config = buildGoodConfig(); |
| 277 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 278 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 279 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 280 | vector<sp<MetricProducer>> allMetricProducers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 281 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 282 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 283 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 284 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 285 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 286 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 287 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 288 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 289 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 290 | std::set<int64_t> noReportMetricIds; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 291 | |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 292 | EXPECT_TRUE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 293 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 294 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 295 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 296 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 297 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 298 | alertTrackerMap, metricsWithActivation, |
| 299 | noReportMetricIds)); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 300 | EXPECT_EQ(1u, allMetricProducers.size()); |
| 301 | EXPECT_EQ(1u, allAnomalyTrackers.size()); |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 302 | EXPECT_EQ(1u, noReportMetricIds.size()); |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 303 | EXPECT_EQ(1u, alertTrackerMap.size()); |
| 304 | EXPECT_NE(alertTrackerMap.find(kAlertId), alertTrackerMap.end()); |
| 305 | EXPECT_EQ(alertTrackerMap.find(kAlertId)->second, 0); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 306 | } |
| 307 | |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 308 | TEST(MetricsManagerTest, TestDimensionMetricsWithMultiTags) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 309 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 310 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 311 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 312 | sp<AlarmMonitor> periodicAlarmMonitor; |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 313 | StatsdConfig config = buildDimensionMetricsWithMultiTags(); |
| 314 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 315 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 316 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 317 | vector<sp<MetricProducer>> allMetricProducers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 318 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 319 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 320 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 321 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 322 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 323 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 324 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 325 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 326 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 327 | std::set<int64_t> noReportMetricIds; |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 328 | |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 329 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 330 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 331 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 332 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 333 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 334 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 335 | alertTrackerMap, metricsWithActivation, |
| 336 | noReportMetricIds)); |
Yao Chen | 5154a379 | 2017-10-30 22:57:06 -0700 | [diff] [blame] | 337 | } |
| 338 | |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 339 | TEST(MetricsManagerTest, TestCircleLogMatcherDependency) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 340 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 341 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 342 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 343 | sp<AlarmMonitor> periodicAlarmMonitor; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 344 | StatsdConfig config = buildCircleMatchers(); |
| 345 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 346 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 347 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 348 | vector<sp<MetricProducer>> allMetricProducers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 349 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 350 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 351 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 352 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 353 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 354 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 355 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 356 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 357 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 358 | std::set<int64_t> noReportMetricIds; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 359 | |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 360 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 361 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 362 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 363 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 364 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 365 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 366 | alertTrackerMap, metricsWithActivation, |
| 367 | noReportMetricIds)); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 368 | } |
| 369 | |
| 370 | TEST(MetricsManagerTest, TestMissingMatchers) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 371 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 372 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 373 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 374 | sp<AlarmMonitor> periodicAlarmMonitor; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 375 | StatsdConfig config = buildMissingMatchers(); |
| 376 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 377 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 378 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 379 | vector<sp<MetricProducer>> allMetricProducers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 380 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 381 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 382 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 383 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 384 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 385 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 386 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 387 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 388 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 389 | std::set<int64_t> noReportMetricIds; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 390 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 391 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 392 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 393 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 394 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 395 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 396 | alertTrackerMap, metricsWithActivation, |
| 397 | noReportMetricIds)); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 398 | } |
| 399 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 400 | TEST(MetricsManagerTest, TestMissingPredicate) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 401 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 402 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 403 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 404 | sp<AlarmMonitor> periodicAlarmMonitor; |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 405 | StatsdConfig config = buildMissingPredicate(); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 406 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 407 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 408 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 409 | vector<sp<MetricProducer>> allMetricProducers; |
| 410 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 411 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 412 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 413 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 414 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 415 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 416 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 417 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 418 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 419 | std::set<int64_t> noReportMetricIds; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 420 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 421 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 422 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 423 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 424 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 425 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 426 | alertTrackerMap, metricsWithActivation, noReportMetricIds)); |
Yao Chen | 10535b9 | 2017-11-27 11:31:55 -0800 | [diff] [blame] | 427 | } |
| 428 | |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 429 | TEST(MetricsManagerTest, TestCirclePredicateDependency) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 430 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 431 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 432 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 433 | sp<AlarmMonitor> periodicAlarmMonitor; |
Stefan Lafon | 12d01fa | 2017-12-04 20:56:09 -0800 | [diff] [blame] | 434 | StatsdConfig config = buildCirclePredicates(); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 435 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 436 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 437 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 438 | vector<sp<MetricProducer>> allMetricProducers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 439 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 440 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 441 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 442 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 443 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 444 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 445 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 446 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 447 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 448 | std::set<int64_t> noReportMetricIds; |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 449 | |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 450 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 451 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 452 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 453 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 454 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 455 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 456 | alertTrackerMap, metricsWithActivation, |
| 457 | noReportMetricIds)); |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 458 | } |
| 459 | |
| 460 | TEST(MetricsManagerTest, testAlertWithUnknownMetric) { |
Yangster-mac | 2087716 | 2017-12-22 17:19:39 -0800 | [diff] [blame] | 461 | UidMap uidMap; |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 462 | sp<StatsPullerManager> pullerManager = new StatsPullerManager(); |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 463 | sp<AlarmMonitor> anomalyAlarmMonitor; |
| 464 | sp<AlarmMonitor> periodicAlarmMonitor; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 465 | StatsdConfig config = buildAlertWithUnknownMetric(); |
| 466 | set<int> allTagIds; |
Stefan Lafon | b8c9aa8 | 2017-12-03 14:27:25 -0800 | [diff] [blame] | 467 | vector<sp<LogMatchingTracker>> allAtomMatchers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 468 | vector<sp<ConditionTracker>> allConditionTrackers; |
| 469 | vector<sp<MetricProducer>> allMetricProducers; |
| 470 | std::vector<sp<AnomalyTracker>> allAnomalyTrackers; |
Yangster-mac | 932ecec | 2018-02-01 10:23:52 -0800 | [diff] [blame] | 471 | std::vector<sp<AlarmTracker>> allAlarmTrackers; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 472 | unordered_map<int, std::vector<int>> conditionToMetricMap; |
| 473 | unordered_map<int, std::vector<int>> trackerToMetricMap; |
| 474 | unordered_map<int, std::vector<int>> trackerToConditionMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 475 | unordered_map<int, std::vector<int>> activationAtomTrackerToMetricMap; |
| 476 | unordered_map<int, std::vector<int>> deactivationAtomTrackerToMetricMap; |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 477 | unordered_map<int64_t, int> alertTrackerMap; |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 478 | vector<int> metricsWithActivation; |
Yangster-mac | 94e197c | 2018-01-02 16:03:03 -0800 | [diff] [blame] | 479 | std::set<int64_t> noReportMetricIds; |
Yangster-mac | e2cd6d5 | 2017-11-09 20:38:30 -0800 | [diff] [blame] | 480 | |
Chenjie Yu | e221920 | 2018-06-08 10:07:51 -0700 | [diff] [blame] | 481 | EXPECT_FALSE(initStatsdConfig(kConfigKey, config, uidMap, pullerManager, anomalyAlarmMonitor, |
| 482 | periodicAlarmMonitor, timeBaseSec, timeBaseSec, allTagIds, |
| 483 | allAtomMatchers, allConditionTrackers, allMetricProducers, |
| 484 | allAnomalyTrackers, allAlarmTrackers, conditionToMetricMap, |
Yangster-mac | 849dfdc2 | 2018-10-12 15:41:45 -0700 | [diff] [blame] | 485 | trackerToMetricMap, trackerToConditionMap, |
Muhammad Qureshi | 3a5ebf5 | 2019-03-28 12:38:21 -0700 | [diff] [blame] | 486 | activationAtomTrackerToMetricMap, deactivationAtomTrackerToMetricMap, |
Jeffrey Huang | 38d7026 | 2020-03-19 17:28:59 -0700 | [diff] [blame^] | 487 | alertTrackerMap, metricsWithActivation, |
| 488 | noReportMetricIds)); |
Yao Chen | caf339d | 2017-10-06 16:01:10 -0700 | [diff] [blame] | 489 | } |
| 490 | |
| 491 | #else |
| 492 | GTEST_LOG_(INFO) << "This test does nothing.\n"; |
| 493 | #endif |