InstantExtended: Send search URLs to renderers.

content/public/test/mock_render_thread.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/public/test/mock_render_thread.h"
 
 #include "base/message_loop/message_loop_proxy.h"
 #include "content/common/view_messages.h"
+#include "content/public/renderer/render_process_observer.h"
 #include "ipc/ipc_message_utils.h"
 #include "ipc/ipc_sync_message.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/web/WebScriptController.h"
 
 namespace content {
 
 MockRenderThread::MockRenderThread()
     : routing_id_(0),
       surface_id_(0),
@@ skipping 28 matching lines @@
       reply_deserializer_->SerializeOutputParameters(*msg);
       reply_deserializer_.reset();
     }
   } else {
     if (msg->is_sync()) {
       // We actually need to handle deleting the reply deserializer for sync
       // messages.
       reply_deserializer_.reset(
           static_cast<IPC::SyncMessage*>(msg)->GetReplyDeserializer());
     }
-    OnMessageReceived(*msg);
+    if (msg->routing_id() == MSG_ROUTING_CONTROL)
+      OnControlMessageReceived(*msg);
+    else
+      OnMessageReceived(*msg);
   }
   delete msg;
   return true;
 }
 
 base::MessageLoop* MockRenderThread::GetMessageLoop() {
   return NULL;
 }
 
 IPC::SyncChannel* MockRenderThread::GetChannel() {
@@ skipping 49 matching lines @@
     }
   }
   NOTREACHED() << "filter to be removed not found";
 }
 
 void MockRenderThread::SetOutgoingMessageFilter(
     IPC::ChannelProxy::OutgoingMessageFilter* filter) {
 }
 
 void MockRenderThread::AddObserver(RenderProcessObserver* observer) {
+  observers_.AddObserver(observer);
 }
 
 void MockRenderThread::RemoveObserver(RenderProcessObserver* observer) {
+  observers_.RemoveObserver(observer);
 }
 
 void MockRenderThread::SetResourceDispatcherDelegate(
     ResourceDispatcherDelegate* delegate) {
 }
 
 void MockRenderThread::WidgetHidden() {
 }
 
 void MockRenderThread::WidgetRestored() {
@@ skipping 79 matching lines @@
     int* route_id,
     int* main_frame_route_id,
     int* surface_id,
     int64* cloned_session_storage_namespace_id) {
   *route_id = new_window_routing_id_;
   *main_frame_route_id = new_window_main_frame_routing_id_;
   *surface_id = surface_id_;
   *cloned_session_storage_namespace_id = 0;
 }
 
+bool MockRenderThread::OnControlMessageReceived(const IPC::Message& msg) {
+  ObserverListBase<RenderProcessObserver>::Iterator it(observers_);
+  RenderProcessObserver* observer;
+  while ((observer = it.GetNext()) != NULL) {
+    if (observer->OnControlMessageReceived(msg))
+      return true;
+  }
+  return OnMessageReceived(msg);
+}
+
 bool MockRenderThread::OnMessageReceived(const IPC::Message& msg) {
   // Save the message in the sink.
   sink_.OnMessageReceived(msg);
 
   bool handled = true;
   bool msg_is_ok = true;
   IPC_BEGIN_MESSAGE_MAP_EX(MockRenderThread, msg, msg_is_ok)
     IPC_MESSAGE_HANDLER(ViewHostMsg_CreateWidget, OnCreateWidget)
     IPC_MESSAGE_HANDLER(ViewHostMsg_CreateWindow, OnCreateWindow)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP_EX()
   return handled;
 }
 
 #if defined(OS_WIN)
 void MockRenderThread::OnDuplicateSection(
     base::SharedMemoryHandle renderer_handle,
     base::SharedMemoryHandle* browser_handle) {
   // We don't have to duplicate the input handles since RenderViewTest does not
   // separate a browser process from a renderer process.
   *browser_handle = renderer_handle;
 }
 #endif  // defined(OS_WIN)
 
 }  // namespace content