Port identity_internals to mojo

chrome/test/base/web_ui_browser_test.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 "chrome/test/base/web_ui_browser_test.h"
 
 #include <string>
 #include <vector>
 
 #include "base/lazy_instance.h"
 #include "base/memory/ref_counted_memory.h"
+#include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/strings/string_util.h"
 #include "base/values.h"
 #include "chrome/browser/chrome_content_browser_client.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/webui/web_ui_test_handler.h"
 #include "chrome/common/chrome_paths.h"
@@ skipping 53 matching lines @@
     browser_test_->PreLoadJavascriptLibraries(
         preload_test_fixture_, preload_test_name_, rvh);
   }
 
  private:
   WebUIBrowserTest* browser_test_;
   std::string preload_test_fixture_;
   std::string preload_test_name_;
 };
 
+class WebUIMainObserver : public content::WebContentsObserver {
+ public:
+  explicit WebUIMainObserver(content::WebContents* web_contents)
+      : content::WebContentsObserver(web_contents),
+        message_loop_runner_(new content::MessageLoopRunner),
+        has_fired_(false) {
+  }
+
+  void Wait() {
+    if (has_fired_)
+      return;
+
+    message_loop_runner_->Run();
+  }
+
+  virtual void DidRunWebUIMojoMain() OVERRIDE {
+    has_fired_ = true;
+    message_loop_runner_->Quit();
+  }
+
+ private:
+  // The MessageLoopRunner used to spin the message loop.
+  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
+
+  bool has_fired_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebUIMainObserver);
+};
+
 }  // namespace
 
 WebUIBrowserTest::~WebUIBrowserTest() {
 }
 
 bool WebUIBrowserTest::RunJavascriptFunction(const std::string& function_name) {
   ConstValueVector empty_args;
   return RunJavascriptFunction(function_name, empty_args);
 }
 
@@ skipping 118 matching lines @@
   WebUIJsInjectionReadyObserver injection_observer(
       web_contents, this, preload_test_fixture_, preload_test_name_);
   content::TestNavigationObserver navigation_observer(web_contents);
   chrome::NavigateParams params(
       browser(), GURL(browse_to), content::PAGE_TRANSITION_TYPED);
   params.disposition = CURRENT_TAB;
   chrome::Navigate(&params);
   navigation_observer.Wait();
 }
 
+void WebUIBrowserTest::BrowsePreloadAndWaitForMain(const GURL& browse_to) {
+  content::WebContents* web_contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  WebUIMainObserver webui_main_observer(web_contents);
+  BrowsePreload(browse_to);
+  webui_main_observer.Wait();
+}
+
 #if defined(ENABLE_FULL_PRINTING)
 
 // This custom ContentBrowserClient is used to get notified when a WebContents
 // for the print preview dialog gets created.
 class PrintContentBrowserClient : public chrome::ChromeContentBrowserClient {
  public:
   PrintContentBrowserClient(WebUIBrowserTest* browser_test,
                             const std::string& preload_test_fixture,
                             const std::string& preload_test_name)
       : browser_test_(browser_test),
@@ skipping 243 matching lines @@
 }
 
 GURL WebUIBrowserTest::WebUITestDataPathToURL(
     const base::FilePath::StringType& path) {
   base::FilePath dir_test_data;
   EXPECT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &dir_test_data));
   base::FilePath test_path(dir_test_data.Append(kWebUITestFolder).Append(path));
   EXPECT_TRUE(base::PathExists(test_path));
   return net::FilePathToFileURL(test_path);
 }