| Index: chrome/browser/chromeos/policy/device_status_collector_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/policy/device_status_collector_browsertest.cc b/chrome/browser/chromeos/policy/device_status_collector_browsertest.cc
|
| index 38e3478e97323246d47c21e758bab7229e1b256b..bd321d3f238506d339f722d31bff4c8c444af0b7 100644
|
| --- a/chrome/browser/chromeos/policy/device_status_collector_browsertest.cc
|
| +++ b/chrome/browser/chromeos/policy/device_status_collector_browsertest.cc
|
| @@ -125,9 +125,14 @@ class TestingDeviceStatusCollector : public policy::DeviceStatusCollector {
|
|
|
| mock_cpu_usage_ = usage;
|
|
|
| - // Refresh our samples.
|
| - for (int i = 0; i < static_cast<int>(kMaxCPUSamples); ++i)
|
| - SampleCPUUsage();
|
| + RefreshSampleResourceUsage();
|
| + }
|
| +
|
| + void RefreshSampleResourceUsage() {
|
| + // Refresh our samples. Sample more than kMaxHardwareSamples times to
|
| + // make sure that the code correctly caps the number of cached samples.
|
| + for (int i = 0; i < static_cast<int>(kMaxResourceUsageSamples + 1); ++i)
|
| + SampleResourceUsage();
|
| }
|
|
|
| protected:
|
| @@ -790,8 +795,10 @@ TEST_F(DeviceStatusCollectorTest, TestVolumeInfo) {
|
| }
|
|
|
| TEST_F(DeviceStatusCollectorTest, TestAvailableMemory) {
|
| + status_collector_->RefreshSampleResourceUsage();
|
| GetStatus();
|
| - EXPECT_TRUE(status_.has_system_ram_free());
|
| + EXPECT_EQ(static_cast<int>(DeviceStatusCollector::kMaxResourceUsageSamples),
|
| + status_.system_ram_free().size());
|
| EXPECT_TRUE(status_.has_system_ram_total());
|
| // No good way to inject specific test values for available system RAM, so
|
| // just make sure it's > 0.
|
| @@ -803,7 +810,7 @@ TEST_F(DeviceStatusCollectorTest, TestCPUSamples) {
|
| const int full_cpu_usage = 100;
|
| status_collector_->set_mock_cpu_usage(full_cpu_usage, 2);
|
| GetStatus();
|
| - EXPECT_EQ(static_cast<int>(DeviceStatusCollector::kMaxCPUSamples),
|
| + EXPECT_EQ(static_cast<int>(DeviceStatusCollector::kMaxResourceUsageSamples),
|
| status_.cpu_utilization_pct().size());
|
| for (const auto utilization : status_.cpu_utilization_pct())
|
| EXPECT_EQ(full_cpu_usage, utilization);
|
| @@ -812,7 +819,7 @@ TEST_F(DeviceStatusCollectorTest, TestCPUSamples) {
|
| const int idle_cpu_usage = 0;
|
| status_collector_->set_mock_cpu_usage(idle_cpu_usage, 2);
|
| GetStatus();
|
| - EXPECT_EQ(static_cast<int>(DeviceStatusCollector::kMaxCPUSamples),
|
| + EXPECT_EQ(static_cast<int>(DeviceStatusCollector::kMaxResourceUsageSamples),
|
| status_.cpu_utilization_pct().size());
|
| for (const auto utilization : status_.cpu_utilization_pct())
|
| EXPECT_EQ(idle_cpu_usage, utilization);
|
|
|