Index: content/common/power_monitor_message_broadcaster_unittest.cc |
diff --git a/content/common/power_monitor_message_broadcaster_unittest.cc b/content/common/power_monitor_message_broadcaster_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..31b20ed1f052fc19746743823e318035328e64ce |
--- /dev/null |
+++ b/content/common/power_monitor_message_broadcaster_unittest.cc |
@@ -0,0 +1,109 @@ |
+// 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/common/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 |