OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 <utility> | 5 #include <utility> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
10 #include "chromeos/dbus/dbus_thread_manager.h" | 10 #include "chromeos/dbus/dbus_thread_manager.h" |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
61 | 61 |
62 case ArcBridgeService::State::STOPPED: | 62 case ArcBridgeService::State::STOPPED: |
63 message_loop_.PostTask(FROM_HERE, message_loop_.QuitWhenIdleClosure()); | 63 message_loop_.PostTask(FROM_HERE, message_loop_.QuitWhenIdleClosure()); |
64 break; | 64 break; |
65 | 65 |
66 default: | 66 default: |
67 break; | 67 break; |
68 } | 68 } |
69 } | 69 } |
70 | 70 |
71 void OnInstanceBootPhase(InstanceBootPhase boot_phase) override { | |
72 boot_phase_ = boot_phase; | |
73 } | |
74 | |
75 bool ready() const { return ready_; } | 71 bool ready() const { return ready_; } |
76 InstanceBootPhase boot_phase() const { return boot_phase_; } | |
77 ArcBridgeService::State state() const { return state_; } | 72 ArcBridgeService::State state() const { return state_; } |
78 | 73 |
79 protected: | 74 protected: |
80 scoped_ptr<ArcBridgeService> service_; | 75 scoped_ptr<ArcBridgeService> service_; |
81 scoped_ptr<FakeArcBridgeInstance> instance_; | 76 scoped_ptr<FakeArcBridgeInstance> instance_; |
82 | 77 |
83 private: | 78 private: |
84 void SetUp() override { | 79 void SetUp() override { |
85 chromeos::DBusThreadManager::Initialize(); | 80 chromeos::DBusThreadManager::Initialize(); |
86 | 81 |
87 ready_ = false; | 82 ready_ = false; |
88 state_ = ArcBridgeService::State::STOPPED; | 83 state_ = ArcBridgeService::State::STOPPED; |
89 boot_phase_ = INSTANCE_BOOT_PHASE_NOT_RUNNING; | |
90 | 84 |
91 ipc_support_.reset(new IPC::ScopedIPCSupport(message_loop_.task_runner())); | 85 ipc_support_.reset(new IPC::ScopedIPCSupport(message_loop_.task_runner())); |
92 instance_.reset(new FakeArcBridgeInstance()); | 86 instance_.reset(new FakeArcBridgeInstance()); |
93 service_.reset(new ArcBridgeServiceImpl( | 87 service_.reset(new ArcBridgeServiceImpl( |
94 make_scoped_ptr(new FakeArcBridgeBootstrap(instance_.get())))); | 88 make_scoped_ptr(new FakeArcBridgeBootstrap(instance_.get())))); |
95 | 89 |
96 service_->AddObserver(this); | 90 service_->AddObserver(this); |
97 } | 91 } |
98 | 92 |
99 void TearDown() override { | 93 void TearDown() override { |
100 service_->RemoveObserver(this); | 94 service_->RemoveObserver(this); |
101 instance_.reset(); | 95 instance_.reset(); |
102 service_.reset(); | 96 service_.reset(); |
103 ipc_support_.reset(); | 97 ipc_support_.reset(); |
104 | 98 |
105 chromeos::DBusThreadManager::Shutdown(); | 99 chromeos::DBusThreadManager::Shutdown(); |
106 } | 100 } |
107 | 101 |
108 bool ready_; | 102 bool ready_; |
109 InstanceBootPhase boot_phase_; | |
110 ArcBridgeService::State state_; | 103 ArcBridgeService::State state_; |
111 scoped_ptr<IPC::ScopedIPCSupport> ipc_support_; | 104 scoped_ptr<IPC::ScopedIPCSupport> ipc_support_; |
112 base::MessageLoopForUI message_loop_; | 105 base::MessageLoopForUI message_loop_; |
113 | 106 |
114 DISALLOW_COPY_AND_ASSIGN(ArcBridgeTest); | 107 DISALLOW_COPY_AND_ASSIGN(ArcBridgeTest); |
115 }; | 108 }; |
116 | 109 |
117 // Shuts down the instance reports booted. | 110 // Shuts down the instance reports booted. |
118 class ScopedShutdownWhenBooted : public ArcBridgeService::Observer { | 111 class ScopedShutdownWhenReady : public ArcBridgeService::Observer { |
119 public: | 112 public: |
120 explicit ScopedShutdownWhenBooted(ArcBridgeService* service) | 113 explicit ScopedShutdownWhenReady(ArcBridgeService* service) |
121 : service_(service) { | 114 : service_(service) { |
122 service_->AddObserver(this); | 115 service_->AddObserver(this); |
123 } | 116 } |
124 | 117 |
125 ~ScopedShutdownWhenBooted() override { service_->RemoveObserver(this); } | 118 ~ScopedShutdownWhenReady() override { service_->RemoveObserver(this); } |
126 | 119 |
127 void OnInstanceBootPhase(InstanceBootPhase boot_phase) override { | 120 void OnStateChanged(ArcBridgeService::State state) override { |
128 if (boot_phase == INSTANCE_BOOT_PHASE_BOOT_COMPLETED) { | 121 if (state == ArcBridgeService::State::READY) { |
129 service_->Shutdown(); | 122 service_->Shutdown(); |
130 } | 123 } |
131 } | 124 } |
132 | 125 |
133 private: | 126 private: |
134 ArcBridgeService* service_; | 127 ArcBridgeService* service_; |
135 | 128 |
136 DISALLOW_COPY_AND_ASSIGN(ScopedShutdownWhenBooted); | 129 DISALLOW_COPY_AND_ASSIGN(ScopedShutdownWhenReady); |
137 }; | 130 }; |
138 | 131 |
139 // Exercises the basic functionality of the ARC Bridge Service. A message from | 132 // Exercises the basic functionality of the ARC Bridge Service. A message from |
140 // within the instance should cause the observer to be notified. | 133 // within the instance should cause the observer to be notified. |
141 TEST_F(ArcBridgeTest, Basic) { | 134 TEST_F(ArcBridgeTest, Basic) { |
142 ASSERT_FALSE(ready()); | 135 ASSERT_FALSE(ready()); |
143 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 136 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
144 | 137 |
145 ScopedShutdownWhenBooted shutdown(service_.get()); | 138 ScopedShutdownWhenReady shutdown(service_.get()); |
146 | 139 |
147 service_->SetAvailable(true); | 140 service_->SetAvailable(true); |
148 service_->HandleStartup(); | 141 service_->HandleStartup(); |
149 | 142 |
150 ASSERT_EQ(ArcBridgeService::State::CONNECTED, state()); | 143 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
151 | 144 |
152 base::RunLoop run_loop; | 145 base::RunLoop run_loop; |
153 run_loop.Run(); | 146 run_loop.Run(); |
154 | 147 |
155 EXPECT_TRUE(ready()); | 148 EXPECT_TRUE(ready()); |
156 ASSERT_EQ(INSTANCE_BOOT_PHASE_BOOT_COMPLETED, boot_phase()); | |
157 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 149 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
158 } | 150 } |
159 | 151 |
160 // If not all pre-requisites are met, the instance is not started. | 152 // If not all pre-requisites are met, the instance is not started. |
161 TEST_F(ArcBridgeTest, Prerequisites) { | 153 TEST_F(ArcBridgeTest, Prerequisites) { |
162 ASSERT_FALSE(ready()); | 154 ASSERT_FALSE(ready()); |
163 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 155 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
164 service_->SetAvailable(true); | 156 service_->SetAvailable(true); |
165 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 157 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
166 service_->SetAvailable(false); | 158 service_->SetAvailable(false); |
167 service_->HandleStartup(); | 159 service_->HandleStartup(); |
168 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 160 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
169 } | 161 } |
170 | 162 |
171 // If the ArcBridgeService is shut down, it should be stopped, even | 163 // If the ArcBridgeService is shut down, it should be stopped, even |
172 // mid-startup. | 164 // mid-startup. |
173 TEST_F(ArcBridgeTest, ShutdownMidStartup) { | 165 TEST_F(ArcBridgeTest, ShutdownMidStartup) { |
174 ASSERT_FALSE(ready()); | 166 ASSERT_FALSE(ready()); |
175 | 167 |
176 service_->SetAvailable(true); | 168 service_->SetAvailable(true); |
177 service_->HandleStartup(); | 169 service_->HandleStartup(); |
178 | 170 |
179 ASSERT_EQ(ArcBridgeService::State::CONNECTED, state()); | 171 ASSERT_EQ(ArcBridgeService::State::READY, state()); |
180 service_->Shutdown(); | 172 service_->Shutdown(); |
181 // Some machines can reach the STOPPED state immediately. | 173 // Some machines can reach the STOPPED state immediately. |
182 ASSERT_TRUE(state() == ArcBridgeService::State::STOPPING || | 174 ASSERT_TRUE(state() == ArcBridgeService::State::STOPPING || |
183 state() == ArcBridgeService::State::STOPPED); | 175 state() == ArcBridgeService::State::STOPPED); |
184 | 176 |
185 base::RunLoop run_loop; | 177 base::RunLoop run_loop; |
186 run_loop.Run(); | 178 run_loop.Run(); |
187 | 179 |
188 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); | 180 ASSERT_EQ(ArcBridgeService::State::STOPPED, state()); |
189 } | 181 } |
190 | 182 |
191 } // namespace arc | 183 } // namespace arc |
OLD | NEW |