Enable new sign in flow for E2E tests.

chrome/browser/ui/webui/signin/login_ui_test_utils.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/browser/signin/signin_promo.h"
 #include "chrome/browser/signin/signin_tracker_factory.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/webui/signin/inline_login_ui.h"
 #include "chrome/test/base/ui_test_utils.h"
@@ skipping 66 matching lines @@
   bool signed_in_;
   scoped_refptr<MessageLoopRunner> message_loop_runner_;
 };
 
 }  // anonymous namespace
 
 
 namespace login_ui_test_utils {
 
 void WaitUntilUIReady(Browser* browser) {
-  content::DOMMessageQueue message_queue;
-  ASSERT_TRUE(content::ExecuteScript(
+  std::string message;
+  ASSERT_TRUE(content::ExecuteScriptAndExtractString(
       browser->tab_strip_model()->GetActiveWebContents(),
       "if (!inline.login.getAuthExtHost())"
       "  inline.login.initialize();"
       "var handler = function() {"
-      "  window.domAutomationController.setAutomationId(0);"
       "  window.domAutomationController.send('ready');"
       "};"
       "if (inline.login.isAuthReady())"
       "  handler();"
       "else"
-      "  inline.login.getAuthExtHost().addEventListener('ready', handler);"));
-
-  std::string message;
-  do {
-    ASSERT_TRUE(message_queue.WaitForMessage(&message));
-  } while (message != "\"ready\"");
+      "  inline.login.getAuthExtHost().addEventListener('ready', handler);",
+      &message));
+  ASSERT_EQ("ready", message);
 }
 
-void ExecuteJsToSigninInSigninFrame(Browser* browser,
-                                    const std::string& email,
-                                    const std::string& password) {
-   std::string js =
+void WaitUntilElementExistsInSigninFrame(Browser* browser,
+                                         const std::string& element_id) {
+  std::string message;
+  std::string js =
+      "function WaitForElementById(elementId) {"
+      "  var retries = 10; /* 10 seconds. */"
+      "  function CheckelementExists() {"
+      "    if (document.getElementById(elementId) != null) {"
+      "      window.domAutomationController.send('found');"
+      "    } else if (retries > 0) { "
+      "      retries--;"
+      "      window.setTimeout(CheckelementExists, 1000);"
+      "    } else {"
+      "      window.domAutomationController.send('failed');"
+      "    }"
+      "  }"
+      "  CheckelementExists();"
+      "}"
+      "WaitForElementById('" + element_id + "');";
+  content::WebContents* web_contents =
+      browser->tab_strip_model()->GetActiveWebContents();
+  ASSERT_TRUE(content::ExecuteScriptAndExtractString(
+      InlineLoginUI::GetAuthFrame(web_contents, GURL(), "signin-frame"),
+      js, &message));
+
+  ASSERT_EQ("found", message) <<
+      "Failed to find element with id " << element_id;
+}
+
+bool ElementExistsInSigninFrame(Browser* browser,
+                                const std::string& element_id) {
+  content::WebContents* web_contents =
+      browser->tab_strip_model()->GetActiveWebContents();
+  bool result = false;
+  EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
+      InlineLoginUI::GetAuthFrame(web_contents, GURL(), "signin-frame"),
+      "window.domAutomationController.send("
+      "  document.getElementById('" + element_id + "') != null);",
+      &result));
+  return result;
+}
+
+void SigninInNewGaiaFlow(Browser* browser,
+                         const std::string& email,
+                         const std::string& password) {
+  std::string js = "document.getElementById('Email').value = '" + email + "';"
+                   "document.getElementById('next').click();";
+
+  content::WebContents* web_contents =
+      browser->tab_strip_model()->GetActiveWebContents();
+  ASSERT_TRUE(content::ExecuteScript(InlineLoginUI::GetAuthFrame(
+      web_contents, GURL(), "signin-frame"), js));
+
+  WaitUntilElementExistsInSigninFrame(browser, "Passwd");
+  js = "document.getElementById('Passwd').value = '" + password + "';"
+       "document.getElementById('signIn').click();";
+
+  ASSERT_TRUE(content::ExecuteScript(InlineLoginUI::GetAuthFrame(
+      web_contents, GURL(), "signin-frame"), js));
+}
+
+void SigninInOldGaiaFlow(Browser* browser,
+                         const std::string& email,
+                         const std::string& password) {
+  std::string js =
       "document.getElementById('Email').value = '" + email + "';"
       "document.getElementById('Passwd').value = '" + password + "';"
       "document.getElementById('signIn').click();";
 
   content::WebContents* web_contents =
       browser->tab_strip_model()->GetActiveWebContents();
   ASSERT_TRUE(content::ExecuteScript(InlineLoginUI::GetAuthFrame(
       web_contents, GURL(), "signin-frame"), js));
 }
 
+void ExecuteJsToSigninInSigninFrame(Browser* browser,
+                                    const std::string& email,
+                                    const std::string& password) {
+  WaitUntilElementExistsInSigninFrame(browser, "Email");
+  if (ElementExistsInSigninFrame(browser, "next"))
+    SigninInNewGaiaFlow(browser, email, password);
+  else
+    SigninInOldGaiaFlow(browser, email, password);
+}
+
 bool SignInWithUI(Browser* browser,
                   const std::string& username,
                   const std::string& password) {
 
   SignInObserver signin_observer;
   scoped_ptr<SigninTracker> tracker =
       SigninTrackerFactory::CreateForProfile(browser->profile(),
                                              &signin_observer);
 
   GURL signin_url = signin::GetPromoURL(
@@ skipping 11 matching lines @@
 
   DVLOG(1) << "Wait for login UI to be ready.";
   WaitUntilUIReady(browser);
   DVLOG(1) << "Sign in user: " << username;
   ExecuteJsToSigninInSigninFrame(browser, username, password);
   signin_observer.Wait();
   return signin_observer.DidSignIn();
 }
 
 }  // namespace login_ui_test_utils