Added uma for pin unlock set up.

chrome/test/data/webui/settings/quick_unlock_authenticate_browsertest_chromeos.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.
 
 cr.define('settings_people_page_quick_unlock', function() {
   var element = null;
   var quickUnlockPrivateApi = null;
   var QuickUnlockMode = chrome.quickUnlockPrivate.QuickUnlockMode;
+  var fakeUma = null;
 
   /**
    * Returns if the element is visible.
    * @param {!Element} element
    */
   function isVisible(element) {
     while (element) {
       if (element.offsetWidth <= 0 || element.offsetHeight <= 0 ||
           element.hidden) {
         return false;
@@ skipping 35 matching lines @@
   }
 
   function registerAuthenticateTests() {
     suite('authenticate', function() {
       var passwordElement = null;
 
       setup(function() {
         PolymerTest.clearBody();
 
         quickUnlockPrivateApi = new settings.FakeQuickUnlockPrivate();
+        fakeUma = new settings.FakeQuickUnlockUma();
 
         element = document.createElement('settings-password-prompt-dialog');
         element.quickUnlockPrivate_ = quickUnlockPrivateApi;
+        element.writeUma_ = fakeUma.recordProgress.bind(fakeUma);
         document.body.appendChild(element);
 
         passwordElement = getFromElement('#passwordInput');
       });
 
       test('PasswordCheckDoesNotChangeActiveMode', function() {
         // No active modes.
         quickUnlockPrivateApi.activeModes = [];
         passwordElement.value = 'foo';
         element.submitPassword_();
         assertDeepEquals([], quickUnlockPrivateApi.activeModes);
         assertDeepEquals([], quickUnlockPrivateApi.credentials);
 
         // PIN is active.
         quickUnlockPrivateApi.activeModes = [QuickUnlockMode.PIN];
         passwordElement.value = 'foo';
         element.submitPassword_();
         assertDeepEquals([QuickUnlockMode.PIN],
                          quickUnlockPrivateApi.activeModes);
         assertDeepEquals([''], quickUnlockPrivateApi.credentials);
       });
 
-      // A bad password does not provide an authenticated setModes object.
+      // A bad password does not provide an authenticated setModes object, and a
+      // entered password correctly uma should not be recorded.
       test('InvalidPasswordDoesNotProvideAuthentication', function() {
         quickUnlockPrivateApi.accountPassword = 'bar';
 
         passwordElement.value = 'foo';
         element.submitPassword_();
 
+        assertEquals(0, fakeUma.getHistogramValue(
+            LockScreenProgress.ENTER_PASSWORD_CORRECTLY));
         assertFalse(!!element.setModes);
       });
 
-      // A valid password provides an authenticated setModes object.
+      // A valid password provides an authenticated setModes object, and a
+      // entered password correctly uma should be recorded.
       test('ValidPasswordProvidesAuthentication', function() {
         quickUnlockPrivateApi.accountPassword = 'foo';
 
         passwordElement.value = 'foo';
         element.submitPassword_();
 
+        assertEquals(1, fakeUma.getHistogramValue(
+            LockScreenProgress.ENTER_PASSWORD_CORRECTLY));
         assertTrue(!!element.setModes);
       });
 
       // The setModes objects times out after a delay.
       test('AuthenticationTimesOut', function(done) {
         quickUnlockPrivateApi.accountPassword = 'foo';
 
         element.passwordActiveDurationMs_ = 0;
         passwordElement.value = 'foo';
         element.submitPassword_();
@@ skipping 75 matching lines @@
 
         CrSettingsPrefs.deferInitialization = true;
 
         // Build pref fakes.
         var fakePrefs = [{
           key: ENABLE_LOCK_SCREEN_PREF,
           type: chrome.settingsPrivate.PrefType.BOOLEAN,
           value: true
         }];
         fakeSettings = new settings.FakeSettingsPrivate(fakePrefs);
+        fakeUma = new settings.FakeQuickUnlockUma();
         setLockScreenPref(true);
         var prefElement = document.createElement('settings-prefs');
         prefElement.initialize(fakeSettings);
         document.body.appendChild(prefElement);
 
         // Wait for prefElement to finish initializing; it takes some time for
         // the prefs element to get allocated.
         prefElement.addEventListener('prefs-changed', function prefsReady() {
           prefElement.removeEventListener('prefs-changed', prefsReady);
 
           quickUnlockPrivateApi = new settings.FakeQuickUnlockPrivate();
 
           // Create choose-method element.
           element = document.createElement('settings-lock-screen');
           element.settingsPrivate_ = fakeSettings;
           element.quickUnlockPrivate_ = quickUnlockPrivateApi;
           element.prefs = prefElement.prefs;
+          element.writeUma_ = fakeUma.recordProgress.bind(fakeUma);
 
           document.body.appendChild(element);
           Polymer.dom.flush();
 
           element.setModes_ =
               quickUnlockPrivateApi.setModes.bind(quickUnlockPrivateApi, '');
 
           passwordRadioButton =
               getFromElement('paper-radio-button[name="password"]');
           pinPasswordRadioButton =
@@ skipping 50 matching lines @@
       test('EnablingQuickUnlockChangesButtonState', function() {
         setActiveModes([QuickUnlockMode.PIN]);
         assertRadioButtonActive(pinPasswordRadioButton);
         assertTrue(isVisible(configureButton));
 
         setActiveModes([]);
         assertRadioButtonActive(passwordRadioButton);
         assertDeepEquals([], quickUnlockPrivateApi.activeModes);
       });
 
-      // Tapping the PIN configure button opens up the setup PIN dialog.
+      // Tapping the PIN configure button opens up the setup PIN dialog, and
+      // records a chose pin or password uma.
       test('TappingConfigureOpensSetupPin', function() {
+        assertEquals(0, fakeUma.getHistogramValue(
+            LockScreenProgress.CHOOSE_PIN_OR_PASSWORD));
         assertRadioButtonActive(passwordRadioButton);
 
         MockInteractions.tap(pinPasswordRadioButton);
         assertTrue(isVisible(configureButton));
         assertRadioButtonActive(pinPasswordRadioButton)
 
         MockInteractions.tap(configureButton);
         var setupPinDialog = getFromElement('#setupPin');
         assertTrue(setupPinDialog.$.dialog.open);
+        assertEquals(1, fakeUma.getHistogramValue(
+            LockScreenProgress.CHOOSE_PIN_OR_PASSWORD));
       });
     });
   }
 
   function registerSetupPinDialogTests() {
     suite('setup-pin-dialog', function() {
       var titleDiv = null;
       var problemDiv = null;
       var pinKeyboard = null;
       var backButton = null;
       var continueButton = null;
 
       suiteSetup(function() {
         var urls = ['chrome://md-settings/i18n_setup.html'];
         return Promise.all(urls.map(PolymerTest.importHtml));
       });
 
       setup(function() {
         PolymerTest.clearBody();
 
         quickUnlockPrivateApi = new settings.FakeQuickUnlockPrivate();
+        fakeUma = new settings.FakeQuickUnlockUma();
 
         // Create setup-pin element.
         element = document.createElement('settings-setup-pin-dialog');
         element.setModes =
             quickUnlockPrivateApi.setModes.bind(quickUnlockPrivateApi, '');
+        element.writeUma_ = fakeUma.recordProgress.bind(fakeUma);
 
         document.body.appendChild(element);
         Polymer.dom.flush();
 
         element.open();
 
         titleDiv = getFromElement('div[class="title"]');
         problemDiv = getFromElement('#problemDiv');
         pinKeyboard = getFromElement('pin-keyboard');
         backButton = getFromElement('paper-button[class="cancel-button"]');
@@ skipping 96 matching lines @@
       });
 
       // User has to re-enter PIN for confirm step.
       test('PinKeyboardIsResetForConfirmStep', function() {
         pinKeyboard.value = '1111';
         MockInteractions.tap(continueButton);
         assertEquals('', pinKeyboard.value);
       });
 
       // Completing the flow results in a call to the quick unlock private API.
+      // Check that uma stats are called as expected.
       test('SubmittingPinCallsQuickUnlockApi', function() {
         // Entering the same (even weak) pin twice calls the quick unlock API
         // and sets up a PIN.
+        assertEquals(0, fakeUma.getHistogramValue(
+            LockScreenProgress.ENTER_PIN));
+        assertEquals(0, fakeUma.getHistogramValue(
+            LockScreenProgress.CONFIRM_PIN));
         pinKeyboard.value = '1111';
         MockInteractions.tap(continueButton);
+        assertEquals(1, fakeUma.getHistogramValue(
+            LockScreenProgress.ENTER_PIN));
+
         pinKeyboard.value = '1111';
         MockInteractions.tap(continueButton);
 
+        assertEquals(1, fakeUma.getHistogramValue(
+            LockScreenProgress.CONFIRM_PIN));
         assertDeepEquals(['PIN'], quickUnlockPrivateApi.activeModes);
         assertDeepEquals(['1111'], quickUnlockPrivateApi.credentials);
       });
     });
   }
 
   return {
     registerAuthenticateTests: registerAuthenticateTests,
     registerLockScreenTests: registerLockScreenTests,
     registerSetupPinDialogTests: registerSetupPinDialogTests
   };
 });