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
3 // found in the LICENSE file.
5 #include "content/public/test/mock_render_process_host.h"
7 #include "base/lazy_instance.h"
8 #include "base/message_loop/message_loop.h"
9 #include "base/time/time.h"
10 #include "content/browser/child_process_security_policy_impl.h"
11 #include "content/browser/renderer_host/render_process_host_impl.h"
12 #include "content/browser/renderer_host/render_view_host_impl.h"
13 #include "content/browser/renderer_host/render_widget_host_impl.h"
14 #include "content/common/child_process_host_impl.h"
15 #include "content/public/browser/global_request_id.h"
16 #include "content/public/browser/render_widget_host_iterator.h"
17 #include "content/public/browser/storage_partition.h"
21 MockRenderProcessHost::MockRenderProcessHost(BrowserContext* browser_context)
24 id_(ChildProcessHostImpl::GenerateChildProcessUniqueId()),
25 browser_context_(browser_context),
27 fast_shutdown_started_(false),
28 deletion_callback_called_(false),
29 is_isolated_guest_(false) {
30 // Child process security operations can't be unit tested unless we add
31 // ourselves as an existing child process.
32 ChildProcessSecurityPolicyImpl::GetInstance()->Add(GetID());
34 RenderProcessHostImpl::RegisterHost(GetID(), this);
37 MockRenderProcessHost::~MockRenderProcessHost() {
38 ChildProcessSecurityPolicyImpl::GetInstance()->Remove(GetID());
40 factory_->Remove(this);
42 // In unit tests, Cleanup() might not have been called.
43 if (!deletion_callback_called_) {
44 FOR_EACH_OBSERVER(RenderProcessHostObserver,
46 RenderProcessHostDestroyed(this));
47 RenderProcessHostImpl::UnregisterHost(GetID());
51 void MockRenderProcessHost::EnableSendQueue() {
54 bool MockRenderProcessHost::Init() {
58 int MockRenderProcessHost::GetNextRoutingID() {
59 return ++prev_routing_id_;
62 void MockRenderProcessHost::AddRoute(
64 IPC::Listener* listener) {
65 listeners_.AddWithID(listener, routing_id);
68 void MockRenderProcessHost::RemoveRoute(int32 routing_id) {
69 DCHECK(listeners_.Lookup(routing_id) != NULL);
70 listeners_.Remove(routing_id);
74 void MockRenderProcessHost::AddObserver(RenderProcessHostObserver* observer) {
75 observers_.AddObserver(observer);
78 void MockRenderProcessHost::RemoveObserver(
79 RenderProcessHostObserver* observer) {
80 observers_.RemoveObserver(observer);
83 void MockRenderProcessHost::ReceivedBadMessage() {
87 void MockRenderProcessHost::WidgetRestored() {
90 void MockRenderProcessHost::WidgetHidden() {
93 int MockRenderProcessHost::VisibleWidgetCount() const {
97 bool MockRenderProcessHost::IsIsolatedGuest() const {
98 return is_isolated_guest_;
101 StoragePartition* MockRenderProcessHost::GetStoragePartition() const {
105 void MockRenderProcessHost::AddWord(const base::string16& word) {
108 bool MockRenderProcessHost::FastShutdownIfPossible() {
109 // We aren't actually going to do anything, but set |fast_shutdown_started_|
110 // to true so that tests know we've been called.
111 fast_shutdown_started_ = true;
115 bool MockRenderProcessHost::FastShutdownStarted() const {
116 return fast_shutdown_started_;
119 void MockRenderProcessHost::DumpHandles() {
122 base::ProcessHandle MockRenderProcessHost::GetHandle() const {
123 // Return the current-process handle for the IPC::GetFileHandleForProcess
125 return base::Process::Current().handle();
128 bool MockRenderProcessHost::Send(IPC::Message* msg) {
129 // Save the message in the sink.
130 sink_.OnMessageReceived(*msg);
135 int MockRenderProcessHost::GetID() const {
139 bool MockRenderProcessHost::HasConnection() const {
143 void MockRenderProcessHost::SetIgnoreInputEvents(bool ignore_input_events) {
146 bool MockRenderProcessHost::IgnoreInputEvents() const {
150 void MockRenderProcessHost::Cleanup() {
151 if (listeners_.IsEmpty()) {
152 FOR_EACH_OBSERVER(RenderProcessHostObserver,
154 RenderProcessHostDestroyed(this));
155 base::MessageLoop::current()->DeleteSoon(FROM_HERE, this);
156 RenderProcessHostImpl::UnregisterHost(GetID());
157 deletion_callback_called_ = true;
161 void MockRenderProcessHost::AddPendingView() {
164 void MockRenderProcessHost::RemovePendingView() {
167 void MockRenderProcessHost::SetSuddenTerminationAllowed(bool allowed) {
170 bool MockRenderProcessHost::SuddenTerminationAllowed() const {
174 BrowserContext* MockRenderProcessHost::GetBrowserContext() const {
175 return browser_context_;
178 bool MockRenderProcessHost::InSameStoragePartition(
179 StoragePartition* partition) const {
180 // Mock RPHs only have one partition.
184 IPC::ChannelProxy* MockRenderProcessHost::GetChannel() {
188 void MockRenderProcessHost::AddFilter(BrowserMessageFilter* filter) {
191 int MockRenderProcessHost::GetActiveViewCount() {
192 int num_active_views = 0;
193 scoped_ptr<RenderWidgetHostIterator> widgets(
194 RenderWidgetHost::GetRenderWidgetHosts());
195 while (RenderWidgetHost* widget = widgets->GetNextHost()) {
196 // Count only RenderWidgetHosts in this process.
197 if (widget->GetProcess()->GetID() == GetID())
200 return num_active_views;
203 bool MockRenderProcessHost::FastShutdownForPageCount(size_t count) {
204 if (static_cast<size_t>(GetActiveViewCount()) == count)
205 return FastShutdownIfPossible();
209 base::TimeDelta MockRenderProcessHost::GetChildProcessIdleTime() const {
210 return base::TimeDelta::FromMilliseconds(0);
213 void MockRenderProcessHost::ResumeRequestsForView(int route_id) {
216 void MockRenderProcessHost::NotifyTimezoneChange() {
219 ServiceRegistry* MockRenderProcessHost::GetServiceRegistry() {
223 void MockRenderProcessHost::FilterURL(bool empty_allowed, GURL* url) {
224 RenderProcessHostImpl::FilterURL(this, empty_allowed, url);
227 #if defined(ENABLE_WEBRTC)
228 void MockRenderProcessHost::EnableAecDump(const base::FilePath& file) {
231 void MockRenderProcessHost::DisableAecDump() {
234 void MockRenderProcessHost::SetWebRtcLogMessageCallback(
235 base::Callback<void(const std::string&)> callback) {
238 RenderProcessHost::WebRtcStopRtpDumpCallback
239 MockRenderProcessHost::StartRtpDump(
242 const WebRtcRtpPacketCallback& packet_callback) {
243 return WebRtcStopRtpDumpCallback();
247 void MockRenderProcessHost::ResumeDeferredNavigation(
248 const GlobalRequestID& request_id) {}
250 bool MockRenderProcessHost::OnMessageReceived(const IPC::Message& msg) {
251 IPC::Listener* listener = listeners_.Lookup(msg.routing_id());
253 return listener->OnMessageReceived(msg);
257 void MockRenderProcessHost::OnChannelConnected(int32 peer_pid) {
260 MockRenderProcessHostFactory::MockRenderProcessHostFactory() {}
262 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() {
263 // Detach this object from MockRenderProcesses to prevent STLDeleteElements()
264 // from calling MockRenderProcessHostFactory::Remove().
265 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin();
266 it != processes_.end(); ++it) {
267 (*it)->SetFactory(NULL);
271 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost(
272 BrowserContext* browser_context,
273 SiteInstance* site_instance) const {
274 MockRenderProcessHost* host = new MockRenderProcessHost(browser_context);
276 processes_.push_back(host);
277 host->SetFactory(this);
282 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const {
283 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin();
284 it != processes_.end(); ++it) {
286 processes_.weak_erase(it);