Add WebWidgetTestProxy.

content/shell/renderer/layout_test/layout_test_content_renderer_client.cc

 // Copyright (c) 2014 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/shell/renderer/layout_test/layout_test_content_renderer_client.h"
 
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/debug/debugger.h"
 #include "components/test_runner/mock_credential_manager_client.h"
@@ skipping 11 matching lines @@
 #include "content/shell/renderer/layout_test/blink_test_helpers.h"
 #include "content/shell/renderer/layout_test/blink_test_runner.h"
 #include "content/shell/renderer/layout_test/layout_test_render_frame_observer.h"
 #include "content/shell/renderer/layout_test/layout_test_render_thread_observer.h"
 #include "content/shell/renderer/layout_test/test_media_stream_renderer_factory.h"
 #include "content/shell/renderer/shell_render_view_observer.h"
 #include "content/test/mock_webclipboard_impl.h"
 #include "ppapi/shared_impl/ppapi_switches.h"
 #include "third_party/WebKit/public/platform/WebMediaStreamCenter.h"
 #include "third_party/WebKit/public/platform/modules/app_banner/WebAppBannerClient.h"
+#include "third_party/WebKit/public/web/WebFrameWidget.h"
 #include "third_party/WebKit/public/web/WebPluginParams.h"
 #include "third_party/WebKit/public/web/WebTestingSupport.h"
 #include "third_party/WebKit/public/web/WebView.h"
 #include "v8/include/v8.h"
 
 using blink::WebAudioDevice;
 using blink::WebClipboard;
 using blink::WebFrame;
 using blink::WebLocalFrame;
 using blink::WebMIDIAccessor;
@@ skipping 23 matching lines @@
   static bool first_test_runner = true;
   if (first_test_runner) {
     first_test_runner = false;
     interfaces->SetDelegate(test_runner);
   }
 
   proxy->set_delegate(test_runner);
   proxy->set_view_test_client(LayoutTestRenderThreadObserver::GetInstance()
                                   ->test_interfaces()
                                   ->CreateWebViewTestClient(proxy));
-  proxy->set_widget_test_client(LayoutTestRenderThreadObserver::GetInstance()
-                                    ->test_interfaces()
-                                    ->CreateWebWidgetTestClient(proxy));
+  std::unique_ptr<test_runner::WebWidgetTestClient> widget_test_client =
+      LayoutTestRenderThreadObserver::GetInstance()
+          ->test_interfaces()
+          ->CreateWebWidgetTestClient(proxy);
+  widget_test_client->set_web_view_test_proxy_base(proxy);
+  proxy->set_widget_test_client(std::move(widget_test_client));
   proxy->SetInterfaces(interfaces);
 }
 
+void WebWidgetTestProxyCreated(blink::WebWidget* web_widget,
+                               test_runner::WebWidgetTestProxyBase* proxy) {
+  CHECK(web_widget->isWebFrameWidget());
+  proxy->set_web_widget(web_widget);
+  blink::WebFrameWidget* web_frame_widget =
+      static_cast<blink::WebFrameWidget*>(web_widget);
+  blink::WebView* web_view = web_frame_widget->localRoot()->view();
+  RenderView* render_view = RenderView::FromWebView(web_view);
+  test_runner::WebViewTestProxyBase* view_proxy =
+      GetWebViewTestProxyBase(render_view);
+  std::unique_ptr<test_runner::WebWidgetTestClient> widget_test_client =
+      LayoutTestRenderThreadObserver::GetInstance()
+          ->test_interfaces()
+          ->CreateWebWidgetTestClient(proxy);
+  widget_test_client->set_web_view_test_proxy_base(view_proxy);
+  proxy->set_widget_test_client(std::move(widget_test_client));
+}
+
 void WebFrameTestProxyCreated(RenderFrame* render_frame,
                               test_runner::WebFrameTestProxyBase* proxy) {
   test_runner::WebViewTestProxyBase* web_view_test_proxy_base =
       GetWebViewTestProxyBase(render_frame->GetRenderView());
   proxy->set_test_client(
       LayoutTestRenderThreadObserver::GetInstance()
           ->test_interfaces()
           ->CreateWebFrameTestClient(web_view_test_proxy_base, proxy));
 }
 
 }  // namespace
 
 LayoutTestContentRendererClient::LayoutTestContentRendererClient() {
   EnableWebTestProxyCreation(base::Bind(&WebViewTestProxyCreated),
+                             base::Bind(&WebWidgetTestProxyCreated),
                              base::Bind(&WebFrameTestProxyCreated));
 }
 
 LayoutTestContentRendererClient::~LayoutTestContentRendererClient() {
 }
 
 void LayoutTestContentRendererClient::RenderThreadStarted() {
   ShellContentRendererClient::RenderThreadStarted();
   shell_observer_.reset(new LayoutTestRenderThreadObserver());
 }
@@ skipping 97 matching lines @@
   return nullptr;
 #endif
 }
 
 void LayoutTestContentRendererClient::DidInitializeWorkerContextOnWorkerThread(
     v8::Local<v8::Context> context) {
   blink::WebTestingSupport::injectInternalsObject(context);
 }
 
 }  // namespace content