Index: content/child/power_monitor_broadcast_source_unittest.cc |
diff --git a/content/child/power_monitor_broadcast_source_unittest.cc b/content/child/power_monitor_broadcast_source_unittest.cc |
index 8049ecfcb2c9f14df33f29843574700e50ef33ee..19f046901b02f9e99e0e14c5629e3ba2208b228d 100644 |
--- a/content/child/power_monitor_broadcast_source_unittest.cc |
+++ b/content/child/power_monitor_broadcast_source_unittest.cc |
@@ -7,8 +7,6 @@ |
#include "base/run_loop.h" |
#include "base/test/power_monitor_test_base.h" |
#include "content/child/power_monitor_broadcast_source.h" |
-#include "content/common/power_monitor_messages.h" |
-#include "ipc/message_filter.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace content { |
@@ -35,61 +33,54 @@ class PowerMonitorBroadcastSourceTest : public testing::Test { |
}; |
TEST_F(PowerMonitorBroadcastSourceTest, PowerMessageReceiveBroadcast) { |
- IPC::MessageFilter* message_filter = source()->GetMessageFilter(); |
base::PowerMonitorTestObserver observer; |
monitor()->AddObserver(&observer); |
- PowerMonitorMsg_Suspend suspend_msg; |
- PowerMonitorMsg_Resume resume_msg; |
- |
// Sending resume when not suspended should have no effect. |
- message_filter->OnMessageReceived(resume_msg); |
+ source()->Resume(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.resumes(), 0); |
// Pretend we suspended. |
- message_filter->OnMessageReceived(suspend_msg); |
+ source()->Suspend(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.suspends(), 1); |
// Send a second suspend notification. This should be suppressed. |
- message_filter->OnMessageReceived(suspend_msg); |
+ source()->Suspend(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.suspends(), 1); |
// Pretend we were awakened. |
- message_filter->OnMessageReceived(resume_msg); |
+ source()->Resume(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.resumes(), 1); |
// Send a duplicate resume notification. This should be suppressed. |
- message_filter->OnMessageReceived(resume_msg); |
+ source()->Resume(); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.resumes(), 1); |
- PowerMonitorMsg_PowerStateChange on_battery_msg(true); |
- PowerMonitorMsg_PowerStateChange off_battery_msg(false); |
- |
// Pretend the device has gone on battery power |
- message_filter->OnMessageReceived(on_battery_msg); |
+ source()->PowerStateChange(true); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.power_state_changes(), 1); |
EXPECT_EQ(observer.last_power_state(), true); |
// Repeated indications the device is on battery power should be suppressed. |
- message_filter->OnMessageReceived(on_battery_msg); |
+ source()->PowerStateChange(true); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.power_state_changes(), 1); |
// Pretend the device has gone off battery power |
- message_filter->OnMessageReceived(off_battery_msg); |
+ source()->PowerStateChange(false); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.power_state_changes(), 2); |
EXPECT_EQ(observer.last_power_state(), false); |
// Repeated indications the device is off battery power should be suppressed. |
- message_filter->OnMessageReceived(off_battery_msg); |
+ source()->PowerStateChange(false); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(observer.power_state_changes(), 2); |
} |