Aggregate device activity, and report per-day activity in device status reports.

chrome/browser/policy/device_status_collector_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/policy/device_status_collector.h"
 
 #include "base/message_loop.h"
 #include "base/time.h"
 #include "chrome/browser/idle.h"
 #include "chrome/browser/chromeos/cros_settings.h"
@@ skipping 14 matching lines @@
 namespace em = enterprise_management;
 
 namespace {
 
 class TestingDeviceStatusCollector : public policy::DeviceStatusCollector {
  public:
   TestingDeviceStatusCollector(
       PrefService* local_state,
       chromeos::system::StatisticsProvider* provider)
     :  policy::DeviceStatusCollector(local_state, provider),
-       local_state_(local_state),
-       baseline_time_(Time::Now()) {
+       local_state_(local_state) {
+    // Set the baseline time to a fixed value (1 AM) to prevent test flakiness
+    // due to a single activity period spanning two days.
+    SetBaselineTime(Time::Now().LocalMidnight() + TimeDelta::FromHours(1));
   }
 
   void Simulate(IdleState* states, int len) {
     for (int i = 0; i < len; i++)
       IdleStateCallback(states[i]);
   }
 
-  void SimulateWithSleep(IdleState* states, int len, int ) {
-    for (int i = 0; i < len; i++)
-      IdleStateCallback(states[i]);
+  void set_max_stored_past_activity_days(unsigned int value) {
+    max_stored_past_activity_days_ = value;
   }
 
-  void set_max_stored_active_periods(unsigned int value) {
-    max_stored_active_periods_ = value;
+  void set_max_stored_future_activity_days(unsigned int value) {
+    max_stored_future_activity_days_ = value;
+  }
+
+  // Reset the baseline time.
+  void SetBaselineTime(Time time) {
+    baseline_time_ = time;
+    baseline_offset_periods_ = 0;
   }
 
  protected:
   virtual void CheckIdleState() OVERRIDE {
     // This should never be called in testing, as it results in a dbus call.
     NOTREACHED();
   }
 
   // Each time this is called, returns a time that is a fixed increment
   // later than the previous time.
   virtual Time GetCurrentTime() OVERRIDE {
-    static int call_count = 0;
-    return baseline_time_ + TimeDelta::FromSeconds(
-        policy::DeviceStatusCollector::kPollIntervalSeconds * call_count++);
+    int poll_interval = policy::DeviceStatusCollector::kPollIntervalSeconds;
+    return baseline_time_ +
+        TimeDelta::FromSeconds(poll_interval * baseline_offset_periods_++);
   }
 
  private:
   PrefService* local_state_;
 
   // Baseline time for the fake times returned from GetCurrentTime().
-  // It doesn't really matter what this is, as long as it stays the same over
-  // the lifetime of the object.
   Time baseline_time_;
+
+  // The number of simulated periods since the baseline time.
+  int baseline_offset_periods_;
 };
 
 // Return the total number of active milliseconds contained in a device
 // status report.
 int64 GetActiveMilliseconds(em::DeviceStatusReportRequest& status) {
   int64 active_milliseconds = 0;
-  for (int i = 0; i < status.active_time_size(); i++) {
-    const em::TimePeriod& period = status.active_time(i);
-    active_milliseconds += period.end_timestamp() - period.start_timestamp();
+  for (int i = 0; i < status.active_period_size(); i++) {
+    active_milliseconds += status.active_period(i).active_duration();
   }
   return active_milliseconds;
 }
 
 }  // namespace
 
 namespace policy {
 
 using ::testing::_;
 using ::testing::NotNull;
@@ skipping 52 matching lines @@
 TEST_F(DeviceStatusCollectorTest, AllIdle) {
   IdleState test_states[] = {
     IDLE_STATE_IDLE,
     IDLE_STATE_IDLE,
     IDLE_STATE_IDLE
   };
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
 
   // Test reporting with no data.
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(0, status_.active_time_size());
+  EXPECT_EQ(0, status_.active_period_size());
   EXPECT_EQ(0, GetActiveMilliseconds(status_));
 
   // Test reporting with a single idle sample.
   status_collector_.Simulate(test_states, 1);
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(0, status_.active_time_size());
+  EXPECT_EQ(0, status_.active_period_size());
   EXPECT_EQ(0, GetActiveMilliseconds(status_));
 
   // Test reporting with multiple consecutive idle samples.
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(0, status_.active_time_size());
+  EXPECT_EQ(0, status_.active_period_size());
   EXPECT_EQ(0, GetActiveMilliseconds(status_));
 }
 
 TEST_F(DeviceStatusCollectorTest, AllActive) {
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE
   };
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
 
   // Test a single active sample.
   status_collector_.Simulate(test_states, 1);
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(1, status_.active_time_size());
+  EXPECT_EQ(1, status_.active_period_size());
   EXPECT_EQ(1 * ActivePeriodMilliseconds(), GetActiveMilliseconds(status_));
-  status_.clear_active_time(); // Clear the result protobuf.
+  status_.clear_active_period(); // Clear the result protobuf.
 
-  // Test multiple consecutive active samples -- they should be coalesced
-  // into a single active period.
+  // Test multiple consecutive active samples.
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(1, status_.active_time_size());
+  EXPECT_EQ(1, status_.active_period_size());
   EXPECT_EQ(3 * ActivePeriodMilliseconds(), GetActiveMilliseconds(status_));
 }
 
 TEST_F(DeviceStatusCollectorTest, MixedStates) {
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_IDLE,
     IDLE_STATE_ACTIVE
   };
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(3, status_.active_time_size());
   EXPECT_EQ(4 * ActivePeriodMilliseconds(), GetActiveMilliseconds(status_));
 }
 
 TEST_F(DeviceStatusCollectorTest, StateKeptInPref) {
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_IDLE
   };
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
 
   // Process the list a second time with a different collector.
   // It should be able to count the active periods found by the first
   // collector, because the results are stored in a pref.
   TestingDeviceStatusCollector second_collector(&prefs_,
                                                 &statistics_provider_);
   second_collector.Simulate(test_states,
                             sizeof(test_states) / sizeof(IdleState));
 
   second_collector.GetStatus(&status_);
-  EXPECT_EQ(4, status_.active_time_size());
   EXPECT_EQ(6 * ActivePeriodMilliseconds(), GetActiveMilliseconds(status_));
 }
 
 TEST_F(DeviceStatusCollectorTest, Times) {
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE,
     IDLE_STATE_IDLE
   };
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(2, status_.active_time_size());
-
   EXPECT_EQ(3 * ActivePeriodMilliseconds(), GetActiveMilliseconds(status_));
 }
 
 TEST_F(DeviceStatusCollectorTest, MaxStoredPeriods) {
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_IDLE
   };
-  unsigned int max_periods = 10;
+  unsigned int max_days = 10;
 
   cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
-  status_collector_.set_max_stored_active_periods(max_periods);
+  status_collector_.set_max_stored_past_activity_days(max_days - 1);
+  status_collector_.set_max_stored_future_activity_days(1);
+  Time baseline = Time::Now().LocalMidnight();
 
   // Simulate 12 active periods.
-  for (int i = 0; i < 12; i++) {
+  for (int i = 0; i < static_cast<int>(max_days) + 2; i++) {
     status_collector_.Simulate(test_states,
                                sizeof(test_states) / sizeof(IdleState));
+    // Advance the simulated clock by a day.
+    baseline += TimeDelta::FromDays(1);
+    status_collector_.SetBaselineTime(baseline);
   }
 
   // Check that we don't exceed the max number of periods.
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(static_cast<int>(max_periods), status_.active_time_size());
+  EXPECT_EQ(static_cast<int>(max_days), status_.active_period_size());
+
+  // Simulate some future times.
+  for (int i = 0; i < static_cast<int>(max_days) + 2; i++) {
+    status_collector_.Simulate(test_states,
+                               sizeof(test_states) / sizeof(IdleState));
+    // Advance the simulated clock by a day.
+    baseline += TimeDelta::FromDays(1);
+    status_collector_.SetBaselineTime(baseline);
+  }
+  // Set the clock back so the previous simulated times are in the future.
+  baseline -= TimeDelta::FromDays(20);
+  status_collector_.SetBaselineTime(baseline);
+
+  // Collect one more data point to trigger pruning.
+  status_collector_.Simulate(test_states, 1);
+
+  // Check that we don't exceed the max number of periods.
+  status_.clear_active_period();
+  status_collector_.GetStatus(&status_);
+  EXPECT_LT(status_.active_period_size(), static_cast<int>(max_days));
 }
 
 TEST_F(DeviceStatusCollectorTest, ActivityTimesDisabledByDefault) {
   // If the pref for collecting device activity times isn't explicitly turned
   // on, no data on activity times should be reported.
 
   IdleState test_states[] = {
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE,
     IDLE_STATE_ACTIVE
   };
   status_collector_.Simulate(test_states,
                              sizeof(test_states) / sizeof(IdleState));
   status_collector_.GetStatus(&status_);
-  EXPECT_EQ(0, status_.active_time_size());
+  EXPECT_EQ(0, status_.active_period_size());
   EXPECT_EQ(0, GetActiveMilliseconds(status_));
 }
 
+TEST_F(DeviceStatusCollectorTest, ActivityCrossingMidnight) {
+  IdleState test_states[] = {
+    IDLE_STATE_ACTIVE
+  };
+  cros_settings_->SetBoolean(chromeos::kReportDeviceActivityTimes, true);
+  status_collector_.SetBaselineTime(
+      Time::Now().LocalMidnight() + TimeDelta::FromSeconds(10));
+
+  status_collector_.Simulate(test_states, 1);
+  status_collector_.GetStatus(&status_);
+  EXPECT_EQ(2, status_.active_period_size());

[Mattias Nissler (ping if slow), 2012/02/15 14:56:21]

We should check that we get correct times for both of the days.

[Patrick Dubroy, 2012/02/15 15:37:25]

Done.

+}
+
 TEST_F(DeviceStatusCollectorTest, DevSwitchBootMode) {
   // Test that boot mode data is not reported if the pref is not turned on.
   status_collector_.GetStatus(&status_);
   EXPECT_EQ(false, status_.has_boot_mode());
 
   EXPECT_CALL(statistics_provider_,
               GetMachineStatistic("devsw_boot", NotNull()))
       .WillRepeatedly(DoAll(SetArgPointee<1>("0"), Return(true)));
   EXPECT_EQ(false, status_.has_boot_mode());
 
@@ skipping 40 matching lines @@
   EXPECT_EQ(true, status_.has_os_version());
   EXPECT_EQ(true, status_.has_firmware_version());
 
   // Check that the browser version is not empty. OS version & firmware
   // don't have any reasonable values inside the unit test, so those
   // aren't checked.
   EXPECT_NE("", status_.browser_version());
 }
 
 }  // namespace policy