MD Settings: show when the default search engine is controlled by policy

chrome/test/data/webui/settings/search_page_test.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_search_page', function() {
   function generateSearchEngineInfo() {
     var searchEngines0 = settings_search.createSampleSearchEngine(
         true, false, false);
     searchEngines0.default = true;
     var searchEngines1 = settings_search.createSampleSearchEngine(
@@ skipping 16 matching lines @@
       var page = null;
 
       var browserProxy = null;
 
       setup(function() {
         browserProxy = new settings_search.TestSearchEnginesBrowserProxy();
         browserProxy.setSearchEnginesInfo(generateSearchEngineInfo());
         settings.SearchEnginesBrowserProxyImpl.instance_ = browserProxy;
         PolymerTest.clearBody();
         page = document.createElement('settings-search-page');
+        page.prefs = {
+          default_search_provider_data: {
+            template_url_data: {},
+          },
+        };
         document.body.appendChild(page);
       });
 
       teardown(function() { page.remove(); });
 
       // Tests that the page is querying and displaying search engine info on
       // startup.
       test('Initialization', function() {
         var selectElement = page.$$('select');
 
@@ skipping 26 matching lines @@
           });
         });
       });
 
       test('ControlledByExtension', function() {
         return browserProxy.whenCalled('getSearchEnginesList').then(function() {
           var selectElement = page.$$('select');
           assertFalse(selectElement.disabled);
           assertFalse(!!page.$$('extension-controlled-indicator'));
 
-          page.prefs = {
-            default_search_provider: {
-              enabled: {
-                controlledBy: chrome.settingsPrivate.ControlledBy.EXTENSION,
-                controlledByName: 'fake extension name',
-                enforcement: chrome.settingsPrivate.Enforcement.ENFORCED,
-                extensionId: 'fake extension id',
-                extensionCanBeDisabled: true,
-                value: true,
-              },
-            },
-          };
+          page.set('prefs.default_search_provider_data.template_url_data', {
+            controlledBy: chrome.settingsPrivate.ControlledBy.EXTENSION,
+            controlledByName: 'fake extension name',
+            enforcement: chrome.settingsPrivate.Enforcement.ENFORCED,
+            extensionId: 'fake extension id',
+            extensionCanBeDisabled: true,
+            value: {},
+          });
           Polymer.dom.flush();
 
           assertTrue(selectElement.disabled);
           assertTrue(!!page.$$('extension-controlled-indicator'));
+          assertFalse(!!page.$$('cr-policy-pref-indicator'));
         });
       });
 
+      test('ControlledByPolicy', function() {
+        return browserProxy.whenCalled('getSearchEnginesList').then(function() {
+          var selectElement = page.$$('select');
+          assertFalse(selectElement.disabled);
+          assertFalse(!!page.$$('extension-controlled-indicator'));
+
+          page.set('prefs.default_search_provider_data.template_url_data', {
+            controlledBy: chrome.settingsPrivate.ControlledBy.USER_POLICY,
+            enforcement: chrome.settingsPrivate.Enforcement.ENFORCED,
+            value: {},
+          });
+          Polymer.dom.flush();
+
+          assertTrue(selectElement.disabled);
+          assertFalse(!!page.$$('extension-controlled-indicator'));
+          assertTrue(!!page.$$('cr-policy-pref-indicator'));
+        });
+      });
+
       // Tests the UI when Hotword 'alwaysOn' is true.
       test('HotwordAlwaysOn', function() {
         return browserProxy.whenCalled('getSearchEnginesList').then(function() {
           return browserProxy.whenCalled('getHotwordInfo');
         }).then(function() {
           Polymer.dom.flush();
           assertTrue(page.hotwordInfo_.allowed);
           assertTrue(page.hotwordInfo_.alwaysOn);
           assertFalse(page.hotwordInfo_.enabled);
           assertFalse(browserProxy.hotwordSearchEnabled);
@@ skipping 45 matching lines @@
         return browserProxy.whenCalled('getGoogleNowAvailability').then(
             function() {
           Polymer.dom.flush();
           assertTrue(page.googleNowAvailable_);
 
           var control = page.$$('#googleNowEnable');
           assertTrue(!!control);
           assertFalse(control.disabled);
           assertFalse(control.checked);
 
-          page.prefs = {
-            google_now_launcher: {
-              enabled: {
-                type: chrome.settingsPrivate.PrefType.BOOLEAN,
-                value: true,
-              }
-            }
-          };
+          page.set('prefs.google_now_launcher', {
+            enabled: {
+              type: chrome.settingsPrivate.PrefType.BOOLEAN,
+              value: true,
+            },
+          });
           Polymer.dom.flush();
           assertFalse(control.disabled);
           assertTrue(control.checked);
         });
       });
     });
   }
 
   return {
     registerTests: registerTests,
   };
 });