MD Settings: slight tweaks to when the metrics reporting restart button shows up

chrome/test/data/webui/settings/metrics_reporting_tests.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.
 
 suite('metrics reporting', function() {
   /** @type {settings.TestPrivacyPageBrowserProxy} */
   var testBrowserProxy;
 
   /** @type {SettingsPrivacyPageElement} */
   var page;
 
   setup(function() {
     testBrowserProxy = new TestPrivacyPageBrowserProxy();
     settings.PrivacyPageBrowserProxyImpl.instance_ = testBrowserProxy;
     PolymerTest.clearBody();
     page = document.createElement('settings-privacy-page');
   });
 
   teardown(function() { page.remove(); });
 
   test('changes to whether metrics reporting is enabled/managed', function() {
     return testBrowserProxy.whenCalled('getMetricsReporting').then(function() {
       Polymer.dom.flush();
-      assertFalse(!!page.$$('#metricsReporting paper-button'));
 
       var checkbox = page.$.metricsReportingCheckbox;
       assertEquals(testBrowserProxy.metricsReporting.enabled, checkbox.checked);
       var indicatorVisible = !!page.$$('#indicator');
       assertEquals(testBrowserProxy.metricsReporting.managed, indicatorVisible);
 
       var changedMetrics = {
         enabled: !testBrowserProxy.metricsReporting.enabled,
         managed: !testBrowserProxy.metricsReporting.managed,
       };
       cr.webUIListenerCallback('metrics-reporting-change', changedMetrics);
       Polymer.dom.flush();
 
-      assertTrue(!!page.$$('#metricsReporting paper-button'));
-
       assertEquals(changedMetrics.enabled, checkbox.checked);
       indicatorVisible = !!page.$$('#indicator');
       assertEquals(changedMetrics.managed, indicatorVisible);
 
       var toggled = !changedMetrics.enabled;
 
       MockInteractions.tap(checkbox);
       return testBrowserProxy.whenCalled('setMetricsReportingEnabled', toggled);
-    }).then(function() {
+    });
+  });
+
+  test('metrics reporting restart button', function() {
+    return testBrowserProxy.whenCalled('getMetricsReporting').then(function() {
+      Polymer.dom.flush();
+
+      // Restart button should be hidden by default (in any state).
+      assertFalse(!!page.$$('#metricsReporting paper-button'));
+
+      // Simulate toggling via policy.
+      cr.webUIListenerCallback('metrics-reporting-change', {
+        enabled: false,
+        managed: true,
+      });
+      Polymer.dom.flush();
+
+      // No restart button should show because the value is managed.
+      assertFalse(!!page.$$('#metricsReporting paper-button'));
+
+      cr.webUIListenerCallback('metrics-reporting-change', {
+        enabled: true,
+        managed: true,
+      });
+      Polymer.dom.flush();
+
+      // Changes in policy should not show the restart button because the value
+      // is still managed.
+      assertFalse(!!page.$$('#metricsReporting paper-button'));
+
+      // Remove the policy and toggle the value.
+      cr.webUIListenerCallback('metrics-reporting-change', {
+        enabled: false,
+        managed: false,
+      });
+      Polymer.dom.flush();
+
+      // Now the restart button should be showing.
+      assertTrue(!!page.$$('#metricsReporting paper-button'));
+
+      // Receiving the same values should have no effect.
+       cr.webUIListenerCallback('metrics-reporting-change', {
+        enabled: false,
+        managed: false,
+      });
+      Polymer.dom.flush();
       assertTrue(!!page.$$('#metricsReporting paper-button'));
     });
   });
 });