| 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 | 9 |
| 10 #include "base/atomic_sequence_num.h" | 10 #include "base/atomic_sequence_num.h" |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/memory/scoped_vector.h" | 12 #include "base/memory/scoped_vector.h" |
| 13 #include "base/thread_task_runner_handle.h" | 13 #include "base/thread_task_runner_handle.h" |
| 14 #include "content/browser/message_port_message_filter.h" | 14 #include "content/browser/message_port_message_filter.h" |
| 15 #include "content/browser/service_worker/embedded_worker_instance.h" | 15 #include "content/browser/service_worker/embedded_worker_instance.h" |
| 16 #include "content/browser/service_worker/embedded_worker_registry.h" | 16 #include "content/browser/service_worker/embedded_worker_registry.h" |
| 17 #include "content/browser/service_worker/service_worker_context_core.h" | 17 #include "content/browser/service_worker/service_worker_context_core.h" |
| 18 #include "content/browser/service_worker/service_worker_context_wrapper.h" | 18 #include "content/browser/service_worker/service_worker_context_wrapper.h" |
| 19 #include "content/common/service_worker/embedded_worker_messages.h" | 19 #include "content/common/service_worker/embedded_worker_messages.h" |
| 20 #include "content/common/service_worker/service_worker_messages.h" | 20 #include "content/common/service_worker/service_worker_messages.h" |
| 21 #include "content/public/test/mock_render_process_host.h" |
| 22 #include "content/public/test/test_browser_context.h" |
| 21 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
| 22 | 24 |
| 23 namespace content { | 25 namespace content { |
| 24 | 26 |
| 27 namespace { |
| 28 |
| 25 class MockMessagePortMessageFilter : public MessagePortMessageFilter { | 29 class MockMessagePortMessageFilter : public MessagePortMessageFilter { |
| 26 public: | 30 public: |
| 27 MockMessagePortMessageFilter() | 31 MockMessagePortMessageFilter() |
| 28 : MessagePortMessageFilter( | 32 : MessagePortMessageFilter( |
| 29 base::Bind(&base::AtomicSequenceNumber::GetNext, | 33 base::Bind(&base::AtomicSequenceNumber::GetNext, |
| 30 base::Unretained(&next_routing_id_))) {} | 34 base::Unretained(&next_routing_id_))) {} |
| 31 | 35 |
| 32 bool Send(IPC::Message* message) override { | 36 bool Send(IPC::Message* message) override { |
| 33 message_queue_.push_back(message); | 37 message_queue_.push_back(message); |
| 34 return true; | 38 return true; |
| 35 } | 39 } |
| 36 | 40 |
| 37 private: | 41 private: |
| 38 ~MockMessagePortMessageFilter() override {} | 42 ~MockMessagePortMessageFilter() override {} |
| 39 base::AtomicSequenceNumber next_routing_id_; | 43 base::AtomicSequenceNumber next_routing_id_; |
| 40 ScopedVector<IPC::Message> message_queue_; | 44 ScopedVector<IPC::Message> message_queue_; |
| 41 }; | 45 }; |
| 42 | 46 |
| 47 } // namespace |
| 48 |
| 43 EmbeddedWorkerTestHelper::EmbeddedWorkerTestHelper( | 49 EmbeddedWorkerTestHelper::EmbeddedWorkerTestHelper( |
| 44 const base::FilePath& user_data_directory, | 50 const base::FilePath& user_data_directory, |
| 45 int mock_render_process_id) | 51 int mock_render_process_id) |
| 46 : wrapper_(new ServiceWorkerContextWrapper(NULL)), | 52 : wrapper_(new ServiceWorkerContextWrapper(NULL)), |
| 47 next_thread_id_(0), | 53 next_thread_id_(0), |
| 48 mock_render_process_id_(mock_render_process_id), | 54 mock_render_process_id_(mock_render_process_id), |
| 49 weak_factory_(this) { | 55 weak_factory_(this) { |
| 50 scoped_ptr<MockServiceWorkerDatabaseTaskManager> database_task_manager( | 56 scoped_ptr<MockServiceWorkerDatabaseTaskManager> database_task_manager( |
| 51 new MockServiceWorkerDatabaseTaskManager( | 57 new MockServiceWorkerDatabaseTaskManager( |
| 52 base::ThreadTaskRunnerHandle::Get())); | 58 base::ThreadTaskRunnerHandle::Get())); |
| 53 wrapper_->InitInternal(user_data_directory, | 59 wrapper_->InitInternal(user_data_directory, |
| 54 database_task_manager.Pass(), | 60 database_task_manager.Pass(), |
| 55 base::ThreadTaskRunnerHandle::Get(), | 61 base::ThreadTaskRunnerHandle::Get(), |
| 56 NULL, | 62 NULL, |
| 57 NULL); | 63 NULL); |
| 58 wrapper_->process_manager()->SetProcessIdForTest(mock_render_process_id); | 64 wrapper_->process_manager()->SetProcessIdForTest(mock_render_process_id); |
| 59 registry()->AddChildProcessSender(mock_render_process_id, this, | 65 registry()->AddChildProcessSender(mock_render_process_id, this, |
| 60 NewMessagePortMessageFilter()); | 66 NewMessagePortMessageFilter()); |
| 61 } | 67 } |
| 62 | 68 |
| 69 EmbeddedWorkerTestHelper::EmbeddedWorkerTestHelper( |
| 70 const base::FilePath& user_data_directory) |
| 71 : browser_context_(new TestBrowserContext), |
| 72 render_process_host_(new MockRenderProcessHost(browser_context_.get())), |
| 73 wrapper_(new ServiceWorkerContextWrapper(nullptr)), |
| 74 next_thread_id_(0), |
| 75 mock_render_process_id_(render_process_host_->GetID()), |
| 76 weak_factory_(this) { |
| 77 scoped_ptr<MockServiceWorkerDatabaseTaskManager> database_task_manager( |
| 78 new MockServiceWorkerDatabaseTaskManager( |
| 79 base::ThreadTaskRunnerHandle::Get())); |
| 80 wrapper_->InitInternal(user_data_directory, database_task_manager.Pass(), |
| 81 base::ThreadTaskRunnerHandle::Get(), nullptr, nullptr); |
| 82 wrapper_->process_manager()->SetProcessIdForTest(mock_render_process_id_); |
| 83 registry()->AddChildProcessSender(mock_render_process_id_, this, |
| 84 NewMessagePortMessageFilter()); |
| 85 } |
| 86 |
| 63 EmbeddedWorkerTestHelper::~EmbeddedWorkerTestHelper() { | 87 EmbeddedWorkerTestHelper::~EmbeddedWorkerTestHelper() { |
| 64 if (wrapper_.get()) | 88 if (wrapper_.get()) |
| 65 wrapper_->Shutdown(); | 89 wrapper_->Shutdown(); |
| 66 } | 90 } |
| 67 | 91 |
| 68 void EmbeddedWorkerTestHelper::SimulateAddProcessToPattern( | 92 void EmbeddedWorkerTestHelper::SimulateAddProcessToPattern( |
| 69 const GURL& pattern, | 93 const GURL& pattern, |
| 70 int process_id) { | 94 int process_id) { |
| 71 registry()->AddChildProcessSender(process_id, this, | 95 registry()->AddChildProcessSender(process_id, this, |
| 72 NewMessagePortMessageFilter()); | 96 NewMessagePortMessageFilter()); |
| (...skipping 266 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 339 | 363 |
| 340 MessagePortMessageFilter* | 364 MessagePortMessageFilter* |
| 341 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() { | 365 EmbeddedWorkerTestHelper::NewMessagePortMessageFilter() { |
| 342 scoped_refptr<MessagePortMessageFilter> filter( | 366 scoped_refptr<MessagePortMessageFilter> filter( |
| 343 new MockMessagePortMessageFilter); | 367 new MockMessagePortMessageFilter); |
| 344 message_port_message_filters_.push_back(filter); | 368 message_port_message_filters_.push_back(filter); |
| 345 return filter.get(); | 369 return filter.get(); |
| 346 } | 370 } |
| 347 | 371 |
| 348 } // namespace content | 372 } // namespace content |
| OLD | NEW |