Setting TestRunner's main window from BlinkTestRunner.

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 35 matching lines @@
 using blink::WebRTCPeerConnectionHandler;
 using blink::WebRTCPeerConnectionHandlerClient;
 using blink::WebThemeEngine;
 
 namespace content {
 
 namespace {
 
 void WebTestProxyCreated(RenderView* render_view,
                          test_runner::WebTestProxyBase* proxy) {
+  test_runner::WebTestInterfaces* interfaces =
+      LayoutTestRenderThreadObserver::GetInstance()->test_interfaces();
+
   BlinkTestRunner* test_runner = new BlinkTestRunner(render_view);
+  // TODO(lukasza): Using the 1st BlinkTestRunner as the main delegate is wrong,
+  // but it is difficult to change because this behavior has been baked for a
+  // long time into test assumptions (i.e. which PrintMessage gets delivered to
+  // the browser depends on this).
+  static bool first_test_runner = true;
+  if (first_test_runner) {
+    first_test_runner = false;
+    interfaces->SetDelegate(test_runner);
+  }
+
   proxy->set_delegate(test_runner);
-
-  if (!LayoutTestRenderThreadObserver::GetInstance()->test_delegate()) {
-    LayoutTestRenderThreadObserver::GetInstance()->SetTestDelegate(
-        test_runner);
-  }
   proxy->set_view_test_client(LayoutTestRenderThreadObserver::GetInstance()
                                   ->test_interfaces()
                                   ->CreateWebViewTestClient(proxy));
-  proxy->SetInterfaces(
-      LayoutTestRenderThreadObserver::GetInstance()->test_interfaces());
+  proxy->SetInterfaces(interfaces);
 }
 
 void WebFrameTestProxyCreated(RenderFrame* render_frame,
                               test_runner::WebFrameTestProxyBase* proxy) {
   test_runner::WebTestProxyBase* web_test_proxy_base =
       GetWebTestProxyBase(render_frame->GetRenderView());
   proxy->set_test_client(LayoutTestRenderThreadObserver::GetInstance()
                              ->test_interfaces()
                              ->CreateWebFrameTestClient(web_test_proxy_base));
 }
@@ skipping 28 matching lines @@
   proxy->Reset();
   proxy->SetSendWheelGestures(UseGestureBasedWheelScrolling());
 
   BlinkTestRunner* test_runner = BlinkTestRunner::Get(render_view);
   test_runner->Reset(false /* for_new_test */);
 
   LayoutTestRenderThreadObserver::GetInstance()
       ->test_interfaces()
       ->TestRunner()
       ->InitializeWebViewWithMocks(render_view->GetWebView());
-
-  test_runner::WebTestDelegate* delegate =
-      LayoutTestRenderThreadObserver::GetInstance()->test_delegate();
-  if (delegate == static_cast<test_runner::WebTestDelegate*>(test_runner)) {
-    // TODO(lukasza): Should this instead by done by BlinkTestRunner,
-    // when it gets notified by the browser that it is the main window?
-
-    // Let test_runner layer know what is the main test window.
-    LayoutTestRenderThreadObserver::GetInstance()
-        ->test_interfaces()
-        ->SetWebView(render_view->GetWebView(), proxy);
-  }
 }
 
 WebMediaStreamCenter*
 LayoutTestContentRendererClient::OverrideCreateWebMediaStreamCenter(
     WebMediaStreamCenterClient* client) {
 #if defined(ENABLE_WEBRTC)
   test_runner::WebTestInterfaces* interfaces =
       LayoutTestRenderThreadObserver::GetInstance()->test_interfaces();
   return interfaces->CreateMediaStreamCenter(client);
 #else
@@ skipping 53 matching lines @@
 LayoutTestContentRendererClient::CreateMediaStreamRendererFactory() {
 #if defined(ENABLE_WEBRTC)
   return std::unique_ptr<MediaStreamRendererFactory>(
       new TestMediaStreamRendererFactory());
 #else
   return nullptr;
 #endif
 }
 
 }  // namespace content