PlzNavigate: Eliminate test_runner bailing out of layout tests too early

components/test_runner/test_runner.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_runner.h"
 
 #include <stddef.h>
 #include <limits>
 #include <utility>
 
@@ skipping 1537 matching lines @@
 TestRunner::TestRunner(TestInterfaces* interfaces)
     : test_is_running_(false),
       close_remaining_windows_(false),
       work_queue_(this),
       web_history_item_count_(0),
       test_interfaces_(interfaces),
       delegate_(nullptr),
       main_view_(nullptr),
       mock_content_settings_client_(
           new MockContentSettingsClient(&layout_test_runtime_flags_)),
+      will_navigate_(false),
       credential_manager_client_(new MockCredentialManagerClient),
       mock_screen_orientation_client_(new MockScreenOrientationClient),
       spellcheck_(new SpellCheckClient(this)),
       chooser_count_(0),
       previously_focused_view_(nullptr),
       is_web_platform_tests_mode_(false),
       weak_factory_(this) {}
 
 TestRunner::~TestRunner() {}
 
@@ skipping 11 matching lines @@
   if (speech_recognizer_)
     speech_recognizer_->SetDelegate(delegate);
 }
 
 void TestRunner::SetMainView(WebView* web_view) {
   main_view_ = web_view;
 }
 
 void TestRunner::Reset() {
   is_web_platform_tests_mode_ = false;
+  will_navigate_ = false;
   top_loading_frame_ = nullptr;
   layout_test_runtime_flags_.Reset();
   mock_screen_orientation_client_->ResetData();
   drag_image_.reset();
   widgets_with_scheduled_animations_.clear();
 
   WebSecurityPolicy::resetOriginAccessWhitelists();
 #if defined(__linux__) || defined(ANDROID)
   WebFontRendering::setSubpixelPositioning(false);
 #endif
@@ skipping 230 matching lines @@
 }
 
 const std::set<std::string>* TestRunner::httpHeadersToClear() const {
   return &http_headers_to_clear_;
 }
 
 bool TestRunner::IsFramePartOfMainTestWindow(blink::WebFrame* frame) const {
   return test_is_running_ && frame->top()->view() == main_view_;
 }
 
+void TestRunner::OnNavigationBegin(WebFrame* frame) {
+  if (IsFramePartOfMainTestWindow(frame))
+    will_navigate_ = true;
+}
+
 bool TestRunner::tryToSetTopLoadingFrame(WebFrame* frame) {
+  will_navigate_ = false;
   if (!IsFramePartOfMainTestWindow(frame))
     return false;
 
   if (top_loading_frame_ || layout_test_runtime_flags_.have_top_loading_frame())
     return false;
 
   top_loading_frame_ = frame;
   layout_test_runtime_flags_.set_have_top_loading_frame(true);
   OnLayoutTestRuntimeFlagsChanged();
   return true;
 }
 
 bool TestRunner::tryToClearTopLoadingFrame(WebFrame* frame) {
+  will_navigate_ = false;
   if (!IsFramePartOfMainTestWindow(frame))
     return false;
 
   if (frame != top_loading_frame_)
     return false;
 
   top_loading_frame_ = nullptr;
   DCHECK(layout_test_runtime_flags_.have_top_loading_frame());
   layout_test_runtime_flags_.set_have_top_loading_frame(false);
   OnLayoutTestRuntimeFlagsChanged();
@@ skipping 865 matching lines @@
   if (mimeType != "text/plain")
     return;
 
   layout_test_runtime_flags_.set_dump_as_text(true);
   layout_test_runtime_flags_.set_generate_pixel_results(false);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
 void TestRunner::NotifyDone() {
   if (layout_test_runtime_flags_.wait_until_done() && !topLoadingFrame() &&
-      work_queue_.is_empty())
+      !will_navigate_ && work_queue_.is_empty())
     delegate_->TestFinished();
   layout_test_runtime_flags_.set_wait_until_done(false);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
 }  // namespace test_runner