Created multi-process-friendly PowerMonitor interface.

content/browser/power_monitor_message_broadcaster_unittest.cc

+// Copyright 2013 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 "base/power_monitor/power_monitor_test_base.h"
+#include "content/browser/power_monitor_message_broadcaster.h"
+#include "content/common/power_monitor_messages.h"
+#include "ipc/ipc_sender.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+class PowerMonitorMessageSender : public IPC::Sender {
+ public:
+  PowerMonitorMessageSender()
+      : power_state_changes_(0),
+        suspends_(0),
+        resumes_(0) {
+  }
+  virtual ~PowerMonitorMessageSender() {}
+
+  virtual bool Send(IPC::Message* msg) OVERRIDE {
+    switch (msg->type()) {
+      case PowerMonitorMsg_Suspend::ID:
+        suspends_++;
+        break;
+      case PowerMonitorMsg_Resume::ID:
+        resumes_++;
+        break;
+      case PowerMonitorMsg_PowerStateChange::ID:
+        power_state_changes_++;
+        break;
+    }
+    delete msg;
+    return true;
+  };
+
+  // Test status counts.
+  int power_state_changes() { return power_state_changes_; }
+  int suspends() { return suspends_; }
+  int resumes() { return resumes_; }
+
+ private:
+  int power_state_changes_;  // Count of OnPowerStateChange notifications.
+  int suspends_;  // Count of OnSuspend notifications.
+  int resumes_;  // Count of OnResume notifications.
+};
+
+class PowerMonitorMessageBroadcasterTest : public testing::Test {
+ protected:
+  PowerMonitorMessageBroadcasterTest() {
+    power_monitor_source_ = new base::PowerMonitorTestSource();
+    power_monitor_.reset(new base::PowerMonitor(
+        scoped_ptr<base::PowerMonitorSource>(power_monitor_source_)));
+  }
+  virtual ~PowerMonitorMessageBroadcasterTest() {};
+
+  base::PowerMonitorTestSource* source() { return power_monitor_source_; }
+  base::PowerMonitor* monitor() { return power_monitor_.get(); }
+
+ private:
+  base::PowerMonitorTestSource* power_monitor_source_;
+  scoped_ptr<base::PowerMonitor> power_monitor_;
+
+  DISALLOW_COPY_AND_ASSIGN(PowerMonitorMessageBroadcasterTest);
+};
+
+TEST_F(PowerMonitorMessageBroadcasterTest, PowerMessageBroadcast) {
+  PowerMonitorMessageSender sender;
+  PowerMonitorMessageBroadcaster broadcaster(&sender);
+
+  // Sending resume when not suspended should have no effect.
+  source()->GenerateResumeEvent();
+  EXPECT_EQ(sender.resumes(), 0);
+
+  // Pretend we suspended.
+  source()->GenerateSuspendEvent();
+  EXPECT_EQ(sender.suspends(), 1);
+
+  // Send a second suspend notification.  This should be suppressed.
+  source()->GenerateSuspendEvent();
+  EXPECT_EQ(sender.suspends(), 1);
+
+  // Pretend we were awakened.
+  source()->GenerateResumeEvent();
+  EXPECT_EQ(sender.resumes(), 1);
+
+  // Send a duplicate resume notification.  This should be suppressed.
+  source()->GenerateResumeEvent();
+  EXPECT_EQ(sender.resumes(), 1);
+
+  // Pretend the device has gone on battery power
+  source()->GeneratePowerStateEvent(true);
+  EXPECT_EQ(sender.power_state_changes(), 1);
+
+  // Repeated indications the device is on battery power should be suppressed.
+  source()->GeneratePowerStateEvent(true);
+  EXPECT_EQ(sender.power_state_changes(), 1);
+
+  // Pretend the device has gone off battery power
+  source()->GeneratePowerStateEvent(false);
+  EXPECT_EQ(sender.power_state_changes(), 2);
+
+  // Repeated indications the device is off battery power should be suppressed.
+  source()->GeneratePowerStateEvent(false);
+  EXPECT_EQ(sender.power_state_changes(), 2);
+}
+
+}  // namespace base