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 67 matching lines @@
   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(

[shadi, 2015/04/23 18:17:11]

ExecuteScript is a bad use here since it already adds a
domAutomationController.send(0) to the default script:
https://code.google.com/p/chromium/codesearch#chromium/src/content/public/tes...

       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));

[shadi, 2015/04/23 18:17:11]

Using ExecuteScriptAndExtractString should return only one message back and make
this more stable here and below.

   } while (message != "\"ready\"");
 }
 
-void ExecuteJsToSigninInSigninFrame(Browser* browser,
-                                    const std::string& email,
-                                    const std::string& password) {
-   std::string js =
+void WaitUntilElementExistsInSigninFrame(Browser* browser,
+                                         const std::string& element_id) {
+  content::DOMMessageQueue message_queue;
+  std::string js =
+      "function WaitForElementById(elementId) {"
+      "  var retries = 10; /* 10 seconds. */"
+      "  function CheckelementExists() {"
+      "    if (document.getElementById(elementId) != null) {"
+      "      window.domAutomationController.setAutomationId(0);"
+      "      window.domAutomationController.send('found');"
+      "    } else if (retries > 0) { "
+      "      retries--;"
+      "      window.setTimeout(CheckelementExists, 1000);"
+      "    } else {"
+      "      window.domAutomationController.setAutomationId(0);"
+      "      window.domAutomationController.send('failed');"
+      "    }"
+      "  }"
+      "  CheckelementExists();"
+      "}"
+      "WaitForElementById('" + element_id + "');";
+  content::WebContents* web_contents =
+      browser->tab_strip_model()->GetActiveWebContents();
+  ASSERT_TRUE(content::ExecuteScript(InlineLoginUI::GetAuthFrame(
+      web_contents, GURL(), "signin-frame"), js));
+
+  std::string message;
+  do {
+    ASSERT_TRUE(message_queue.WaitForMessage(&message));
+    if (message == "\"failed\"")
+      LOG(ERROR) << "Failed to find element with id " << element_id;
+  } while (message != "\"found\"");
+}
+
+bool ElementExistsInSigninFrame(Browser* browser,
+                                const std::string& element_id) {
+  content::DOMMessageQueue message_queue;
+  std::string js =
+      "window.domAutomationController.setAutomationId(0);"
+      "if (document.getElementById('" + element_id + "') != null) {"
+      "  window.domAutomationController.send('found');"
+      "} else { "
+      "  window.domAutomationController.send('not found');"
+      "}";
+  content::WebContents* web_contents =
+      browser->tab_strip_model()->GetActiveWebContents();
+  EXPECT_TRUE(content::ExecuteScript(InlineLoginUI::GetAuthFrame(
+      web_contents, GURL(), "signin-frame"), js));
+
+  std::string message;
+  EXPECT_TRUE(message_queue.WaitForMessage(&message));
+  return message == "\"found\"";
+}
+
+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