| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "services/device/power_monitor/power_monitor_message_broadcaster.h" | 5 #include "services/device/power_monitor/power_monitor_message_broadcaster.h" |
| 6 | 6 |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "base/message_loop/message_loop.h" | |
| 9 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
| 10 #include "base/test/power_monitor_test_base.h" | 9 #include "base/test/power_monitor_test_base.h" |
| 11 #include "mojo/public/cpp/bindings/binding.h" | 10 #include "services/device/device_service_test_base.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "services/device/public/cpp/power_monitor/power_monitor_broadcast_sourc
e.h" |
| 13 | 12 |
| 14 namespace device { | 13 namespace device { |
| 15 | 14 |
| 16 class FakePowerMonitorClient : public device::mojom::PowerMonitorClient { | 15 class MockPowerMonitorBroadcastSource : public PowerMonitorBroadcastSource { |
| 17 public: | 16 public: |
| 18 FakePowerMonitorClient(device::mojom::PowerMonitorClientRequest request) | 17 MockPowerMonitorBroadcastSource(service_manager::Connector* connector, |
| 19 : binding_(this, std::move(request)), | 18 base::Closure service_connected) |
| 20 power_state_changes_(0), | 19 : PowerMonitorBroadcastSource(connector), |
| 21 suspends_(0), | 20 service_connected_(service_connected) {} |
| 22 resumes_(0) {} | 21 ~MockPowerMonitorBroadcastSource() override = default; |
| 23 ~FakePowerMonitorClient() override {} | |
| 24 | 22 |
| 25 // Implement device::mojom::PowerMonitorClient | 23 // Implement device::mojom::PowerMonitorClient |
| 26 void PowerStateChange(bool on_battery_power) override { | 24 void PowerStateChange(bool on_battery_power) override { |
| 27 power_state_changes_++; | 25 power_state_changes_++; |
| 26 service_connected_.Run(); |
| 28 } | 27 } |
| 29 void Suspend() override { suspends_++; } | 28 void Suspend() override { suspends_++; } |
| 30 void Resume() override { resumes_++; } | 29 void Resume() override { resumes_++; } |
| 31 | 30 |
| 32 // Test status counts. | 31 // Test status counts. |
| 33 int power_state_changes() { return power_state_changes_; } | 32 int power_state_changes() { return power_state_changes_; } |
| 34 int suspends() { return suspends_; } | 33 int suspends() { return suspends_; } |
| 35 int resumes() { return resumes_; } | 34 int resumes() { return resumes_; } |
| 36 | 35 |
| 37 private: | 36 private: |
| 38 mojo::Binding<device::mojom::PowerMonitorClient> binding_; | 37 int power_state_changes_ = 0; // Count of OnPowerStateChange notifications. |
| 39 int power_state_changes_; // Count of OnPowerStateChange notifications. | 38 int suspends_ = 0; // Count of OnSuspend notifications. |
| 40 int suspends_; // Count of OnSuspend notifications. | 39 int resumes_ = 0; // Count of OnResume notifications. |
| 41 int resumes_; // Count of OnResume notifications. | 40 base::Closure service_connected_; |
| 42 }; | 41 }; |
| 43 | 42 |
| 44 class PowerMonitorMessageBroadcasterTest : public testing::Test { | 43 class PowerMonitorMessageBroadcasterTest : public DeviceServiceTestBase { |
| 45 protected: | 44 protected: |
| 46 PowerMonitorMessageBroadcasterTest() { | 45 PowerMonitorMessageBroadcasterTest() = default; |
| 46 ~PowerMonitorMessageBroadcasterTest() override = default; |
| 47 |
| 48 void SetUp() override { |
| 49 DeviceServiceTestBase::SetUp(); |
| 50 |
| 47 power_monitor_source_ = new base::PowerMonitorTestSource(); | 51 power_monitor_source_ = new base::PowerMonitorTestSource(); |
| 48 power_monitor_.reset(new base::PowerMonitor( | 52 power_monitor_.reset(new base::PowerMonitor( |
| 49 std::unique_ptr<base::PowerMonitorSource>(power_monitor_source_))); | 53 std::unique_ptr<base::PowerMonitorSource>(power_monitor_source_))); |
| 50 } | 54 } |
| 51 ~PowerMonitorMessageBroadcasterTest() override {} | |
| 52 | 55 |
| 53 base::PowerMonitorTestSource* source() { return power_monitor_source_; } | 56 base::PowerMonitorTestSource* source() { return power_monitor_source_; } |
| 54 base::PowerMonitor* monitor() { return power_monitor_.get(); } | |
| 55 | 57 |
| 56 private: | 58 private: |
| 57 base::MessageLoop message_loop_; | |
| 58 base::PowerMonitorTestSource* power_monitor_source_; | 59 base::PowerMonitorTestSource* power_monitor_source_; |
| 59 std::unique_ptr<base::PowerMonitor> power_monitor_; | 60 std::unique_ptr<base::PowerMonitor> power_monitor_; |
| 60 | 61 |
| 61 DISALLOW_COPY_AND_ASSIGN(PowerMonitorMessageBroadcasterTest); | 62 DISALLOW_COPY_AND_ASSIGN(PowerMonitorMessageBroadcasterTest); |
| 62 }; | 63 }; |
| 63 | 64 |
| 64 TEST_F(PowerMonitorMessageBroadcasterTest, PowerMessageBroadcast) { | 65 TEST_F(PowerMonitorMessageBroadcasterTest, PowerMessageBroadcast) { |
| 65 device::mojom::PowerMonitorClientPtr proxy; | 66 base::RunLoop run_loop; |
| 66 FakePowerMonitorClient client(mojo::MakeRequest(&proxy)); | 67 MockPowerMonitorBroadcastSource client(connector(), run_loop.QuitClosure()); |
| 67 PowerMonitorMessageBroadcaster broadcaster; | 68 run_loop.Run(); |
| 68 | 69 |
| 69 // Calling AddClient should invoke a power state change. | 70 // Above PowerMonitorBroadcastSource ctor will connect to Device Service to |
| 70 broadcaster.AddClient(std::move(proxy)); | 71 // bind device::mojom::PowerMonitor interface, on which AddClient() will be |
| 71 | 72 // called then, this should invoke immediatelly a power state change back to |
| 73 // PowerMonitorBroadcastSource. |
| 72 base::RunLoop().RunUntilIdle(); | 74 base::RunLoop().RunUntilIdle(); |
| 73 EXPECT_EQ(client.power_state_changes(), 1); | 75 EXPECT_EQ(client.power_state_changes(), 1); |
| 74 | 76 |
| 75 // Sending resume when not suspended should have no effect. | 77 // Sending resume when not suspended should have no effect. |
| 76 source()->GenerateResumeEvent(); | 78 source()->GenerateResumeEvent(); |
| 77 EXPECT_EQ(client.resumes(), 0); | 79 EXPECT_EQ(client.resumes(), 0); |
| 78 | 80 |
| 79 // Pretend we suspended. | 81 // Pretend we suspended. |
| 80 source()->GenerateSuspendEvent(); | 82 source()->GenerateSuspendEvent(); |
| 81 EXPECT_EQ(client.suspends(), 1); | 83 EXPECT_EQ(client.suspends(), 1); |
| (...skipping 21 matching lines...) Expand all Loading... |
| 103 // Pretend the device has gone off battery power | 105 // Pretend the device has gone off battery power |
| 104 source()->GeneratePowerStateEvent(false); | 106 source()->GeneratePowerStateEvent(false); |
| 105 EXPECT_EQ(client.power_state_changes(), 3); | 107 EXPECT_EQ(client.power_state_changes(), 3); |
| 106 | 108 |
| 107 // Repeated indications the device is off battery power should be suppressed. | 109 // Repeated indications the device is off battery power should be suppressed. |
| 108 source()->GeneratePowerStateEvent(false); | 110 source()->GeneratePowerStateEvent(false); |
| 109 EXPECT_EQ(client.power_state_changes(), 3); | 111 EXPECT_EQ(client.power_state_changes(), 3); |
| 110 } | 112 } |
| 111 | 113 |
| 112 } // namespace device | 114 } // namespace device |
| OLD | NEW |