content/public/test/mock_render_process_host.cc

   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.
   4
   5 #include "content/public/test/mock_render_process_host.h"
   6
   7 #include "base/lazy_instance.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/process/process_handle.h"
  10 #include "base/time/time.h"
  11 #include "content/browser/child_process_security_policy_impl.h"
  12 #include "content/browser/renderer_host/render_process_host_impl.h"
  13 #include "content/browser/renderer_host/render_view_host_impl.h"
  14 #include "content/browser/renderer_host/render_widget_host_impl.h"
  15 #include "content/common/child_process_host_impl.h"
  16 #include "content/public/browser/browser_context.h"
  17 #include "content/public/browser/global_request_id.h"
  18 #include "content/public/browser/render_widget_host_iterator.h"
  19 #include "content/public/browser/storage_partition.h"
  20
  21 namespace content {
  22
  23 MockRenderProcessHost::MockRenderProcessHost(BrowserContext* browser_context)
  24     : bad_msg_count_(0),
  25       factory_(NULL),
  26       id_(ChildProcessHostImpl::GenerateChildProcessUniqueId()),
  27       browser_context_(browser_context),
  28       prev_routing_id_(0),
  29       fast_shutdown_started_(false),
  30       deletion_callback_called_(false),
  31       is_isolated_guest_(false) {
  32   // Child process security operations can't be unit tested unless we add
  33   // ourselves as an existing child process.
  34   ChildProcessSecurityPolicyImpl::GetInstance()->Add(GetID());
  35
  36   RenderProcessHostImpl::RegisterHost(GetID(), this);
  37 }
  38
  39 MockRenderProcessHost::~MockRenderProcessHost() {
  40   ChildProcessSecurityPolicyImpl::GetInstance()->Remove(GetID());
  41   if (factory_)
  42     factory_->Remove(this);
  43
  44   // In unit tests, Cleanup() might not have been called.
  45   if (!deletion_callback_called_) {
  46     FOR_EACH_OBSERVER(RenderProcessHostObserver,
  47                       observers_,
  48                       RenderProcessHostDestroyed(this));
  49     RenderProcessHostImpl::UnregisterHost(GetID());
  50   }
  51 }
  52
  53 void MockRenderProcessHost::EnableSendQueue() {
  54 }
  55
  56 bool MockRenderProcessHost::Init() {
  57   return true;
  58 }
  59
  60 int MockRenderProcessHost::GetNextRoutingID() {
  61   return ++prev_routing_id_;
  62 }
  63
  64 void MockRenderProcessHost::AddRoute(
  65     int32 routing_id,
  66     IPC::Listener* listener) {
  67   listeners_.AddWithID(listener, routing_id);
  68 }
  69
  70 void MockRenderProcessHost::RemoveRoute(int32 routing_id) {
  71   DCHECK(listeners_.Lookup(routing_id) != NULL);
  72   listeners_.Remove(routing_id);
  73   Cleanup();
  74 }
  75
  76 void MockRenderProcessHost::AddObserver(RenderProcessHostObserver* observer) {
  77   observers_.AddObserver(observer);
  78 }
  79
  80 void MockRenderProcessHost::RemoveObserver(
  81     RenderProcessHostObserver* observer) {
  82   observers_.RemoveObserver(observer);
  83 }
  84
  85 void MockRenderProcessHost::ReceivedBadMessage() {
  86   ++bad_msg_count_;
  87 }
  88
  89 void MockRenderProcessHost::WidgetRestored() {
  90 }
  91
  92 void MockRenderProcessHost::WidgetHidden() {
  93 }
  94
  95 int MockRenderProcessHost::VisibleWidgetCount() const {
  96   return 1;
  97 }
  98
  99 bool MockRenderProcessHost::IsIsolatedGuest() const {
 100   return is_isolated_guest_;
 101 }
 102
 103 StoragePartition* MockRenderProcessHost::GetStoragePartition() const {
 104   return BrowserContext::GetDefaultStoragePartition(browser_context_);
 105 }
 106
 107 void MockRenderProcessHost::AddWord(const base::string16& word) {
 108 }
 109
 110 bool MockRenderProcessHost::Shutdown(int exit_code, bool wait) {
 111   return true;
 112 }
 113
 114 bool MockRenderProcessHost::FastShutdownIfPossible() {
 115   // We aren't actually going to do anything, but set |fast_shutdown_started_|
 116   // to true so that tests know we've been called.
 117   fast_shutdown_started_ = true;
 118   return true;
 119 }
 120
 121 bool MockRenderProcessHost::FastShutdownStarted() const {
 122   return fast_shutdown_started_;
 123 }
 124
 125 void MockRenderProcessHost::DumpHandles() {
 126 }
 127
 128 base::ProcessHandle MockRenderProcessHost::GetHandle() const {
 129   // Return the current-process handle for the IPC::GetFileHandleForProcess
 130   // function.
 131   if (process_handle)
 132     return *process_handle;
 133   return base::GetCurrentProcessHandle();
 134 }
 135
 136 bool MockRenderProcessHost::Send(IPC::Message* msg) {
 137   // Save the message in the sink.
 138   sink_.OnMessageReceived(*msg);
 139   delete msg;
 140   return true;
 141 }
 142
 143 int MockRenderProcessHost::GetID() const {
 144   return id_;
 145 }
 146
 147 bool MockRenderProcessHost::HasConnection() const {
 148   return true;
 149 }
 150
 151 void MockRenderProcessHost::SetIgnoreInputEvents(bool ignore_input_events) {
 152 }
 153
 154 bool MockRenderProcessHost::IgnoreInputEvents() const {
 155   return false;
 156 }
 157
 158 void MockRenderProcessHost::Cleanup() {
 159   if (listeners_.IsEmpty()) {
 160     FOR_EACH_OBSERVER(RenderProcessHostObserver,
 161                       observers_,
 162                       RenderProcessHostDestroyed(this));
 163     base::MessageLoop::current()->DeleteSoon(FROM_HERE, this);
 164     RenderProcessHostImpl::UnregisterHost(GetID());
 165     deletion_callback_called_ = true;
 166   }
 167 }
 168
 169 void MockRenderProcessHost::AddPendingView() {
 170 }
 171
 172 void MockRenderProcessHost::RemovePendingView() {
 173 }
 174
 175 void MockRenderProcessHost::SetSuddenTerminationAllowed(bool allowed) {
 176 }
 177
 178 bool MockRenderProcessHost::SuddenTerminationAllowed() const {
 179   return true;
 180 }
 181
 182 BrowserContext* MockRenderProcessHost::GetBrowserContext() const {
 183   return browser_context_;
 184 }
 185
 186 bool MockRenderProcessHost::InSameStoragePartition(
 187     StoragePartition* partition) const {
 188   // Mock RPHs only have one partition.
 189   return true;
 190 }
 191
 192 IPC::ChannelProxy* MockRenderProcessHost::GetChannel() {
 193   return NULL;
 194 }
 195
 196 void MockRenderProcessHost::AddFilter(BrowserMessageFilter* filter) {
 197 }
 198
 199 int MockRenderProcessHost::GetActiveViewCount() {
 200   int num_active_views = 0;
 201   scoped_ptr<RenderWidgetHostIterator> widgets(
 202       RenderWidgetHost::GetRenderWidgetHosts());
 203   while (RenderWidgetHost* widget = widgets->GetNextHost()) {
 204     // Count only RenderWidgetHosts in this process.
 205     if (widget->GetProcess()->GetID() == GetID())
 206       num_active_views++;
 207   }
 208   return num_active_views;
 209 }
 210
 211 bool MockRenderProcessHost::FastShutdownForPageCount(size_t count) {
 212   if (static_cast<size_t>(GetActiveViewCount()) == count)
 213     return FastShutdownIfPossible();
 214   return false;
 215 }
 216
 217 base::TimeDelta MockRenderProcessHost::GetChildProcessIdleTime() const {
 218   return base::TimeDelta::FromMilliseconds(0);
 219 }
 220
 221 void MockRenderProcessHost::ResumeRequestsForView(int route_id) {
 222 }
 223
 224 void MockRenderProcessHost::NotifyTimezoneChange() {
 225 }
 226
 227 ServiceRegistry* MockRenderProcessHost::GetServiceRegistry() {
 228   return NULL;
 229 }
 230
 231 const base::TimeTicks& MockRenderProcessHost::GetInitTimeForNavigationMetrics()
 232     const {
 233   static base::TimeTicks dummy_time = base::TimeTicks::Now();
 234   return dummy_time;
 235 }
 236
 237 bool MockRenderProcessHost::SubscribeUniformEnabled() const {
 238   return false;
 239 }
 240
 241 void MockRenderProcessHost::OnAddSubscription(unsigned int target) {
 242 }
 243
 244 void MockRenderProcessHost::OnRemoveSubscription(unsigned int target) {
 245 }
 246
 247 void MockRenderProcessHost::SendUpdateValueState(
 248     unsigned int target, const gpu::ValueState& state) {
 249 }
 250
 251 void MockRenderProcessHost::FilterURL(bool empty_allowed, GURL* url) {
 252   RenderProcessHostImpl::FilterURL(this, empty_allowed, url);
 253 }
 254
 255 #if defined(ENABLE_WEBRTC)
 256 void MockRenderProcessHost::EnableAecDump(const base::FilePath& file) {
 257 }
 258
 259 void MockRenderProcessHost::DisableAecDump() {
 260 }
 261
 262 void MockRenderProcessHost::SetWebRtcLogMessageCallback(
 263     base::Callback<void(const std::string&)> callback) {
 264 }
 265
 266 RenderProcessHost::WebRtcStopRtpDumpCallback
 267 MockRenderProcessHost::StartRtpDump(
 268     bool incoming,
 269     bool outgoing,
 270     const WebRtcRtpPacketCallback& packet_callback) {
 271   return WebRtcStopRtpDumpCallback();
 272 }
 273 #endif
 274
 275 void MockRenderProcessHost::ResumeDeferredNavigation(
 276     const GlobalRequestID& request_id) {}
 277
 278 bool MockRenderProcessHost::OnMessageReceived(const IPC::Message& msg) {
 279   IPC::Listener* listener = listeners_.Lookup(msg.routing_id());
 280   if (listener)
 281     return listener->OnMessageReceived(msg);
 282   return false;
 283 }
 284
 285 void MockRenderProcessHost::OnChannelConnected(int32 peer_pid) {
 286 }
 287
 288 MockRenderProcessHostFactory::MockRenderProcessHostFactory() {}
 289
 290 MockRenderProcessHostFactory::~MockRenderProcessHostFactory() {
 291   // Detach this object from MockRenderProcesses to prevent STLDeleteElements()
 292   // from calling MockRenderProcessHostFactory::Remove().
 293   for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin();
 294        it != processes_.end(); ++it) {
 295     (*it)->SetFactory(NULL);
 296   }
 297 }
 298
 299 RenderProcessHost* MockRenderProcessHostFactory::CreateRenderProcessHost(
 300     BrowserContext* browser_context,
 301     SiteInstance* site_instance) const {
 302   MockRenderProcessHost* host = new MockRenderProcessHost(browser_context);
 303   if (host) {
 304     processes_.push_back(host);
 305     host->SetFactory(this);
 306   }
 307   return host;
 308 }
 309
 310 void MockRenderProcessHostFactory::Remove(MockRenderProcessHost* host) const {
 311   for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin();
 312        it != processes_.end(); ++it) {
 313     if (*it == host) {
 314       processes_.weak_erase(it);
 315       break;
 316     }
 317   }
 318 }
 319
 320 }  // content