| 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 |
| 11 #include "base/atomic_sequence_num.h" | 11 #include "base/atomic_sequence_num.h" |
| 12 #include "base/bind.h" | 12 #include "base/bind.h" |
| 13 #include "base/memory/scoped_vector.h" | 13 #include "base/memory/scoped_vector.h" |
| 14 #include "base/threading/thread_task_runner_handle.h" | 14 #include "base/threading/thread_task_runner_handle.h" |
| 15 #include "base/time/time.h" | 15 #include "base/time/time.h" |
| 16 #include "content/browser/message_port_message_filter.h" | |
| 17 #include "content/browser/service_worker/embedded_worker_instance.h" | 16 #include "content/browser/service_worker/embedded_worker_instance.h" |
| 18 #include "content/browser/service_worker/embedded_worker_registry.h" | 17 #include "content/browser/service_worker/embedded_worker_registry.h" |
| 19 #include "content/browser/service_worker/embedded_worker_status.h" | 18 #include "content/browser/service_worker/embedded_worker_status.h" |
| 20 #include "content/browser/service_worker/service_worker_context_core.h" | 19 #include "content/browser/service_worker/service_worker_context_core.h" |
| 21 #include "content/browser/service_worker/service_worker_context_wrapper.h" | 20 #include "content/browser/service_worker/service_worker_context_wrapper.h" |
| 22 #include "content/common/service_worker/embedded_worker_messages.h" | 21 #include "content/common/service_worker/embedded_worker_messages.h" |
| 23 #include "content/common/service_worker/embedded_worker_setup.mojom.h" | 22 #include "content/common/service_worker/embedded_worker_setup.mojom.h" |
| 24 #include "content/common/service_worker/embedded_worker_start_params.h" | 23 #include "content/common/service_worker/embedded_worker_start_params.h" |
| 25 #include "content/common/service_worker/service_worker_messages.h" | 24 #include "content/common/service_worker/service_worker_messages.h" |
| 26 #include "content/common/service_worker/service_worker_utils.h" | 25 #include "content/common/service_worker/service_worker_utils.h" |
| 27 #include "content/public/common/push_event_payload.h" | 26 #include "content/public/common/push_event_payload.h" |
| 28 #include "content/public/test/mock_render_process_host.h" | 27 #include "content/public/test/mock_render_process_host.h" |
| 29 #include "content/public/test/test_browser_context.h" | 28 #include "content/public/test/test_browser_context.h" |
| 30 #include "mojo/public/cpp/bindings/interface_request.h" | 29 #include "mojo/public/cpp/bindings/interface_request.h" |
| 31 #include "mojo/public/cpp/bindings/strong_binding.h" | 30 #include "mojo/public/cpp/bindings/strong_binding.h" |
| 32 #include "services/service_manager/public/cpp/interface_provider.h" | 31 #include "services/service_manager/public/cpp/interface_provider.h" |
| 33 #include "services/service_manager/public/cpp/interface_registry.h" | 32 #include "services/service_manager/public/cpp/interface_registry.h" |
| 34 #include "services/service_manager/public/interfaces/interface_provider.mojom.h" | 33 #include "services/service_manager/public/interfaces/interface_provider.mojom.h" |
| 35 #include "testing/gtest/include/gtest/gtest.h" | 34 #include "testing/gtest/include/gtest/gtest.h" |
| 36 | 35 |
| 37 namespace content { | 36 namespace content { |
| 38 | 37 |
| 39 namespace { | |
| 40 | |
| 41 class MockMessagePortMessageFilter : public MessagePortMessageFilter { | |
| 42 public: | |
| 43 MockMessagePortMessageFilter() | |
| 44 : MessagePortMessageFilter( | |
| 45 base::Bind(&base::AtomicSequenceNumber::GetNext, | |
| 46 base::Unretained(&next_routing_id_))) {} | |
| 47 | |
| 48 bool Send(IPC::Message* message) override { | |
| 49 message_queue_.push_back(message); | |
| 50 return true; | |
| 51 } | |
| 52 | |
| 53 private: | |
| 54 ~MockMessagePortMessageFilter() override {} | |
| 55 base::AtomicSequenceNumber next_routing_id_; | |
| 56 ScopedVector<IPC::Message> message_queue_; | |
| 57 }; | |
| 58 | |
| 59 } // namespace | |
| 60 | |
| 61 class EmbeddedWorkerTestHelper::MockEmbeddedWorkerSetup | 38 class EmbeddedWorkerTestHelper::MockEmbeddedWorkerSetup |
| 62 : public mojom::EmbeddedWorkerSetup { | 39 : public mojom::EmbeddedWorkerSetup { |
| 63 public: | 40 public: |
| 64 explicit MockEmbeddedWorkerSetup( | 41 explicit MockEmbeddedWorkerSetup( |
| 65 const base::WeakPtr<EmbeddedWorkerTestHelper>& helper) | 42 const base::WeakPtr<EmbeddedWorkerTestHelper>& helper) |
| 66 : helper_(helper) {} | 43 : helper_(helper) {} |
| 67 | 44 |
| 68 static void Create(const base::WeakPtr<EmbeddedWorkerTestHelper>& helper, | 45 static void Create(const base::WeakPtr<EmbeddedWorkerTestHelper>& helper, |
| 69 mojom::EmbeddedWorkerSetupRequest request) { | 46 mojom::EmbeddedWorkerSetupRequest request) { |
| 70 mojo::MakeStrongBinding(base::MakeUnique<MockEmbeddedWorkerSetup>(helper), | 47 mojo::MakeStrongBinding(base::MakeUnique<MockEmbeddedWorkerSetup>(helper), |
| (...skipping 140 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 211 mock_render_process_id_(render_process_host_->GetID()), | 188 mock_render_process_id_(render_process_host_->GetID()), |
| 212 new_mock_render_process_id_(new_render_process_host_->GetID()), | 189 new_mock_render_process_id_(new_render_process_host_->GetID()), |
| 213 weak_factory_(this) { | 190 weak_factory_(this) { |
| 214 std::unique_ptr<MockServiceWorkerDatabaseTaskManager> database_task_manager( | 191 std::unique_ptr<MockServiceWorkerDatabaseTaskManager> database_task_manager( |
| 215 new MockServiceWorkerDatabaseTaskManager( | 192 new MockServiceWorkerDatabaseTaskManager( |
| 216 base::ThreadTaskRunnerHandle::Get())); | 193 base::ThreadTaskRunnerHandle::Get())); |
| 217 wrapper_->InitInternal(user_data_directory, std::move(database_task_manager), | 194 wrapper_->InitInternal(user_data_directory, std::move(database_task_manager), |
| 218 base::ThreadTaskRunnerHandle::Get(), nullptr, nullptr); | 195 base::ThreadTaskRunnerHandle::Get(), nullptr, nullptr); |
| 219 wrapper_->process_manager()->SetProcessIdForTest(mock_render_process_id()); | 196 wrapper_->process_manager()->SetProcessIdForTest(mock_render_process_id()); |
| 220 wrapper_->process_manager()->SetNewProcessIdForTest(new_render_process_id()); | 197 wrapper_->process_manager()->SetNewProcessIdForTest(new_render_process_id()); |
| 221 registry()->AddChildProcessSender(mock_render_process_id_, this, | 198 registry()->AddChildProcessSender(mock_render_process_id_, this); |
| 222 NewMessagePortMessageFilter()); | |
| 223 | 199 |
| 224 // Setup process level interface registry. | 200 // Setup process level interface registry. |
| 225 render_process_interface_registry_ = | 201 render_process_interface_registry_ = |
| 226 CreateInterfaceRegistry(render_process_host_.get()); | 202 CreateInterfaceRegistry(render_process_host_.get()); |
| 227 new_render_process_interface_registry_ = | 203 new_render_process_interface_registry_ = |
| 228 CreateInterfaceRegistry(new_render_process_host_.get()); | 204 CreateInterfaceRegistry(new_render_process_host_.get()); |
| 229 } | 205 } |
| 230 | 206 |
| 231 EmbeddedWorkerTestHelper::~EmbeddedWorkerTestHelper() { | 207 EmbeddedWorkerTestHelper::~EmbeddedWorkerTestHelper() { |
| 232 if (wrapper_.get()) | 208 if (wrapper_.get()) |
| 233 wrapper_->Shutdown(); | 209 wrapper_->Shutdown(); |
| 234 } | 210 } |
| 235 | 211 |
| 236 void EmbeddedWorkerTestHelper::SimulateAddProcessToPattern(const GURL& pattern, | 212 void EmbeddedWorkerTestHelper::SimulateAddProcessToPattern(const GURL& pattern, |
| 237 int process_id) { | 213 int process_id) { |
| 238 registry()->AddChildProcessSender(process_id, this, | 214 registry()->AddChildProcessSender(process_id, this); |
| 239 NewMessagePortMessageFilter()); | |
| 240 wrapper_->process_manager()->AddProcessReferenceToPattern(pattern, | 215 wrapper_->process_manager()->AddProcessReferenceToPattern(pattern, |
| 241 process_id); | 216 process_id); |
| 242 } | 217 } |
| 243 | 218 |
| 244 bool EmbeddedWorkerTestHelper::Send(IPC::Message* message) { | 219 bool EmbeddedWorkerTestHelper::Send(IPC::Message* message) { |
| 245 OnMessageReceived(*message); | 220 OnMessageReceived(*message); |
| 246 delete message; | 221 delete message; |
| 247 return true; | 222 return true; |
| 248 } | 223 } |
| 249 | 224 |
| (...skipping 298 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 548 base::ThreadTaskRunnerHandle::Get()->PostTask( | 523 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 549 FROM_HERE, base::Bind(&EmbeddedWorkerTestHelper::OnPushEvent, AsWeakPtr(), | 524 FROM_HERE, base::Bind(&EmbeddedWorkerTestHelper::OnPushEvent, AsWeakPtr(), |
| 550 current_embedded_worker_id_, request_id, payload)); | 525 current_embedded_worker_id_, request_id, payload)); |
| 551 } | 526 } |
| 552 | 527 |
| 553 EmbeddedWorkerRegistry* EmbeddedWorkerTestHelper::registry() { | 528 EmbeddedWorkerRegistry* EmbeddedWorkerTestHelper::registry() { |
| 554 DCHECK(context()); | 529 DCHECK(context()); |
| 555 return context()->embedded_worker_registry(); | 530 return context()->embedded_worker_registry(); |
| 556 } | 531 } |
| 557 | 532 |
| 558 MessagePortMessageFilter* | |
| 559 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() { | |
| 560 scoped_refptr<MessagePortMessageFilter> filter( | |
| 561 new MockMessagePortMessageFilter); | |
| 562 message_port_message_filters_.push_back(filter); | |
| 563 return filter.get(); | |
| 564 } | |
| 565 | |
| 566 std::unique_ptr<service_manager::InterfaceRegistry> | 533 std::unique_ptr<service_manager::InterfaceRegistry> |
| 567 EmbeddedWorkerTestHelper::CreateInterfaceRegistry(MockRenderProcessHost* rph) { | 534 EmbeddedWorkerTestHelper::CreateInterfaceRegistry(MockRenderProcessHost* rph) { |
| 568 auto registry = | 535 auto registry = |
| 569 base::MakeUnique<service_manager::InterfaceRegistry>(std::string()); | 536 base::MakeUnique<service_manager::InterfaceRegistry>(std::string()); |
| 570 registry->AddInterface( | 537 registry->AddInterface( |
| 571 base::Bind(&MockEmbeddedWorkerSetup::Create, AsWeakPtr())); | 538 base::Bind(&MockEmbeddedWorkerSetup::Create, AsWeakPtr())); |
| 572 registry->AddInterface( | 539 registry->AddInterface( |
| 573 base::Bind(&MockEmbeddedWorkerInstanceClient::Bind, AsWeakPtr())); | 540 base::Bind(&MockEmbeddedWorkerInstanceClient::Bind, AsWeakPtr())); |
| 574 | 541 |
| 575 service_manager::mojom::InterfaceProviderPtr interfaces; | 542 service_manager::mojom::InterfaceProviderPtr interfaces; |
| 576 registry->Bind(mojo::MakeRequest(&interfaces), service_manager::Identity(), | 543 registry->Bind(mojo::MakeRequest(&interfaces), service_manager::Identity(), |
| 577 service_manager::InterfaceProviderSpec(), | 544 service_manager::InterfaceProviderSpec(), |
| 578 service_manager::Identity(), | 545 service_manager::Identity(), |
| 579 service_manager::InterfaceProviderSpec()); | 546 service_manager::InterfaceProviderSpec()); |
| 580 | 547 |
| 581 std::unique_ptr<service_manager::InterfaceProvider> remote_interfaces( | 548 std::unique_ptr<service_manager::InterfaceProvider> remote_interfaces( |
| 582 new service_manager::InterfaceProvider); | 549 new service_manager::InterfaceProvider); |
| 583 remote_interfaces->Bind(std::move(interfaces)); | 550 remote_interfaces->Bind(std::move(interfaces)); |
| 584 rph->SetRemoteInterfaces(std::move(remote_interfaces)); | 551 rph->SetRemoteInterfaces(std::move(remote_interfaces)); |
| 585 return registry; | 552 return registry; |
| 586 } | 553 } |
| 587 | 554 |
| 588 } // namespace content | 555 } // namespace content |
| OLD | NEW |