| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/public/test/mock_render_process_host.h" | 5 #include "content/public/test/mock_render_process_host.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <utility> | 8 #include <utility> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/lazy_instance.h" | 11 #include "base/lazy_instance.h" |
| 12 #include "base/location.h" | 12 #include "base/location.h" |
| 13 #include "base/memory/ptr_util.h" |
| 13 #include "base/process/process_handle.h" | 14 #include "base/process/process_handle.h" |
| 14 #include "base/single_thread_task_runner.h" | 15 #include "base/single_thread_task_runner.h" |
| 15 #include "base/threading/thread_task_runner_handle.h" | 16 #include "base/threading/thread_task_runner_handle.h" |
| 16 #include "base/time/time.h" | 17 #include "base/time/time.h" |
| 17 #include "content/browser/child_process_security_policy_impl.h" | 18 #include "content/browser/child_process_security_policy_impl.h" |
| 18 #include "content/browser/renderer_host/render_process_host_impl.h" | 19 #include "content/browser/renderer_host/render_process_host_impl.h" |
| 19 #include "content/browser/renderer_host/render_view_host_impl.h" | 20 #include "content/browser/renderer_host/render_view_host_impl.h" |
| 20 #include "content/browser/renderer_host/render_widget_host_impl.h" | 21 #include "content/browser/renderer_host/render_widget_host_impl.h" |
| 21 #include "content/common/child_process_host_impl.h" | 22 #include "content/common/child_process_host_impl.h" |
| 22 #include "content/common/frame_messages.h" | 23 #include "content/common/frame_messages.h" |
| (...skipping 347 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 370 return false; | 371 return false; |
| 371 } | 372 } |
| 372 | 373 |
| 373 void MockRenderProcessHost::OnChannelConnected(int32_t peer_pid) {} | 374 void MockRenderProcessHost::OnChannelConnected(int32_t peer_pid) {} |
| 374 | 375 |
| 375 MockRenderProcessHostFactory::MockRenderProcessHostFactory() {} | 376 MockRenderProcessHostFactory::MockRenderProcessHostFactory() {} |
| 376 | 377 |
| 377 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() { | 378 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() { |
| 378 // Detach this object from MockRenderProcesses to prevent STLDeleteElements() | 379 // Detach this object from MockRenderProcesses to prevent STLDeleteElements() |
| 379 // from calling MockRenderProcessHostFactory::Remove(). | 380 // from calling MockRenderProcessHostFactory::Remove(). |
| 380 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); | 381 for (const auto& process : processes_) |
| 381 it != processes_.end(); ++it) { | 382 process->SetFactory(nullptr); |
| 382 (*it)->SetFactory(NULL); | |
| 383 } | |
| 384 } | 383 } |
| 385 | 384 |
| 386 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost( | 385 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost( |
| 387 BrowserContext* browser_context, | 386 BrowserContext* browser_context, |
| 388 SiteInstance* site_instance) const { | 387 SiteInstance* site_instance) const { |
| 389 MockRenderProcessHost* host = new MockRenderProcessHost(browser_context); | 388 processes_.push_back( |
| 390 if (host) { | 389 base::MakeUnique<MockRenderProcessHost>(browser_context)); |
| 391 processes_.push_back(host); | 390 processes_.back()->SetFactory(this); |
| 392 host->SetFactory(this); | 391 return processes_.back().get(); |
| 393 } | |
| 394 return host; | |
| 395 } | 392 } |
| 396 | 393 |
| 397 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const { | 394 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const { |
| 398 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); | 395 for (auto it = processes_.begin(); it != processes_.end(); ++it) { |
| 399 it != processes_.end(); ++it) { | 396 if (it->get() == host) { |
| 400 if (*it == host) { | 397 it->release(); |
| 401 processes_.weak_erase(it); | 398 processes_.erase(it); |
| 402 break; | 399 break; |
| 403 } | 400 } |
| 404 } | 401 } |
| 405 } | 402 } |
| 406 | 403 |
| 407 } // namespace content | 404 } // namespace content |
| OLD | NEW |