256 lines
10 KiB
C++
256 lines
10 KiB
C++
// 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/config/ConfigManager.h"
|
|
#include "src/metrics/MetricsManager.h"
|
|
#include "statsd_test_util.h"
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <stdio.h>
|
|
#include <iostream>
|
|
|
|
using namespace android;
|
|
using namespace android::os::statsd;
|
|
using namespace testing;
|
|
using namespace std;
|
|
|
|
namespace android {
|
|
namespace os {
|
|
namespace statsd {
|
|
|
|
static ostream& operator<<(ostream& os, const StatsdConfig& config) {
|
|
return os << "StatsdConfig{id=" << config.id() << "}";
|
|
}
|
|
|
|
} // namespace statsd
|
|
} // namespace os
|
|
} // namespace android
|
|
|
|
/**
|
|
* Mock ConfigListener
|
|
*/
|
|
class MockListener : public ConfigListener {
|
|
public:
|
|
MOCK_METHOD4(OnConfigUpdated, void(const int64_t timestampNs, const ConfigKey& key,
|
|
const StatsdConfig& config, bool modularUpdate));
|
|
MOCK_METHOD1(OnConfigRemoved, void(const ConfigKey& key));
|
|
};
|
|
|
|
/**
|
|
* Validate that the ConfigKey is the one we wanted.
|
|
*/
|
|
MATCHER_P2(ConfigKeyEq, uid, id, "") {
|
|
return arg.GetUid() == uid && (long long)arg.GetId() == (long long)id;
|
|
}
|
|
|
|
/**
|
|
* Validate that the StatsdConfig is the one we wanted.
|
|
*/
|
|
MATCHER_P(StatsdConfigEq, id, 0) {
|
|
return (long long)arg.id() == (long long)id;
|
|
}
|
|
|
|
const int64_t testConfigId = 12345;
|
|
|
|
/**
|
|
* Test the addOrUpdate and remove methods
|
|
*/
|
|
TEST(ConfigManagerTest, TestAddUpdateRemove) {
|
|
sp<MockListener> listener = new StrictMock<MockListener>();
|
|
|
|
sp<ConfigManager> manager = new ConfigManager();
|
|
manager->AddListener(listener);
|
|
|
|
StatsdConfig config91;
|
|
config91.set_id(91);
|
|
StatsdConfig config92;
|
|
config92.set_id(92);
|
|
StatsdConfig config93;
|
|
config93.set_id(93);
|
|
StatsdConfig config94;
|
|
config94.set_id(94);
|
|
|
|
{
|
|
InSequence s;
|
|
|
|
manager->StartupForTest();
|
|
|
|
// Add another one
|
|
EXPECT_CALL(*(listener.get()),
|
|
OnConfigUpdated(_, ConfigKeyEq(1, StringToId("zzz")), StatsdConfigEq(91), true))
|
|
.RetiresOnSaturation();
|
|
manager->UpdateConfig(ConfigKey(1, StringToId("zzz")), config91);
|
|
|
|
// Update It
|
|
EXPECT_CALL(*(listener.get()),
|
|
OnConfigUpdated(_, ConfigKeyEq(1, StringToId("zzz")), StatsdConfigEq(92), true))
|
|
.RetiresOnSaturation();
|
|
manager->UpdateConfig(ConfigKey(1, StringToId("zzz")), config92);
|
|
|
|
// Add one with the same uid but a different name
|
|
EXPECT_CALL(*(listener.get()),
|
|
OnConfigUpdated(_, ConfigKeyEq(1, StringToId("yyy")), StatsdConfigEq(93), true))
|
|
.RetiresOnSaturation();
|
|
manager->UpdateConfig(ConfigKey(1, StringToId("yyy")), config93);
|
|
|
|
// Add one with the same name but a different uid
|
|
EXPECT_CALL(*(listener.get()),
|
|
OnConfigUpdated(_, ConfigKeyEq(2, StringToId("zzz")), StatsdConfigEq(94), true))
|
|
.RetiresOnSaturation();
|
|
manager->UpdateConfig(ConfigKey(2, StringToId("zzz")), config94);
|
|
|
|
// Remove (1,yyy)
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(1, StringToId("yyy"))))
|
|
.RetiresOnSaturation();
|
|
manager->RemoveConfig(ConfigKey(1, StringToId("yyy")));
|
|
|
|
// Remove (2,zzz)
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(2, StringToId("zzz"))))
|
|
.RetiresOnSaturation();
|
|
manager->RemoveConfig(ConfigKey(2, StringToId("zzz")));
|
|
|
|
// Remove (1,zzz)
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(1, StringToId("zzz"))))
|
|
.RetiresOnSaturation();
|
|
manager->RemoveConfig(ConfigKey(1, StringToId("zzz")));
|
|
|
|
// Remove (2,zzz) again and we shouldn't get the callback
|
|
manager->RemoveConfig(ConfigKey(2, StringToId("zzz")));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Test removing all of the configs for a uid.
|
|
*/
|
|
TEST(ConfigManagerTest, TestRemoveUid) {
|
|
sp<MockListener> listener = new StrictMock<MockListener>();
|
|
|
|
sp<ConfigManager> manager = new ConfigManager();
|
|
manager->AddListener(listener);
|
|
|
|
StatsdConfig config;
|
|
|
|
EXPECT_CALL(*(listener.get()), OnConfigUpdated(_, _, _, true)).Times(5);
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(2, StringToId("xxx"))));
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(2, StringToId("yyy"))));
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(2, StringToId("zzz"))));
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(1, StringToId("aaa"))));
|
|
EXPECT_CALL(*(listener.get()), OnConfigRemoved(ConfigKeyEq(3, StringToId("bbb"))));
|
|
|
|
manager->StartupForTest();
|
|
manager->UpdateConfig(ConfigKey(1, StringToId("aaa")), config);
|
|
manager->UpdateConfig(ConfigKey(2, StringToId("xxx")), config);
|
|
manager->UpdateConfig(ConfigKey(2, StringToId("yyy")), config);
|
|
manager->UpdateConfig(ConfigKey(2, StringToId("zzz")), config);
|
|
manager->UpdateConfig(ConfigKey(3, StringToId("bbb")), config);
|
|
|
|
manager->RemoveConfigs(2);
|
|
manager->RemoveConfigs(1);
|
|
manager->RemoveConfigs(3);
|
|
}
|
|
|
|
TEST(ConfigManagerTest, TestSendRestrictedMetricsChangedBroadcast) {
|
|
const string configPackage = "pkg";
|
|
const int64_t configId = 123;
|
|
const int32_t callingUid = 456;
|
|
const vector<int64_t> metricIds = {100, 200, 999};
|
|
|
|
shared_ptr<MockPendingIntentRef> pir = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir, sendRestrictedMetricsChangedBroadcast(metricIds))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
sp<ConfigManager> manager = new ConfigManager();
|
|
manager->SetRestrictedMetricsChangedReceiver(configPackage, configId, callingUid, pir);
|
|
|
|
manager->SendRestrictedMetricsBroadcast({configPackage}, configId, {callingUid}, metricIds);
|
|
}
|
|
|
|
TEST(ConfigManagerTest, TestSendRestrictedMetricsChangedBroadcastMultipleConfigs) {
|
|
const string package1 = "pkg1", package2 = "pkg2", package3 = "pkg3";
|
|
const int64_t configId = 123;
|
|
const int32_t callingUid1 = 456, callingUid2 = 789, callingUid3 = 1111;
|
|
const vector<int64_t> metricIds1 = {100, 200, 999}, metricIds2 = {101}, metricIds3 = {202, 101};
|
|
|
|
shared_ptr<MockPendingIntentRef> pir1 = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir1, sendRestrictedMetricsChangedBroadcast(metricIds1))
|
|
.Times(3)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
EXPECT_CALL(*pir1, sendRestrictedMetricsChangedBroadcast(metricIds2))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
shared_ptr<MockPendingIntentRef> pir2 = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir2, sendRestrictedMetricsChangedBroadcast(metricIds1))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
shared_ptr<MockPendingIntentRef> pir3 = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir3, sendRestrictedMetricsChangedBroadcast(metricIds1))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
EXPECT_CALL(*pir3, sendRestrictedMetricsChangedBroadcast(metricIds2))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
shared_ptr<MockPendingIntentRef> pir4 = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir4, sendRestrictedMetricsChangedBroadcast(metricIds1))
|
|
.Times(2)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
shared_ptr<MockPendingIntentRef> pir5 = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir5, sendRestrictedMetricsChangedBroadcast(metricIds3))
|
|
.Times(1)
|
|
.WillRepeatedly(Invoke([](const vector<int64_t>&) { return Status::ok(); }));
|
|
|
|
sp<ConfigManager> manager = new ConfigManager();
|
|
manager->SetRestrictedMetricsChangedReceiver(package1, configId, callingUid1, pir1);
|
|
manager->SetRestrictedMetricsChangedReceiver(package2, configId, callingUid2, pir2);
|
|
manager->SetRestrictedMetricsChangedReceiver(package1, configId, callingUid2, pir3);
|
|
manager->SetRestrictedMetricsChangedReceiver(package2, configId, callingUid1, pir4);
|
|
manager->SetRestrictedMetricsChangedReceiver(package3, configId, callingUid3, pir5);
|
|
|
|
// Invoke pir 1, 2, 3, 4.
|
|
manager->SendRestrictedMetricsBroadcast({package1, package2}, configId,
|
|
{callingUid1, callingUid2}, metricIds1);
|
|
// Invoke pir 1 only
|
|
manager->SendRestrictedMetricsBroadcast({package1}, configId, {callingUid1}, metricIds1);
|
|
// Invoke pir 1, 4.
|
|
manager->SendRestrictedMetricsBroadcast({package1, package2}, configId, {callingUid1},
|
|
metricIds1);
|
|
// Invoke pir 1, 3
|
|
manager->SendRestrictedMetricsBroadcast({package1}, configId, {callingUid1, callingUid2},
|
|
metricIds2);
|
|
// Invoke pir 5
|
|
manager->SendRestrictedMetricsBroadcast({package3}, configId, {callingUid1, callingUid3},
|
|
metricIds3);
|
|
}
|
|
|
|
TEST(ConfigManagerTest, TestRemoveRestrictedMetricsChangedBroadcast) {
|
|
const string configPackage = "pkg";
|
|
const int64_t configId = 123;
|
|
const int32_t callingUid = 456;
|
|
const vector<int64_t> metricIds = {100, 200, 999};
|
|
|
|
shared_ptr<MockPendingIntentRef> pir = SharedRefBase::make<StrictMock<MockPendingIntentRef>>();
|
|
EXPECT_CALL(*pir, sendRestrictedMetricsChangedBroadcast(metricIds)).Times(0);
|
|
|
|
sp<ConfigManager> manager = new ConfigManager();
|
|
manager->SetRestrictedMetricsChangedReceiver(configPackage, configId, callingUid, pir);
|
|
manager->RemoveRestrictedMetricsChangedReceiver(configPackage, configId, callingUid);
|
|
|
|
manager->SendRestrictedMetricsBroadcast({configPackage}, configId, {callingUid}, metricIds);
|
|
} |