Add WaitForInterstitialReady to browser_test_utils.

content/public/test/browser_test_utils.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/browser_test_utils.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/process/kill.h"
 #include "base/rand_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "base/values.h"
 #include "content/browser/renderer_host/render_widget_host_impl.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/browser/web_contents/web_contents_view.h"
 #include "content/common/input/synthetic_web_input_event_builders.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/dom_operation_notification_details.h"
 #include "content/public/browser/histogram_fetcher.h"
+#include "content/public/browser/interstitial_page.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/test/test_utils.h"
 #include "net/base/filename_util.h"
@@ skipping 682 matching lines @@
       new content::MessageLoopRunner);
   InterstitialObserver observer(web_contents,
                                 base::Closure(),
                                 loop_runner->QuitClosure());
   if (!task.is_null())
     task.Run();
   // At this point, web_contents may have been deleted.
   loop_runner->Run();
 }
 
+bool WaitForInterstitialReady(content::InterstitialPage* interstitial) {
+  content::RenderFrameHost* rfh = interstitial->GetMainFrame();

[mattm, 2015/01/22 02:28:32]

Since the rest of this function isn't interstitial specific, would it be useful
to just take a RenderFrameHost and call it WaitForRenderFrameReady or something
like that?

[meacer, 2015/01/22 03:18:07]

Done.

+  if (!rfh)
+    return false;
+  bool load_complete = false;
+  EXPECT_TRUE(
+      content::ExecuteScriptAndExtractBool(
+          rfh,
+          "(function() {"
+          "  var done = false;"
+          "  function checkState() {"
+          "    if (!done && document.readyState == 'complete') {"
+          "      done = true;"
+          "      window.domAutomationController.send(true);"
+          "    }"
+          "  }"
+          "  checkState();"
+          "  document.addEventListener('readystatechange', checkState);"
+          "})();",
+          &load_complete));
+  return load_complete;
+}
+
 TitleWatcher::TitleWatcher(WebContents* web_contents,
                            const base::string16& expected_title)
     : WebContentsObserver(web_contents),
       message_loop_runner_(new MessageLoopRunner) {
   EXPECT_TRUE(web_contents != NULL);
   expected_titles_.push_back(expected_title);
 }
 
 void TitleWatcher::AlsoWaitForTitle(const base::string16& expected_title) {
   expected_titles_.push_back(expected_title);
@@ skipping 181 matching lines @@
 
 bool WebContentsAddedObserver::RenderViewCreatedCalled() {
   if (child_observer_) {
     return child_observer_->render_view_created_called_ &&
            child_observer_->main_frame_created_called_;
   }
   return false;
 }
 
 }  // namespace content