| 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 |