516 lines
23 KiB
C++
516 lines
23 KiB
C++
|
/*
|
||
|
|
* Copyright 2021 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 "MockPackageInfoResolver.h"
|
||
|
|
#include "MockUidCpuStatsCollector.h"
|
||
|
|
#include "MockUidIoStatsCollector.h"
|
||
|
|
#include "MockUidProcStatsCollector.h"
|
||
|
|
#include "PackageInfoTestUtils.h"
|
||
|
|
#include "UidIoStatsCollector.h"
|
||
|
|
#include "UidProcStatsCollector.h"
|
||
|
|
#include "UidProcStatsCollectorTestUtils.h"
|
||
|
|
#include "UidStatsCollector.h"
|
||
|
|
|
||
|
|
#include <android-base/stringprintf.h>
|
||
|
|
#include <gmock/gmock.h>
|
||
|
|
#include <utils/RefBase.h>
|
||
|
|
|
||
|
|
#include <inttypes.h>
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
namespace android {
|
||
|
|
namespace automotive {
|
||
|
|
namespace watchdog {
|
||
|
|
|
||
|
|
using ::android::automotive::watchdog::internal::PackageInfo;
|
||
|
|
using ::android::automotive::watchdog::internal::UidType;
|
||
|
|
using ::android::base::Error;
|
||
|
|
using ::android::base::Result;
|
||
|
|
using ::android::base::StringAppendF;
|
||
|
|
using ::android::base::StringPrintf;
|
||
|
|
using ::testing::AllOf;
|
||
|
|
using ::testing::Eq;
|
||
|
|
using ::testing::ExplainMatchResult;
|
||
|
|
using ::testing::Field;
|
||
|
|
using ::testing::IsEmpty;
|
||
|
|
using ::testing::Matcher;
|
||
|
|
using ::testing::Return;
|
||
|
|
using ::testing::UnorderedElementsAre;
|
||
|
|
using ::testing::UnorderedElementsAreArray;
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
std::string toString(const UidStats& uidStats) {
|
||
|
|
return StringPrintf("UidStats{packageInfo: %s, cpuTimeMillis: %" PRId64
|
||
|
|
", ioStats: %s, procStats: %s}",
|
||
|
|
uidStats.packageInfo.toString().c_str(), uidStats.cpuTimeMillis,
|
||
|
|
uidStats.ioStats.toString().c_str(), uidStats.procStats.toString().c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string toString(const std::vector<UidStats>& uidStats) {
|
||
|
|
std::string buffer;
|
||
|
|
StringAppendF(&buffer, "{");
|
||
|
|
for (const auto& stats : uidStats) {
|
||
|
|
StringAppendF(&buffer, "%s\n", toString(stats).c_str());
|
||
|
|
}
|
||
|
|
StringAppendF(&buffer, "}");
|
||
|
|
return buffer;
|
||
|
|
}
|
||
|
|
|
||
|
|
MATCHER_P(UidStatsEq, expected, "") {
|
||
|
|
return ExplainMatchResult(AllOf(Field("packageInfo", &UidStats::packageInfo,
|
||
|
|
PackageInfoEq(expected.packageInfo)),
|
||
|
|
Field("cpuTimeMillis", &UidStats::cpuTimeMillis,
|
||
|
|
Eq(expected.cpuTimeMillis)),
|
||
|
|
Field("ioStats", &UidStats::ioStats, Eq(expected.ioStats)),
|
||
|
|
Field("procStats", &UidStats::procStats,
|
||
|
|
UidProcStatsEq(expected.procStats))),
|
||
|
|
arg, result_listener);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<Matcher<const UidStats&>> UidStatsMatchers(const std::vector<UidStats>& uidStats) {
|
||
|
|
std::vector<Matcher<const UidStats&>> matchers;
|
||
|
|
for (const auto& stats : uidStats) {
|
||
|
|
matchers.push_back(UidStatsEq(stats));
|
||
|
|
}
|
||
|
|
return matchers;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::unordered_map<uid_t, PackageInfo> samplePackageInfoByUid() {
|
||
|
|
return {{1001234, constructPackageInfo("system.daemon", 1001234, UidType::NATIVE)},
|
||
|
|
{1005678, constructPackageInfo("kitchensink.app", 1005678, UidType::APPLICATION)}};
|
||
|
|
}
|
||
|
|
|
||
|
|
std::unordered_map<uid_t, UidIoStats> sampleUidIoStatsByUid() {
|
||
|
|
return {{1001234,
|
||
|
|
UidIoStats{/*fgRdBytes=*/3'000, /*bgRdBytes=*/0,
|
||
|
|
/*fgWrBytes=*/500,
|
||
|
|
/*bgWrBytes=*/0, /*fgFsync=*/20,
|
||
|
|
/*bgFsync=*/0}},
|
||
|
|
{1005678,
|
||
|
|
UidIoStats{/*fgRdBytes=*/30, /*bgRdBytes=*/100,
|
||
|
|
/*fgWrBytes=*/50, /*bgWrBytes=*/200,
|
||
|
|
/*fgFsync=*/45, /*bgFsync=*/60}}};
|
||
|
|
}
|
||
|
|
|
||
|
|
std::unordered_map<uid_t, UidProcStats> sampleUidProcStatsByUid() {
|
||
|
|
return {{1001234,
|
||
|
|
UidProcStats{.cpuTimeMillis = 10,
|
||
|
|
.totalMajorFaults = 220,
|
||
|
|
.totalTasksCount = 2,
|
||
|
|
.ioBlockedTasksCount = 1,
|
||
|
|
.processStatsByPid = {{1, {"init", 0, 10, 220, 2, 1}}}}},
|
||
|
|
{1005678,
|
||
|
|
UidProcStats{.cpuTimeMillis = 43,
|
||
|
|
.totalMajorFaults = 600,
|
||
|
|
.totalTasksCount = 2,
|
||
|
|
.ioBlockedTasksCount = 2,
|
||
|
|
.processStatsByPid = {
|
||
|
|
{1000, {"system_server", 13'400, 43, 600, 2, 2}}}}}};
|
||
|
|
}
|
||
|
|
|
||
|
|
std::unordered_map<uid_t, int64_t> sampleUidCpuStatsByUid() {
|
||
|
|
return {{1001234, 15}, {1005678, 43}};
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<UidStats> sampleUidStats() {
|
||
|
|
return {{.packageInfo = constructPackageInfo("system.daemon", 1001234, UidType::NATIVE),
|
||
|
|
.cpuTimeMillis = 15,
|
||
|
|
.ioStats = UidIoStats{/*fgRdBytes=*/3'000, /*bgRdBytes=*/0, /*fgWrBytes=*/500,
|
||
|
|
/*bgWrBytes=*/0, /*fgFsync=*/20, /*bgFsync=*/0},
|
||
|
|
.procStats = UidProcStats{.cpuTimeMillis = 10,
|
||
|
|
.totalMajorFaults = 220,
|
||
|
|
.totalTasksCount = 2,
|
||
|
|
.ioBlockedTasksCount = 1,
|
||
|
|
.processStatsByPid = {{1, {"init", 0, 10, 220, 2, 1}}}}},
|
||
|
|
{.packageInfo = constructPackageInfo("kitchensink.app", 1005678, UidType::APPLICATION),
|
||
|
|
.cpuTimeMillis = 43,
|
||
|
|
.ioStats = UidIoStats{/*fgRdBytes=*/30, /*bgRdBytes=*/100, /*fgWrBytes=*/50,
|
||
|
|
/*bgWrBytes=*/200,
|
||
|
|
/*fgFsync=*/45, /*bgFsync=*/60},
|
||
|
|
.procStats = UidProcStats{.cpuTimeMillis = 43,
|
||
|
|
.totalMajorFaults = 600,
|
||
|
|
.totalTasksCount = 2,
|
||
|
|
.ioBlockedTasksCount = 2,
|
||
|
|
.processStatsByPid = {
|
||
|
|
{1000, {"system_server", 13'400, 43, 600, 2, 2}}}}}};
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
namespace internal {
|
||
|
|
|
||
|
|
class UidStatsCollectorPeer final : public RefBase {
|
||
|
|
public:
|
||
|
|
explicit UidStatsCollectorPeer(sp<UidStatsCollector> collector) : mCollector(collector) {}
|
||
|
|
|
||
|
|
void setPackageInfoResolver(const sp<PackageInfoResolverInterface>& packageInfoResolver) {
|
||
|
|
mCollector->mPackageInfoResolver = packageInfoResolver;
|
||
|
|
}
|
||
|
|
|
||
|
|
void setUidIoStatsCollector(const sp<UidIoStatsCollectorInterface>& uidIoStatsCollector) {
|
||
|
|
mCollector->mUidIoStatsCollector = uidIoStatsCollector;
|
||
|
|
}
|
||
|
|
|
||
|
|
void setUidProcStatsCollector(const sp<UidProcStatsCollectorInterface>& uidProcStatsCollector) {
|
||
|
|
mCollector->mUidProcStatsCollector = uidProcStatsCollector;
|
||
|
|
}
|
||
|
|
|
||
|
|
void setUidCpuStatsCollector(const sp<UidCpuStatsCollectorInterface>& uidCpuStatsCollector) {
|
||
|
|
mCollector->mUidCpuStatsCollector = uidCpuStatsCollector;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
sp<UidStatsCollector> mCollector;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace internal
|
||
|
|
|
||
|
|
class UidStatsCollectorTest : public ::testing::Test {
|
||
|
|
protected:
|
||
|
|
virtual void SetUp() {
|
||
|
|
mUidStatsCollector = sp<UidStatsCollector>::make();
|
||
|
|
mUidStatsCollectorPeer = sp<internal::UidStatsCollectorPeer>::make(mUidStatsCollector);
|
||
|
|
mMockPackageInfoResolver = sp<MockPackageInfoResolver>::make();
|
||
|
|
mMockUidIoStatsCollector = sp<MockUidIoStatsCollector>::make();
|
||
|
|
mMockUidProcStatsCollector = sp<MockUidProcStatsCollector>::make();
|
||
|
|
mMockUidCpuStatsCollector = sp<MockUidCpuStatsCollector>::make();
|
||
|
|
mUidStatsCollectorPeer->setPackageInfoResolver(mMockPackageInfoResolver);
|
||
|
|
mUidStatsCollectorPeer->setUidIoStatsCollector(mMockUidIoStatsCollector);
|
||
|
|
mUidStatsCollectorPeer->setUidProcStatsCollector(mMockUidProcStatsCollector);
|
||
|
|
mUidStatsCollectorPeer->setUidCpuStatsCollector(mMockUidCpuStatsCollector);
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void TearDown() {
|
||
|
|
mUidStatsCollector.clear();
|
||
|
|
mUidStatsCollectorPeer.clear();
|
||
|
|
mMockPackageInfoResolver.clear();
|
||
|
|
mMockUidIoStatsCollector.clear();
|
||
|
|
mMockUidProcStatsCollector.clear();
|
||
|
|
mMockUidCpuStatsCollector.clear();
|
||
|
|
}
|
||
|
|
|
||
|
|
sp<UidStatsCollector> mUidStatsCollector;
|
||
|
|
sp<internal::UidStatsCollectorPeer> mUidStatsCollectorPeer;
|
||
|
|
sp<MockPackageInfoResolver> mMockPackageInfoResolver;
|
||
|
|
sp<MockUidIoStatsCollector> mMockUidIoStatsCollector;
|
||
|
|
sp<MockUidProcStatsCollector> mMockUidProcStatsCollector;
|
||
|
|
sp<MockUidCpuStatsCollector> mMockUidCpuStatsCollector;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestInit) {
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, init()).Times(1);
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, init()).Times(1);
|
||
|
|
|
||
|
|
mUidStatsCollector->init();
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollect) {
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, enabled()).WillOnce(Return(true));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, enabled()).WillOnce(Return(true));
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, collect()).WillOnce(Return(Result<void>()));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, collect()).WillOnce(Return(Result<void>()));
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, latestStats())
|
||
|
|
.WillOnce(Return(std::unordered_map<uid_t, UidIoStats>()));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, latestStats())
|
||
|
|
.WillOnce(Return(std::unordered_map<uid_t, UidProcStats>()));
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats())
|
||
|
|
.WillOnce(Return(std::unordered_map<uid_t, UidIoStats>()));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats())
|
||
|
|
.WillOnce(Return(std::unordered_map<uid_t, UidProcStats>()));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestFailsCollectOnUidIoStatsCollectorError) {
|
||
|
|
Result<void> errorResult = Error() << "Failed to collect per-UID I/O stats";
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, collect()).WillOnce(Return(errorResult));
|
||
|
|
|
||
|
|
ASSERT_FALSE(mUidStatsCollector->collect().ok())
|
||
|
|
<< "Must fail to collect when per-UID I/O stats collector fails";
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestFailsCollectOnUidProcStatsCollectorError) {
|
||
|
|
Result<void> errorResult = Error() << "Failed to collect per-UID proc stats";
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, collect()).WillOnce(Return(errorResult));
|
||
|
|
|
||
|
|
ASSERT_FALSE(mUidStatsCollector->collect().ok())
|
||
|
|
<< "Must fail to collect when per-UID proc stats collector fails";
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectLatestStats) {
|
||
|
|
const std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, int64_t> uidCpuStatsByUid = sampleUidCpuStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, latestStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, latestStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidCpuStatsCollector, latestStats()).WillOnce(Return(uidCpuStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
const std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Latest UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Delta UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectDeltaStats) {
|
||
|
|
const std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, int64_t> uidCpuStatsByUid = sampleUidCpuStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidCpuStatsCollector, deltaStats()).WillOnce(Return(uidCpuStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
const std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Delta UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Latest UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectDeltaStatsWithMissingUidIoStats) {
|
||
|
|
const std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
uidIoStatsByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, int64_t> uidCpuStatsByUid = sampleUidCpuStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidCpuStatsCollector, deltaStats()).WillOnce(Return(uidCpuStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
expected[0].ioStats = {};
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Delta UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Latest UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectDeltaStatsWithMissingUidProcStats) {
|
||
|
|
const std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
uidProcStatsByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, int64_t> uidCpuStatsByUid = sampleUidCpuStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidCpuStatsCollector, deltaStats()).WillOnce(Return(uidCpuStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
expected[0].procStats = {};
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Delta UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Latest UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectDeltaStatsWithMissingUidCpuStats) {
|
||
|
|
const std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
// Since no Uid CPU time was recovered from the /proc/uid_cputime/show_uid_stat file, the
|
||
|
|
// collector will default the UID CPU time to the UidProcStats' CPU time.
|
||
|
|
expected[0].cpuTimeMillis = 10;
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Delta UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Latest UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestCollectDeltaStatsWithMissingPackageInfo) {
|
||
|
|
std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
packageInfoByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, int64_t> uidCpuStatsByUid = sampleUidCpuStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidCpuStatsCollector, deltaStats()).WillOnce(Return(uidCpuStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
std::vector<UidStats> expected = sampleUidStats();
|
||
|
|
expected[0].packageInfo = constructPackageInfo("", 1001234);
|
||
|
|
|
||
|
|
auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, UnorderedElementsAreArray(UidStatsMatchers(expected)))
|
||
|
|
<< "Delta UID stats doesn't match.\nExpected: " << toString(expected)
|
||
|
|
<< "\nActual: " << toString(actual);
|
||
|
|
|
||
|
|
actual = mUidStatsCollector->latestStats();
|
||
|
|
|
||
|
|
EXPECT_THAT(actual, IsEmpty()) << "Latest UID stats isn't empty.\nActual: " << toString(actual);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestUidStatsHasPackageInfo) {
|
||
|
|
std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
packageInfoByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
const auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_EQ(actual.size(), static_cast<size_t>(2));
|
||
|
|
for (const auto stats : actual) {
|
||
|
|
if (stats.packageInfo.packageIdentifier.uid == 1001234) {
|
||
|
|
EXPECT_FALSE(stats.hasPackageInfo())
|
||
|
|
<< "Stats without package info should return false";
|
||
|
|
} else if (stats.packageInfo.packageIdentifier.uid == 1005678) {
|
||
|
|
EXPECT_TRUE(stats.hasPackageInfo()) << "Stats without package info should return true";
|
||
|
|
} else {
|
||
|
|
FAIL() << "Unexpected uid " << stats.packageInfo.packageIdentifier.uid;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestUidStatsGenericPackageName) {
|
||
|
|
std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
packageInfoByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
const auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
EXPECT_EQ(actual.size(), static_cast<size_t>(2));
|
||
|
|
for (const auto stats : actual) {
|
||
|
|
if (stats.packageInfo.packageIdentifier.uid == 1001234) {
|
||
|
|
EXPECT_EQ(stats.genericPackageName(), "1001234")
|
||
|
|
<< "Stats without package info should return UID as package name";
|
||
|
|
} else if (stats.packageInfo.packageIdentifier.uid == 1005678) {
|
||
|
|
EXPECT_EQ(stats.genericPackageName(), "kitchensink.app")
|
||
|
|
<< "Stats with package info should return corresponding package name";
|
||
|
|
} else {
|
||
|
|
FAIL() << "Unexpected uid " << stats.packageInfo.packageIdentifier.uid;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(UidStatsCollectorTest, TestUidStatsUid) {
|
||
|
|
std::unordered_map<uid_t, PackageInfo> packageInfoByUid = samplePackageInfoByUid();
|
||
|
|
packageInfoByUid.erase(1001234);
|
||
|
|
const std::unordered_map<uid_t, UidIoStats> uidIoStatsByUid = sampleUidIoStatsByUid();
|
||
|
|
const std::unordered_map<uid_t, UidProcStats> uidProcStatsByUid = sampleUidProcStatsByUid();
|
||
|
|
|
||
|
|
EXPECT_CALL(*mMockPackageInfoResolver,
|
||
|
|
getPackageInfosForUids(UnorderedElementsAre(1001234, 1005678)))
|
||
|
|
.WillOnce(Return(packageInfoByUid));
|
||
|
|
EXPECT_CALL(*mMockUidIoStatsCollector, deltaStats()).WillOnce(Return(uidIoStatsByUid));
|
||
|
|
EXPECT_CALL(*mMockUidProcStatsCollector, deltaStats()).WillOnce(Return(uidProcStatsByUid));
|
||
|
|
|
||
|
|
ASSERT_RESULT_OK(mUidStatsCollector->collect());
|
||
|
|
|
||
|
|
const auto actual = mUidStatsCollector->deltaStats();
|
||
|
|
|
||
|
|
for (const auto stats : actual) {
|
||
|
|
EXPECT_EQ(stats.uid(), static_cast<uid_t>(stats.packageInfo.packageIdentifier.uid));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace watchdog
|
||
|
|
} // namespace automotive
|
||
|
|
} // namespace android
|