| // 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/test/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) { |
| } |
| ~PowerMonitorMessageSender() override {} |
| |
| 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_))); |
| } |
| ~PowerMonitorMessageBroadcasterTest() override {} |
| |
| 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); |
| |
| // Calling Init should invoke a power state change. |
| broadcaster.Init(); |
| EXPECT_EQ(sender.power_state_changes(), 1); |
| |
| // 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(), 2); |
| |
| // Repeated indications the device is on battery power should be suppressed. |
| source()->GeneratePowerStateEvent(true); |
| EXPECT_EQ(sender.power_state_changes(), 2); |
| |
| // Pretend the device has gone off battery power |
| source()->GeneratePowerStateEvent(false); |
| EXPECT_EQ(sender.power_state_changes(), 3); |
| |
| // Repeated indications the device is off battery power should be suppressed. |
| source()->GeneratePowerStateEvent(false); |
| EXPECT_EQ(sender.power_state_changes(), 3); |
| } |
| |
| } // namespace base |