Add enterprise enrollment support for fake users.

chrome/browser/resources/chromeos/login/login_shared.js

 // Copyright 2016 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.
 
 /**
  * @fileoverview Common OOBE controller methods. This method is shared between
  * OOBE, login, and the lock screen. Add only methods that need to be shared
  * between all *three* screens here, as each additional method increases the
  * time it takes to show the lock screen.
  *
@@ skipping 73 matching lines @@
    * @param {boolean} show True if version label should be visible.
    */
   Oobe.showVersion = function(show) {
     Oobe.getInstance().showVersion(show);
   };
 
   /**
    * Update body class to switch between OOBE UI and Login UI.
    */
   Oobe.showOobeUI = function(showOobe) {
-    console.log('Oobe.showOobeUI(' + showOobe + ')');
     if (showOobe) {
       document.body.classList.add('oobe-display');
 
       // Callback to animate the header bar in.
       var showHeaderBar = function() {
         login.HeaderBar.animateIn(false, function() {
           chrome.send('headerBarVisible');
         });
       };
       // Start asynchronously so the OOBE network screen comes in first.
@@ skipping 174 matching lines @@
    */
   Oobe.skipToLoginForTesting = function() {
     Oobe.disableSigninUI();
     chrome.send('skipToLoginForTesting');
   };
 
   /**
    * Login for telemetry.
    * @param {string} username Login username.
    * @param {string} password Login password.
+   * @param {boolean} enterpriseEnroll Login as an enterprise enrollment?
    */
-  Oobe.loginForTesting = function(username, password, gaia_id) {
+  Oobe.loginForTesting = function(username, password, gaia_id,
+                                  enterpriseEnroll = false) {
+    // Helper method that runs |fn| after |screenName| is visible.
+    function waitForOobeScreen(screenName, fn) {
+      if (Oobe.getInstance().currentScreen &&
+          Oobe.getInstance().currentScreen.id === screenName) {
+        fn();
+      } else {
+        $('oobe').addEventListener('screenchanged', function handler(e) {
+          if (e.detail == screenName) {
+            $('oobe').removeEventListener('screenchanged', handler);
+            fn();
+          }
+        });
+      }
+    }
+
     Oobe.disableSigninUI();
     chrome.send('skipToLoginForTesting', [username]);
-    chrome.send('completeLogin', [gaia_id, username, password, false]);
+
+    if (!enterpriseEnroll) {
+      chrome.send('completeLogin', [gaia_id, username, password, false]);
+    } else {
+      waitForOobeScreen('gaia-signin', function() {
+        chrome.send('toggleEnrollmentScreen');
+        chrome.send('toggleFakeEnrollment');
+      });
+
+      waitForOobeScreen('oauth-enrollment', function() {
+        chrome.send('oauthEnrollCompleteLogin', [username, 'authcode']);
+        chrome.send('completeLogin', [gaia_id, username, password, false]);
+      });
+    }
   };
 
   /**
    * Guest login for telemetry.
    */
   Oobe.guestLoginForTesting = function() {
     Oobe.skipToLoginForTesting();
     chrome.send('launchIncognito');
   };
 
@@ skipping 90 matching lines @@
 
 
 (function() {
   'use strict';
 
   document.addEventListener('DOMContentLoaded', function() {
     Oobe.initialize();
   });
 })();