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 |