| // Copyright (C) 2017 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "src/metrics/duration_helper/MaxDurationTracker.h" |
| #include "src/condition/ConditionWizard.h" |
| #include "metrics_test_helper.h" |
| #include "tests/statsd_test_util.h" |
| |
| #include <gmock/gmock.h> |
| #include <gtest/gtest.h> |
| #include <stdio.h> |
| #include <set> |
| #include <unordered_map> |
| #include <vector> |
| |
| using namespace android::os::statsd; |
| using namespace testing; |
| using android::sp; |
| using std::set; |
| using std::unordered_map; |
| using std::vector; |
| |
| #ifdef __ANDROID__ |
| |
| namespace android { |
| namespace os { |
| namespace statsd { |
| |
| const ConfigKey kConfigKey(0, 12345); |
| |
| const int TagId = 1; |
| |
| const HashableDimensionKey eventKey = getMockedDimensionKey(TagId, 0, "1"); |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| const HashableDimensionKey key2 = getMockedDimensionKey(TagId, 1, "2"); |
| const uint64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL; |
| |
| TEST(MaxDurationTrackerTest, TestSimpleMaxDuration) { |
| const MetricDimensionKey eventKey = getMockedMetricDimensionKey(TagId, 0, "1"); |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| const HashableDimensionKey key2 = getMockedDimensionKey(TagId, 1, "2"); |
| |
| FieldMatcher dimensionInCondition; |
| sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>(); |
| |
| unordered_map<MetricDimensionKey, vector<DurationBucket>> buckets; |
| |
| uint64_t bucketStartTimeNs = 10000000000; |
| uint64_t bucketEndTimeNs = bucketStartTimeNs + bucketSizeNs; |
| uint64_t bucketNum = 0; |
| |
| int64_t metricId = 1; |
| MaxDurationTracker tracker(kConfigKey, metricId, eventKey, wizard, -1, dimensionInCondition, |
| false, bucketStartTimeNs, bucketNum, bucketStartTimeNs, bucketSizeNs, |
| false, {}); |
| |
| tracker.noteStart(key1, true, bucketStartTimeNs, ConditionKey()); |
| // Event starts again. This would not change anything as it already starts. |
| tracker.noteStart(key1, true, bucketStartTimeNs + 3, ConditionKey()); |
| // Stopped. |
| tracker.noteStop(key1, bucketStartTimeNs + 10, false); |
| |
| // Another event starts in this bucket. |
| tracker.noteStart(key2, true, bucketStartTimeNs + 20, ConditionKey()); |
| tracker.noteStop(key2, bucketStartTimeNs + 40, false /*stop all*/); |
| |
| tracker.flushIfNeeded(bucketStartTimeNs + bucketSizeNs + 1, &buckets); |
| EXPECT_TRUE(buckets.find(eventKey) != buckets.end()); |
| EXPECT_EQ(1u, buckets[eventKey].size()); |
| EXPECT_EQ(20ULL, buckets[eventKey][0].mDuration); |
| } |
| |
| TEST(MaxDurationTrackerTest, TestStopAll) { |
| const MetricDimensionKey eventKey = getMockedMetricDimensionKey(TagId, 0, "1"); |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| const HashableDimensionKey key2 = getMockedDimensionKey(TagId, 1, "2"); |
| |
| FieldMatcher dimensionInCondition; |
| sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>(); |
| |
| unordered_map<MetricDimensionKey, vector<DurationBucket>> buckets; |
| |
| uint64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL; |
| uint64_t bucketStartTimeNs = 10000000000; |
| uint64_t bucketEndTimeNs = bucketStartTimeNs + bucketSizeNs; |
| uint64_t bucketNum = 0; |
| |
| int64_t metricId = 1; |
| MaxDurationTracker tracker(kConfigKey, metricId, eventKey, wizard, -1, dimensionInCondition, |
| false, bucketStartTimeNs, bucketNum, bucketStartTimeNs, bucketSizeNs, |
| false, {}); |
| |
| tracker.noteStart(key1, true, bucketStartTimeNs + 1, ConditionKey()); |
| |
| // Another event starts in this bucket. |
| tracker.noteStart(key2, true, bucketStartTimeNs + 20, ConditionKey()); |
| tracker.flushIfNeeded(bucketStartTimeNs + bucketSizeNs + 40, &buckets); |
| tracker.noteStopAll(bucketStartTimeNs + bucketSizeNs + 40); |
| EXPECT_TRUE(tracker.mInfos.empty()); |
| EXPECT_TRUE(buckets.find(eventKey) == buckets.end()); |
| |
| tracker.flushIfNeeded(bucketStartTimeNs + 3 * bucketSizeNs + 40, &buckets); |
| EXPECT_TRUE(buckets.find(eventKey) != buckets.end()); |
| EXPECT_EQ(1u, buckets[eventKey].size()); |
| EXPECT_EQ(bucketSizeNs + 40 - 1, buckets[eventKey][0].mDuration); |
| EXPECT_EQ(bucketStartTimeNs + bucketSizeNs, buckets[eventKey][0].mBucketStartNs); |
| EXPECT_EQ(bucketStartTimeNs + 2 * bucketSizeNs, buckets[eventKey][0].mBucketEndNs); |
| } |
| |
| TEST(MaxDurationTrackerTest, TestCrossBucketBoundary) { |
| const MetricDimensionKey eventKey = getMockedMetricDimensionKey(TagId, 0, "1"); |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| const HashableDimensionKey key2 = getMockedDimensionKey(TagId, 1, "2"); |
| FieldMatcher dimensionInCondition; |
| sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>(); |
| |
| unordered_map<MetricDimensionKey, vector<DurationBucket>> buckets; |
| |
| uint64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL; |
| uint64_t bucketStartTimeNs = 10000000000; |
| uint64_t bucketEndTimeNs = bucketStartTimeNs + bucketSizeNs; |
| uint64_t bucketNum = 0; |
| |
| int64_t metricId = 1; |
| MaxDurationTracker tracker(kConfigKey, metricId, eventKey, wizard, -1, dimensionInCondition, |
| false, bucketStartTimeNs, bucketNum, bucketStartTimeNs, bucketSizeNs, |
| false, {}); |
| |
| // The event starts. |
| tracker.noteStart(DEFAULT_DIMENSION_KEY, true, bucketStartTimeNs + 1, ConditionKey()); |
| |
| // Starts again. Does not DEFAULT_DIMENSION_KEY anything. |
| tracker.noteStart(DEFAULT_DIMENSION_KEY, true, bucketStartTimeNs + bucketSizeNs + 1, |
| ConditionKey()); |
| |
| // The event stops at early 4th bucket. |
| // Notestop is called from DurationMetricProducer's onMatchedLogEvent, which calls |
| // flushIfneeded. |
| tracker.flushIfNeeded(bucketStartTimeNs + (3 * bucketSizeNs) + 20, &buckets); |
| tracker.noteStop(DEFAULT_DIMENSION_KEY, bucketStartTimeNs + (3 * bucketSizeNs) + 20, |
| false /*stop all*/); |
| EXPECT_TRUE(buckets.find(eventKey) == buckets.end()); |
| |
| tracker.flushIfNeeded(bucketStartTimeNs + 4 * bucketSizeNs, &buckets); |
| EXPECT_EQ(1u, buckets[eventKey].size()); |
| EXPECT_EQ((3 * bucketSizeNs) + 20 - 1, buckets[eventKey][0].mDuration); |
| EXPECT_EQ(bucketStartTimeNs + 3 * bucketSizeNs, buckets[eventKey][0].mBucketStartNs); |
| EXPECT_EQ(bucketStartTimeNs + 4 * bucketSizeNs, buckets[eventKey][0].mBucketEndNs); |
| } |
| |
| TEST(MaxDurationTrackerTest, TestCrossBucketBoundary_nested) { |
| const MetricDimensionKey eventKey = getMockedMetricDimensionKey(TagId, 0, "1"); |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| const HashableDimensionKey key2 = getMockedDimensionKey(TagId, 1, "2"); |
| FieldMatcher dimensionInCondition; |
| sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>(); |
| |
| unordered_map<MetricDimensionKey, vector<DurationBucket>> buckets; |
| |
| uint64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL; |
| uint64_t bucketStartTimeNs = 10000000000; |
| uint64_t bucketEndTimeNs = bucketStartTimeNs + bucketSizeNs; |
| uint64_t bucketNum = 0; |
| |
| int64_t metricId = 1; |
| MaxDurationTracker tracker(kConfigKey, metricId, eventKey, wizard, -1, dimensionInCondition, |
| true, bucketStartTimeNs, bucketNum, bucketStartTimeNs, bucketSizeNs, |
| false, {}); |
| |
| // 2 starts |
| tracker.noteStart(DEFAULT_DIMENSION_KEY, true, bucketStartTimeNs + 1, ConditionKey()); |
| tracker.noteStart(DEFAULT_DIMENSION_KEY, true, bucketStartTimeNs + 10, ConditionKey()); |
| // one stop |
| tracker.noteStop(DEFAULT_DIMENSION_KEY, bucketStartTimeNs + 20, false /*stop all*/); |
| |
| tracker.flushIfNeeded(bucketStartTimeNs + (2 * bucketSizeNs) + 1, &buckets); |
| // Because of nesting, still not stopped. |
| EXPECT_TRUE(buckets.find(eventKey) == buckets.end()); |
| |
| // real stop now. |
| tracker.noteStop(DEFAULT_DIMENSION_KEY, |
| bucketStartTimeNs + (2 * bucketSizeNs) + 5, false); |
| tracker.flushIfNeeded(bucketStartTimeNs + (3 * bucketSizeNs) + 1, &buckets); |
| |
| EXPECT_EQ(1u, buckets[eventKey].size()); |
| EXPECT_EQ(2 * bucketSizeNs + 5 - 1, buckets[eventKey][0].mDuration); |
| } |
| |
| TEST(MaxDurationTrackerTest, TestMaxDurationWithCondition) { |
| const std::vector<HashableDimensionKey> conditionKey = {getMockedDimensionKey(TagId, 4, "1")}; |
| const HashableDimensionKey key1 = getMockedDimensionKey(TagId, 1, "1"); |
| |
| FieldMatcher dimensionInCondition; |
| sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>(); |
| |
| ConditionKey conditionKey1; |
| MetricDimensionKey eventKey = getMockedMetricDimensionKey(TagId, 2, "maps"); |
| conditionKey1[StringToId("APP_BACKGROUND")] = conditionKey; |
| |
| EXPECT_CALL(*wizard, query(_, conditionKey1, _, _)) // #4 |
| .WillOnce(Return(ConditionState::kFalse)); |
| |
| unordered_map<MetricDimensionKey, vector<DurationBucket>> buckets; |
| |
| uint64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL; |
| uint64_t bucketStartTimeNs = 10000000000; |
| uint64_t bucketEndTimeNs = bucketStartTimeNs + bucketSizeNs; |
| uint64_t bucketNum = 0; |
| uint64_t eventStartTimeNs = bucketStartTimeNs + 1; |
| int64_t durationTimeNs = 2 * 1000; |
| |
| int64_t metricId = 1; |
| MaxDurationTracker tracker(kConfigKey, metricId, eventKey, wizard, 1, dimensionInCondition, |
| false, bucketStartTimeNs, bucketNum, bucketStartTimeNs, bucketSizeNs, |
| true, {}); |
| EXPECT_TRUE(tracker.mAnomalyTrackers.empty()); |
| |
| tracker.noteStart(key1, true, eventStartTimeNs, conditionKey1); |
| |
| tracker.onSlicedConditionMayChange(eventStartTimeNs + 5); |
| |
| tracker.noteStop(key1, eventStartTimeNs + durationTimeNs, false); |
| |
| tracker.flushIfNeeded(bucketStartTimeNs + bucketSizeNs + 1, &buckets); |
| EXPECT_TRUE(buckets.find(eventKey) != buckets.end()); |
| EXPECT_EQ(1u, buckets[eventKey].size()); |
| EXPECT_EQ(5ULL, buckets[eventKey][0].mDuration); |
| } |
| |
| } // namespace statsd |
| } // namespace os |
| } // namespace android |
| #else |
| GTEST_LOG_(INFO) << "This test does nothing.\n"; |
| #endif |