Setting TestRunner's main window from BlinkTestRunner.

components/test_runner/test_interfaces.cc

 // Copyright 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 "components/test_runner/test_interfaces.h"
 
 #include <stddef.h>
 
 #include <string>
 
@@ skipping 11 matching lines @@
 #include "third_party/WebKit/public/platform/WebURL.h"
 #include "third_party/WebKit/public/web/WebCache.h"
 #include "third_party/WebKit/public/web/WebKit.h"
 #include "third_party/WebKit/public/web/WebView.h"
 
 namespace test_runner {
 
 TestInterfaces::TestInterfaces()
     : test_runner_(new TestRunner(this)),
       delegate_(nullptr),
-      app_banner_client_(nullptr) {
+      app_banner_client_(nullptr),
+      main_view_(nullptr) {
   blink::setLayoutTestMode(true);
   // NOTE: please don't put feature specific enable flags here,
   // instead add them to RuntimeEnabledFeatures.in
 
   ResetAll();
 }
 
 TestInterfaces::~TestInterfaces() {
   // gamepad_controller_ doesn't depend on WebView.
-  test_runner_->SetWebView(nullptr);
+  test_runner_->SetMainView(nullptr);
 
   // gamepad_controller_ ignores SetDelegate(nullptr)
   test_runner_->SetDelegate(nullptr);
 }
 
-void TestInterfaces::SetWebView(blink::WebView* web_view,
-                                WebTestProxyBase* proxy) {
+void TestInterfaces::SetMainView(blink::WebView* web_view) {
   // gamepad_controller_ doesn't depend on WebView.
-  test_runner_->SetWebView(web_view);
+  main_view_ = web_view;
+  test_runner_->SetMainView(web_view);
 }
 
 void TestInterfaces::SetDelegate(WebTestDelegate* delegate) {
   gamepad_controller_ = GamepadController::Create(delegate);
   test_runner_->SetDelegate(delegate);
   delegate_ = delegate;
 }
 
 void TestInterfaces::BindTo(blink::WebFrame* frame) {
   if (gamepad_controller_)
@@ skipping 60 matching lines @@
 }
 
 void TestInterfaces::WindowClosed(WebTestProxyBase* proxy) {
   std::vector<WebTestProxyBase*>::iterator pos =
       std::find(window_list_.begin(), window_list_.end(), proxy);
   if (pos == window_list_.end()) {
     NOTREACHED();
     return;
   }
   window_list_.erase(pos);
+
+  if (proxy->web_view() == main_view_)
+    SetMainView(nullptr);
 }
 
 TestRunner* TestInterfaces::GetTestRunner() {
   return test_runner_.get();
 }
 
 WebTestDelegate* TestInterfaces::GetDelegate() {
   return delegate_;
 }
 
 const std::vector<WebTestProxyBase*>& TestInterfaces::GetWindowList() {
   return window_list_;
 }
 
 blink::WebThemeEngine* TestInterfaces::GetThemeEngine() {
   if (!test_runner_->UseMockTheme())
     return 0;
   if (!theme_engine_.get())
     theme_engine_.reset(new MockWebThemeEngine());
   return theme_engine_.get();
 }
 
 AppBannerClient* TestInterfaces::GetAppBannerClient() {
   return app_banner_client_;
 }
 
 }  // namespace test_runner