OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "content/browser/service_worker/embedded_worker_test_helper.h" | 5 #include "content/browser/service_worker/embedded_worker_test_helper.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 #include <string> | 8 #include <string> |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
(...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
123 bool EmbeddedWorkerTestHelper::Send(IPC::Message* message) { | 123 bool EmbeddedWorkerTestHelper::Send(IPC::Message* message) { |
124 OnMessageReceived(*message); | 124 OnMessageReceived(*message); |
125 delete message; | 125 delete message; |
126 return true; | 126 return true; |
127 } | 127 } |
128 | 128 |
129 bool EmbeddedWorkerTestHelper::OnMessageReceived(const IPC::Message& message) { | 129 bool EmbeddedWorkerTestHelper::OnMessageReceived(const IPC::Message& message) { |
130 bool handled = true; | 130 bool handled = true; |
131 IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerTestHelper, message) | 131 IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerTestHelper, message) |
132 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StartWorker, OnStartWorkerStub) | 132 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StartWorker, OnStartWorkerStub) |
133 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_ResumeAfterDownload, | |
134 OnResumeAfterDownloadStub) | |
135 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StopWorker, OnStopWorkerStub) | 133 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StopWorker, OnStopWorkerStub) |
136 IPC_MESSAGE_HANDLER(EmbeddedWorkerContextMsg_MessageToWorker, | 134 IPC_MESSAGE_HANDLER(EmbeddedWorkerContextMsg_MessageToWorker, |
137 OnMessageToWorkerStub) | 135 OnMessageToWorkerStub) |
138 IPC_MESSAGE_UNHANDLED(handled = false) | 136 IPC_MESSAGE_UNHANDLED(handled = false) |
139 IPC_END_MESSAGE_MAP() | 137 IPC_END_MESSAGE_MAP() |
140 | 138 |
141 // IPC::TestSink only records messages that are not handled by filters, | 139 // IPC::TestSink only records messages that are not handled by filters, |
142 // so we just forward all messages to the separate sink. | 140 // so we just forward all messages to the separate sink. |
143 sink_.OnMessageReceived(message); | 141 sink_.OnMessageReceived(message); |
144 | 142 |
145 return handled; | 143 return handled; |
146 } | 144 } |
147 | 145 |
148 ServiceWorkerContextCore* EmbeddedWorkerTestHelper::context() { | 146 ServiceWorkerContextCore* EmbeddedWorkerTestHelper::context() { |
149 return wrapper_->context(); | 147 return wrapper_->context(); |
150 } | 148 } |
151 | 149 |
152 void EmbeddedWorkerTestHelper::ShutdownContext() { | 150 void EmbeddedWorkerTestHelper::ShutdownContext() { |
153 wrapper_->Shutdown(); | 151 wrapper_->Shutdown(); |
154 wrapper_ = NULL; | 152 wrapper_ = NULL; |
155 } | 153 } |
156 | 154 |
157 void EmbeddedWorkerTestHelper::OnStartWorker(int embedded_worker_id, | 155 void EmbeddedWorkerTestHelper::OnStartWorker(int embedded_worker_id, |
158 int64_t service_worker_version_id, | 156 int64_t service_worker_version_id, |
159 const GURL& scope, | 157 const GURL& scope, |
160 const GURL& script_url, | 158 const GURL& script_url) { |
161 bool pause_after_download) { | |
162 embedded_worker_id_service_worker_version_id_map_[embedded_worker_id] = | 159 embedded_worker_id_service_worker_version_id_map_[embedded_worker_id] = |
163 service_worker_version_id; | 160 service_worker_version_id; |
164 SimulateWorkerReadyForInspection(embedded_worker_id); | 161 SimulateWorkerReadyForInspection(embedded_worker_id); |
165 SimulateWorkerScriptCached(embedded_worker_id); | 162 SimulateWorkerScriptCached(embedded_worker_id); |
166 SimulateWorkerScriptLoaded(embedded_worker_id); | 163 SimulateWorkerScriptLoaded(embedded_worker_id); |
167 if (!pause_after_download) | |
168 OnResumeAfterDownload(embedded_worker_id); | |
169 } | |
170 | |
171 void EmbeddedWorkerTestHelper::OnResumeAfterDownload(int embedded_worker_id) { | |
172 SimulateWorkerThreadStarted(GetNextThreadId(), embedded_worker_id); | 164 SimulateWorkerThreadStarted(GetNextThreadId(), embedded_worker_id); |
173 SimulateWorkerScriptEvaluated(embedded_worker_id, true /* success */); | 165 SimulateWorkerScriptEvaluated(embedded_worker_id, true /* success */); |
174 SimulateWorkerStarted(embedded_worker_id); | 166 SimulateWorkerStarted(embedded_worker_id); |
175 } | 167 } |
176 | 168 |
177 void EmbeddedWorkerTestHelper::OnStopWorker(int embedded_worker_id) { | 169 void EmbeddedWorkerTestHelper::OnStopWorker(int embedded_worker_id) { |
178 // By default just notify the sender that the worker is stopped. | 170 // By default just notify the sender that the worker is stopped. |
179 SimulateWorkerStopped(embedded_worker_id); | 171 SimulateWorkerStopped(embedded_worker_id); |
180 } | 172 } |
181 | 173 |
(...skipping 132 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
314 const EmbeddedWorkerMsg_StartWorker_Params& params) { | 306 const EmbeddedWorkerMsg_StartWorker_Params& params) { |
315 EmbeddedWorkerInstance* worker = | 307 EmbeddedWorkerInstance* worker = |
316 registry()->GetWorker(params.embedded_worker_id); | 308 registry()->GetWorker(params.embedded_worker_id); |
317 ASSERT_TRUE(worker != NULL); | 309 ASSERT_TRUE(worker != NULL); |
318 EXPECT_EQ(EmbeddedWorkerInstance::STARTING, worker->status()); | 310 EXPECT_EQ(EmbeddedWorkerInstance::STARTING, worker->status()); |
319 base::ThreadTaskRunnerHandle::Get()->PostTask( | 311 base::ThreadTaskRunnerHandle::Get()->PostTask( |
320 FROM_HERE, | 312 FROM_HERE, |
321 base::Bind(&EmbeddedWorkerTestHelper::OnStartWorker, | 313 base::Bind(&EmbeddedWorkerTestHelper::OnStartWorker, |
322 weak_factory_.GetWeakPtr(), params.embedded_worker_id, | 314 weak_factory_.GetWeakPtr(), params.embedded_worker_id, |
323 params.service_worker_version_id, params.scope, | 315 params.service_worker_version_id, params.scope, |
324 params.script_url, params.pause_after_download)); | 316 params.script_url)); |
325 } | |
326 | |
327 void EmbeddedWorkerTestHelper::OnResumeAfterDownloadStub( | |
328 int embedded_worker_id) { | |
329 EmbeddedWorkerInstance* worker = registry()->GetWorker(embedded_worker_id); | |
330 ASSERT_TRUE(worker); | |
331 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
332 FROM_HERE, base::Bind(&EmbeddedWorkerTestHelper::OnResumeAfterDownload, | |
333 weak_factory_.GetWeakPtr(), embedded_worker_id)); | |
334 } | 317 } |
335 | 318 |
336 void EmbeddedWorkerTestHelper::OnStopWorkerStub(int embedded_worker_id) { | 319 void EmbeddedWorkerTestHelper::OnStopWorkerStub(int embedded_worker_id) { |
337 EmbeddedWorkerInstance* worker = registry()->GetWorker(embedded_worker_id); | 320 EmbeddedWorkerInstance* worker = registry()->GetWorker(embedded_worker_id); |
338 ASSERT_TRUE(worker != NULL); | 321 ASSERT_TRUE(worker != NULL); |
339 base::ThreadTaskRunnerHandle::Get()->PostTask( | 322 base::ThreadTaskRunnerHandle::Get()->PostTask( |
340 FROM_HERE, | 323 FROM_HERE, |
341 base::Bind(&EmbeddedWorkerTestHelper::OnStopWorker, | 324 base::Bind(&EmbeddedWorkerTestHelper::OnStopWorker, |
342 weak_factory_.GetWeakPtr(), | 325 weak_factory_.GetWeakPtr(), |
343 embedded_worker_id)); | 326 embedded_worker_id)); |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
417 | 400 |
418 MessagePortMessageFilter* | 401 MessagePortMessageFilter* |
419 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() { | 402 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() { |
420 scoped_refptr<MessagePortMessageFilter> filter( | 403 scoped_refptr<MessagePortMessageFilter> filter( |
421 new MockMessagePortMessageFilter); | 404 new MockMessagePortMessageFilter); |
422 message_port_message_filters_.push_back(filter); | 405 message_port_message_filters_.push_back(filter); |
423 return filter.get(); | 406 return filter.get(); |
424 } | 407 } |
425 | 408 |
426 } // namespace content | 409 } // namespace content |
OLD | NEW |