| // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #define __STDC_FORMAT_MACROS |
| |
| #include <inttypes.h> |
| #include <utime.h> |
| |
| #include <string> |
| #include <vector> |
| |
| #include <base/at_exit.h> |
| #include <base/files/file_util.h> |
| #include <base/strings/string_number_conversions.h> |
| #include <base/strings/stringprintf.h> |
| #include <chromeos/dbus/service_constants.h> |
| #include <gtest/gtest.h> |
| |
| #include "metrics/metrics_daemon.h" |
| #include "metrics/metrics_library_mock.h" |
| #include "metrics/persistent_integer_mock.h" |
| |
| using base::FilePath; |
| using base::StringPrintf; |
| using base::Time; |
| using base::TimeDelta; |
| using base::TimeTicks; |
| using std::string; |
| using std::vector; |
| using ::testing::_; |
| using ::testing::AtLeast; |
| using ::testing::Return; |
| using ::testing::StrictMock; |
| using chromeos_metrics::PersistentIntegerMock; |
| |
| static const char kFakeDiskStatsName[] = "fake-disk-stats"; |
| static const char kFakeDiskStatsFormat[] = |
| " 1793 1788 %" PRIu64 " 105580 " |
| " 196 175 %" PRIu64 " 30290 " |
| " 0 44060 135850\n"; |
| static const uint64_t kFakeReadSectors[] = {80000, 100000}; |
| static const uint64_t kFakeWriteSectors[] = {3000, 4000}; |
| |
| static const char kFakeVmStatsName[] = "fake-vm-stats"; |
| static const char kFakeScalingMaxFreqPath[] = "fake-scaling-max-freq"; |
| static const char kFakeCpuinfoMaxFreqPath[] = "fake-cpuinfo-max-freq"; |
| static const char kMetricsServer[] = "https://clients4.google.com/uma/v2"; |
| static const char kMetricsFilePath[] = "/var/run/metrics/uma-events"; |
| |
| class MetricsDaemonTest : public testing::Test { |
| protected: |
| std::string kFakeDiskStats0; |
| std::string kFakeDiskStats1; |
| |
| virtual void SetUp() { |
| kFakeDiskStats0 = base::StringPrintf(kFakeDiskStatsFormat, |
| kFakeReadSectors[0], |
| kFakeWriteSectors[0]); |
| kFakeDiskStats1 = base::StringPrintf(kFakeDiskStatsFormat, |
| kFakeReadSectors[1], |
| kFakeWriteSectors[1]); |
| CreateFakeDiskStatsFile(kFakeDiskStats0.c_str()); |
| CreateUint64ValueFile(base::FilePath(kFakeCpuinfoMaxFreqPath), 10000000); |
| CreateUint64ValueFile(base::FilePath(kFakeScalingMaxFreqPath), 10000000); |
| |
| chromeos_metrics::PersistentInteger::SetTestingMode(true); |
| daemon_.Init(true, |
| false, |
| &metrics_lib_, |
| kFakeDiskStatsName, |
| kFakeVmStatsName, |
| kFakeScalingMaxFreqPath, |
| kFakeCpuinfoMaxFreqPath, |
| base::TimeDelta::FromMinutes(30), |
| kMetricsServer, |
| kMetricsFilePath, |
| "/"); |
| |
| // Replace original persistent values with mock ones. |
| daily_active_use_mock_ = |
| new StrictMock<PersistentIntegerMock>("1.mock"); |
| daemon_.daily_active_use_.reset(daily_active_use_mock_); |
| |
| kernel_crash_interval_mock_ = |
| new StrictMock<PersistentIntegerMock>("2.mock"); |
| daemon_.kernel_crash_interval_.reset(kernel_crash_interval_mock_); |
| |
| user_crash_interval_mock_ = |
| new StrictMock<PersistentIntegerMock>("3.mock"); |
| daemon_.user_crash_interval_.reset(user_crash_interval_mock_); |
| |
| unclean_shutdown_interval_mock_ = |
| new StrictMock<PersistentIntegerMock>("4.mock"); |
| daemon_.unclean_shutdown_interval_.reset(unclean_shutdown_interval_mock_); |
| } |
| |
| virtual void TearDown() { |
| EXPECT_EQ(0, unlink(kFakeDiskStatsName)); |
| EXPECT_EQ(0, unlink(kFakeScalingMaxFreqPath)); |
| EXPECT_EQ(0, unlink(kFakeCpuinfoMaxFreqPath)); |
| } |
| |
| // Adds active use aggregation counters update expectations that the |
| // specified count will be added. |
| void ExpectActiveUseUpdate(int count) { |
| EXPECT_CALL(*daily_active_use_mock_, Add(count)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| EXPECT_CALL(*kernel_crash_interval_mock_, Add(count)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| EXPECT_CALL(*user_crash_interval_mock_, Add(count)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| } |
| |
| // As above, but ignore values of counter updates. |
| void IgnoreActiveUseUpdate() { |
| EXPECT_CALL(*daily_active_use_mock_, Add(_)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| EXPECT_CALL(*kernel_crash_interval_mock_, Add(_)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| EXPECT_CALL(*user_crash_interval_mock_, Add(_)) |
| .Times(1) |
| .RetiresOnSaturation(); |
| } |
| |
| // Adds a metrics library mock expectation that the specified metric |
| // will be generated. |
| void ExpectSample(const std::string& name, int sample) { |
| EXPECT_CALL(metrics_lib_, SendToUMA(name, sample, _, _, _)) |
| .Times(1) |
| .WillOnce(Return(true)) |
| .RetiresOnSaturation(); |
| } |
| |
| // Creates a new DBus signal message with zero or more string arguments. |
| // The message can be deallocated through DeleteDBusMessage. |
| // |
| // |path| is the object emitting the signal. |
| // |interface| is the interface the signal is emitted from. |
| // |name| is the name of the signal. |
| // |arg_values| contains the values of the string arguments. |
| DBusMessage* NewDBusSignalString(const string& path, |
| const string& interface, |
| const string& name, |
| const vector<string>& arg_values) { |
| DBusMessage* msg = dbus_message_new_signal(path.c_str(), |
| interface.c_str(), |
| name.c_str()); |
| DBusMessageIter iter; |
| dbus_message_iter_init_append(msg, &iter); |
| for (vector<string>::const_iterator it = arg_values.begin(); |
| it != arg_values.end(); ++it) { |
| const char* str_value = it->c_str(); |
| dbus_message_iter_append_basic(&iter, DBUS_TYPE_STRING, &str_value); |
| } |
| return msg; |
| } |
| |
| // Deallocates the DBus message |msg| previously allocated through |
| // dbus_message_new*. |
| void DeleteDBusMessage(DBusMessage* msg) { |
| dbus_message_unref(msg); |
| } |
| |
| // Creates or overwrites an input file containing fake disk stats. |
| void CreateFakeDiskStatsFile(const char* fake_stats) { |
| if (unlink(kFakeDiskStatsName) < 0) { |
| EXPECT_EQ(errno, ENOENT); |
| } |
| FILE* f = fopen(kFakeDiskStatsName, "w"); |
| EXPECT_EQ(1, fwrite(fake_stats, strlen(fake_stats), 1, f)); |
| EXPECT_EQ(0, fclose(f)); |
| } |
| |
| // Creates or overwrites the file in |path| so that it contains the printable |
| // representation of |value|. |
| void CreateUint64ValueFile(const base::FilePath& path, uint64_t value) { |
| base::DeleteFile(path, false); |
| std::string value_string = base::Uint64ToString(value); |
| ASSERT_EQ(value_string.length(), |
| base::WriteFile(path, value_string.c_str(), |
| value_string.length())); |
| } |
| |
| // The MetricsDaemon under test. |
| MetricsDaemon daemon_; |
| |
| // Mocks. They are strict mock so that all unexpected |
| // calls are marked as failures. |
| StrictMock<MetricsLibraryMock> metrics_lib_; |
| StrictMock<PersistentIntegerMock>* daily_active_use_mock_; |
| StrictMock<PersistentIntegerMock>* kernel_crash_interval_mock_; |
| StrictMock<PersistentIntegerMock>* user_crash_interval_mock_; |
| StrictMock<PersistentIntegerMock>* unclean_shutdown_interval_mock_; |
| }; |
| |
| TEST_F(MetricsDaemonTest, CheckSystemCrash) { |
| static const char kKernelCrashDetected[] = "test-kernel-crash-detected"; |
| EXPECT_FALSE(daemon_.CheckSystemCrash(kKernelCrashDetected)); |
| |
| base::FilePath crash_detected(kKernelCrashDetected); |
| base::WriteFile(crash_detected, "", 0); |
| EXPECT_TRUE(base::PathExists(crash_detected)); |
| EXPECT_TRUE(daemon_.CheckSystemCrash(kKernelCrashDetected)); |
| EXPECT_FALSE(base::PathExists(crash_detected)); |
| EXPECT_FALSE(daemon_.CheckSystemCrash(kKernelCrashDetected)); |
| EXPECT_FALSE(base::PathExists(crash_detected)); |
| base::DeleteFile(crash_detected, false); |
| } |
| |
| TEST_F(MetricsDaemonTest, ReportDailyUse) { |
| ExpectSample("Logging.DailyUseTime", 2); |
| daemon_.ReportDailyUse(90); |
| |
| ExpectSample("Logging.DailyUseTime", 1); |
| daemon_.ReportDailyUse(89); |
| |
| // There should be no metrics generated for the calls below. |
| daemon_.ReportDailyUse(0); |
| daemon_.ReportDailyUse(-5); |
| } |
| |
| TEST_F(MetricsDaemonTest, MessageFilter) { |
| // Ignore calls to SendToUMA. |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, _, _, _, _)).Times(AtLeast(0)); |
| |
| DBusMessage* msg = dbus_message_new(DBUS_MESSAGE_TYPE_METHOD_CALL); |
| DBusHandlerResult res = |
| MetricsDaemon::MessageFilter(/* connection */ nullptr, msg, &daemon_); |
| EXPECT_EQ(DBUS_HANDLER_RESULT_NOT_YET_HANDLED, res); |
| DeleteDBusMessage(msg); |
| |
| IgnoreActiveUseUpdate(); |
| vector<string> signal_args; |
| msg = NewDBusSignalString("/", |
| "org.chromium.CrashReporter", |
| "UserCrash", |
| signal_args); |
| res = MetricsDaemon::MessageFilter(/* connection */ nullptr, msg, &daemon_); |
| EXPECT_EQ(DBUS_HANDLER_RESULT_HANDLED, res); |
| DeleteDBusMessage(msg); |
| |
| signal_args.clear(); |
| signal_args.push_back("randomstate"); |
| signal_args.push_back("bob"); // arbitrary username |
| msg = NewDBusSignalString("/", |
| "org.chromium.UnknownService.Manager", |
| "StateChanged", |
| signal_args); |
| res = MetricsDaemon::MessageFilter(/* connection */ nullptr, msg, &daemon_); |
| EXPECT_EQ(DBUS_HANDLER_RESULT_NOT_YET_HANDLED, res); |
| DeleteDBusMessage(msg); |
| } |
| |
| TEST_F(MetricsDaemonTest, SendSample) { |
| ExpectSample("Dummy.Metric", 3); |
| daemon_.SendSample("Dummy.Metric", /* sample */ 3, |
| /* min */ 1, /* max */ 100, /* buckets */ 50); |
| } |
| |
| TEST_F(MetricsDaemonTest, ReportDiskStats) { |
| uint64_t read_sectors_now, write_sectors_now; |
| |
| CreateFakeDiskStatsFile(kFakeDiskStats1.c_str()); |
| daemon_.DiskStatsReadStats(&read_sectors_now, &write_sectors_now); |
| EXPECT_EQ(read_sectors_now, kFakeReadSectors[1]); |
| EXPECT_EQ(write_sectors_now, kFakeWriteSectors[1]); |
| |
| MetricsDaemon::StatsState s_state = daemon_.stats_state_; |
| EXPECT_CALL(metrics_lib_, |
| SendToUMA(_, (kFakeReadSectors[1] - kFakeReadSectors[0]) / 30, |
| _, _, _)); |
| EXPECT_CALL(metrics_lib_, |
| SendToUMA(_, (kFakeWriteSectors[1] - kFakeWriteSectors[0]) / 30, |
| _, _, _)); |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA(_, _, _)); // SendCpuThrottleMetrics |
| daemon_.StatsCallback(); |
| EXPECT_TRUE(s_state != daemon_.stats_state_); |
| } |
| |
| TEST_F(MetricsDaemonTest, ProcessMeminfo) { |
| string meminfo = |
| "MemTotal: 2000000 kB\nMemFree: 500000 kB\n" |
| "Buffers: 1000000 kB\nCached: 213652 kB\n" |
| "SwapCached: 0 kB\nActive: 133400 kB\n" |
| "Inactive: 183396 kB\nActive(anon): 92984 kB\n" |
| "Inactive(anon): 58860 kB\nActive(file): 40416 kB\n" |
| "Inactive(file): 124536 kB\nUnevictable: 0 kB\n" |
| "Mlocked: 0 kB\nSwapTotal: 0 kB\n" |
| "SwapFree: 0 kB\nDirty: 40 kB\n" |
| "Writeback: 0 kB\nAnonPages: 92652 kB\n" |
| "Mapped: 59716 kB\nShmem: 59196 kB\n" |
| "Slab: 16656 kB\nSReclaimable: 6132 kB\n" |
| "SUnreclaim: 10524 kB\nKernelStack: 1648 kB\n" |
| "PageTables: 2780 kB\nNFS_Unstable: 0 kB\n" |
| "Bounce: 0 kB\nWritebackTmp: 0 kB\n" |
| "CommitLimit: 970656 kB\nCommitted_AS: 1260528 kB\n" |
| "VmallocTotal: 122880 kB\nVmallocUsed: 12144 kB\n" |
| "VmallocChunk: 103824 kB\nDirectMap4k: 9636 kB\n" |
| "DirectMap2M: 1955840 kB\n"; |
| |
| // All enum calls must report percents. |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA(_, _, 100)).Times(AtLeast(1)); |
| // Check that MemFree is correctly computed at 25%. |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA("Platform.MeminfoMemFree", 25, 100)) |
| .Times(AtLeast(1)); |
| // Check that we call SendToUma at least once (log histogram). |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, _, _, _, _)) |
| .Times(AtLeast(1)); |
| // Make sure we don't report fields not in the list. |
| EXPECT_CALL(metrics_lib_, SendToUMA("Platform.MeminfoMlocked", _, _, _, _)) |
| .Times(0); |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA("Platform.MeminfoMlocked", _, _)) |
| .Times(0); |
| EXPECT_TRUE(daemon_.ProcessMeminfo(meminfo)); |
| } |
| |
| TEST_F(MetricsDaemonTest, ProcessMeminfo2) { |
| string meminfo = "MemTotal: 2000000 kB\nMemFree: 1000000 kB\n"; |
| // Not enough fields. |
| EXPECT_FALSE(daemon_.ProcessMeminfo(meminfo)); |
| } |
| |
| TEST_F(MetricsDaemonTest, ParseVmStats) { |
| static char kVmStats[] = "pswpin 1345\npswpout 8896\n" |
| "foo 100\nbar 200\npgmajfault 42\netcetc 300\n"; |
| struct MetricsDaemon::VmstatRecord stats; |
| EXPECT_TRUE(daemon_.VmStatsParseStats(kVmStats, &stats)); |
| EXPECT_EQ(stats.page_faults_, 42); |
| EXPECT_EQ(stats.swap_in_, 1345); |
| EXPECT_EQ(stats.swap_out_, 8896); |
| } |
| |
| TEST_F(MetricsDaemonTest, ReadFreqToInt) { |
| const int fake_scaled_freq = 1666999; |
| const int fake_max_freq = 2000000; |
| int scaled_freq = 0; |
| int max_freq = 0; |
| CreateUint64ValueFile(base::FilePath(kFakeScalingMaxFreqPath), |
| fake_scaled_freq); |
| CreateUint64ValueFile(base::FilePath(kFakeCpuinfoMaxFreqPath), fake_max_freq); |
| EXPECT_TRUE(daemon_.testing_); |
| EXPECT_TRUE(daemon_.ReadFreqToInt(kFakeScalingMaxFreqPath, &scaled_freq)); |
| EXPECT_TRUE(daemon_.ReadFreqToInt(kFakeCpuinfoMaxFreqPath, &max_freq)); |
| EXPECT_EQ(fake_scaled_freq, scaled_freq); |
| EXPECT_EQ(fake_max_freq, max_freq); |
| } |
| |
| TEST_F(MetricsDaemonTest, SendCpuThrottleMetrics) { |
| CreateUint64ValueFile(base::FilePath(kFakeCpuinfoMaxFreqPath), 2001000); |
| // Test the 101% and 100% cases. |
| CreateUint64ValueFile(base::FilePath(kFakeScalingMaxFreqPath), 2001000); |
| EXPECT_TRUE(daemon_.testing_); |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA(_, 101, 101)); |
| daemon_.SendCpuThrottleMetrics(); |
| CreateUint64ValueFile(base::FilePath(kFakeScalingMaxFreqPath), 2000000); |
| EXPECT_CALL(metrics_lib_, SendEnumToUMA(_, 100, 101)); |
| daemon_.SendCpuThrottleMetrics(); |
| } |
| |
| TEST_F(MetricsDaemonTest, SendZramMetrics) { |
| EXPECT_TRUE(daemon_.testing_); |
| |
| // |compr_data_size| is the size in bytes of compressed data. |
| const uint64_t compr_data_size = 50 * 1000 * 1000; |
| // The constant '3' is a realistic but random choice. |
| // |orig_data_size| does not include zero pages. |
| const uint64_t orig_data_size = compr_data_size * 3; |
| const uint64_t page_size = 4096; |
| const uint64_t zero_pages = 10 * 1000 * 1000 / page_size; |
| |
| CreateUint64ValueFile(base::FilePath(MetricsDaemon::kComprDataSizeName), |
| compr_data_size); |
| CreateUint64ValueFile(base::FilePath(MetricsDaemon::kOrigDataSizeName), |
| orig_data_size); |
| CreateUint64ValueFile(base::FilePath(MetricsDaemon::kZeroPagesName), |
| zero_pages); |
| |
| const uint64_t real_orig_size = orig_data_size + zero_pages * page_size; |
| const uint64_t zero_ratio_percent = |
| zero_pages * page_size * 100 / real_orig_size; |
| // Ratio samples are in percents. |
| const uint64_t actual_ratio_sample = real_orig_size * 100 / compr_data_size; |
| |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, compr_data_size >> 20, _, _, _)); |
| EXPECT_CALL(metrics_lib_, |
| SendToUMA(_, (real_orig_size - compr_data_size) >> 20, _, _, _)); |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, actual_ratio_sample, _, _, _)); |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, zero_pages, _, _, _)); |
| EXPECT_CALL(metrics_lib_, SendToUMA(_, zero_ratio_percent, _, _, _)); |
| |
| EXPECT_TRUE(daemon_.ReportZram(base::FilePath("."))); |
| } |
| |
| int main(int argc, char** argv) { |
| testing::InitGoogleTest(&argc, argv); |
| // Some libchrome calls need this. |
| base::AtExitManager at_exit_manager; |
| |
| return RUN_ALL_TESTS(); |
| } |
