| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 "chrome/browser/renderer_host/mock_render_process_host.h" | 5 #include "chrome/browser/renderer_host/mock_render_process_host.h" |
| 6 | 6 |
| 7 #include "chrome/browser/child_process_security_policy.h" | 7 #include "chrome/browser/child_process_security_policy.h" |
| 8 | 8 |
| 9 MockRenderProcessHost::MockRenderProcessHost(Profile* profile) | 9 MockRenderProcessHost::MockRenderProcessHost(Profile* profile) |
| 10 : RenderProcessHost(profile), | 10 : RenderProcessHost(profile), |
| (...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 80 return true; | 80 return true; |
| 81 } | 81 } |
| 82 | 82 |
| 83 bool MockRenderProcessHost::SendWithTimeout(IPC::Message* msg, int timeout_ms) { | 83 bool MockRenderProcessHost::SendWithTimeout(IPC::Message* msg, int timeout_ms) { |
| 84 // Save the message in the sink. Just ignore timeout_ms. | 84 // Save the message in the sink. Just ignore timeout_ms. |
| 85 sink_.OnMessageReceived(*msg); | 85 sink_.OnMessageReceived(*msg); |
| 86 delete msg; | 86 delete msg; |
| 87 return true; | 87 return true; |
| 88 } | 88 } |
| 89 | 89 |
| 90 base::ProcessHandle MockRenderProcessHost::GetHandle() { |
| 91 return base::kNullProcessHandle; |
| 92 } |
| 93 |
| 90 bool MockRenderProcessHost::Send(IPC::Message* msg) { | 94 bool MockRenderProcessHost::Send(IPC::Message* msg) { |
| 91 // Save the message in the sink. | 95 // Save the message in the sink. |
| 92 sink_.OnMessageReceived(*msg); | 96 sink_.OnMessageReceived(*msg); |
| 93 delete msg; | 97 delete msg; |
| 94 return true; | 98 return true; |
| 95 } | 99 } |
| 96 | 100 |
| 97 TransportDIB* MockRenderProcessHost::GetTransportDIB(TransportDIB::Id dib_id) { | 101 TransportDIB* MockRenderProcessHost::GetTransportDIB(TransportDIB::Id dib_id) { |
| 98 if (transport_dib_) | 102 if (transport_dib_) |
| 99 return transport_dib_; | 103 return transport_dib_; |
| (...skipping 12 matching lines...) Expand all Loading... |
| 112 | 116 |
| 113 return transport_dib_; | 117 return transport_dib_; |
| 114 } | 118 } |
| 115 | 119 |
| 116 void MockRenderProcessHost::OnMessageReceived(const IPC::Message& msg) { | 120 void MockRenderProcessHost::OnMessageReceived(const IPC::Message& msg) { |
| 117 } | 121 } |
| 118 | 122 |
| 119 void MockRenderProcessHost::OnChannelConnected(int32 peer_pid) { | 123 void MockRenderProcessHost::OnChannelConnected(int32 peer_pid) { |
| 120 } | 124 } |
| 121 | 125 |
| 126 MockRenderProcessHostFactory::MockRenderProcessHostFactory() {} |
| 127 |
| 122 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() { | 128 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() { |
| 123 // Detach this object from MockRenderProcesses to prevent STLDeleteElements() | 129 // Detach this object from MockRenderProcesses to prevent STLDeleteElements() |
| 124 // from calling MockRenderProcessHostFactory::Remove(). | 130 // from calling MockRenderProcessHostFactory::Remove(). |
| 125 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); | 131 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); |
| 126 it != processes_.end(); ++it) { | 132 it != processes_.end(); ++it) { |
| 127 (*it)->SetFactory(NULL); | 133 (*it)->SetFactory(NULL); |
| 128 } | 134 } |
| 129 } | 135 } |
| 130 | 136 |
| 131 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost( | 137 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost( |
| 132 Profile* profile) const { | 138 Profile* profile) const { |
| 133 MockRenderProcessHost* host = new MockRenderProcessHost(profile); | 139 MockRenderProcessHost* host = new MockRenderProcessHost(profile); |
| 134 if (host) { | 140 if (host) { |
| 135 processes_.push_back(host); | 141 processes_.push_back(host); |
| 136 host->SetFactory(this); | 142 host->SetFactory(this); |
| 137 } | 143 } |
| 138 return host; | 144 return host; |
| 139 } | 145 } |
| 140 | 146 |
| 141 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const { | 147 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const { |
| 142 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); | 148 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); |
| 143 it != processes_.end(); ++it) { | 149 it != processes_.end(); ++it) { |
| 144 if (*it == host) { | 150 if (*it == host) { |
| 145 processes_.weak_erase(it); | 151 processes_.weak_erase(it); |
| 146 break; | 152 break; |
| 147 } | 153 } |
| 148 } | 154 } |
| 149 } | 155 } |
| OLD | NEW |