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