Add AuthPolicyLoginHelper

chrome/browser/chromeos/login/login_browsertest.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 <string>
 
 #include "ash/common/shelf/wm_shelf.h"
 #include "ash/common/system/tray/system_tray.h"
 #include "ash/common/wm_window.h"
 #include "ash/shell.h"
@@ skipping 267 matching lines @@
   // Sets username and password for the Active Directory login and submits it.
   void SubmitActiveDirectoryCredentials(const std::string& username,
                                         const std::string& password) {
     js_checker().ExecuteAsync(JSElement(kAdUserInput) + ".value='" + username +
                               "'");
     js_checker().ExecuteAsync(JSElement(kAdPasswordInput) + ".value='" +
                               password + "'");
     js_checker().Evaluate(JSElement(kAdButton) + ".fire('tap')");
   }
 
+  void SetupActiveDirectoryJSNotifications() {
+    js_checker().Evaluate(
+        "var testInvalidateAd = login.GaiaSigninScreen.invalidateAd;"
+        "login.GaiaSigninScreen.invalidateAd = function(user, errorState) {"
+        "  testInvalidateAd(user, errorState);"
+        "  window.domAutomationController.setAutomationId(0);"
+        "  window.domAutomationController.send('ShowAuthError');"
+        "}");
+  }
+
+  void WaitForMessage(content::DOMMessageQueue* message_queue,
+                      const std::string& expected_message) {
+    std::string message;
+    do {
+      ASSERT_TRUE(message_queue->WaitForMessage(&message));
+    } while (message != expected_message);
+  }
+
  protected:
   // Returns string representing element with id=|element_id| inside Active
   // Directory login element.
   std::string JSElement(const std::string& element_id) {
     return "document.querySelector('#offline-ad-auth /deep/ #" + element_id +
            "')";
   }
   FakeAuthPolicyClient* fake_auth_policy_client_ = nullptr;
 
  private:
@@ skipping 129 matching lines @@
   // TestSystemTrayIsVisible();
 }
 
 // Marks as Active Directory enterprise device and OOBE as completed.
 IN_PROC_BROWSER_TEST_F(ActiveDirectoryLoginTest, PRE_LoginErrors) {
   MarkAsActiveDirectoryEnterprise();
 }
 
 // Test different UI errors for Active Directory login.
 IN_PROC_BROWSER_TEST_F(ActiveDirectoryLoginTest, LoginErrors) {
+  SetupActiveDirectoryJSNotifications();
   TestLoginVisible();
   TestDomainVisible();
 
+  content::DOMMessageQueue message_queue;
+
   SubmitActiveDirectoryCredentials("", "");
   TestUserError();
   TestDomainVisible();
 
   SubmitActiveDirectoryCredentials(kTestActiveDirectoryUser, "");
   TestPasswordError();
   TestDomainVisible();
 
-  fake_auth_policy_client_->set_auth_error(authpolicy::ERROR_BAD_USER_NAME);
   SubmitActiveDirectoryCredentials(std::string(kTestActiveDirectoryUser) + "@",
                                    kPassword);
   TestUserError();
   TestDomainHidden();
 
+  fake_auth_policy_client_->set_auth_error(authpolicy::ERROR_BAD_USER_NAME);
+  SubmitActiveDirectoryCredentials(
+      std::string(kTestActiveDirectoryUser) + "@" + kTestRealm, kPassword);
+  WaitForMessage(&message_queue, "\"ShowAuthError\"");
+  TestUserError();
+  TestDomainVisible();
+
   fake_auth_policy_client_->set_auth_error(authpolicy::ERROR_BAD_PASSWORD);
   SubmitActiveDirectoryCredentials(kTestActiveDirectoryUser, kPassword);
+  WaitForMessage(&message_queue, "\"ShowAuthError\"");
   TestPasswordError();
   TestDomainVisible();
 }
 
 }  // namespace chromeos