Define EnabledComponentUpdates group policy for the component updater.

chrome/browser/component_updater/chrome_component_updater_configurator_unittest.cc

 // Copyright 2015 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 <memory>
 #include <string>
 #include <vector>
 
 #include "base/command_line.h"
+#include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "chrome/browser/component_updater/chrome_component_updater_configurator.h"
 #include "components/component_updater/component_updater_switches.h"
 #include "components/component_updater/component_updater_url_constants.h"
 #include "components/component_updater/configurator_impl.h"
 #include "components/prefs/testing_pref_service.h"
 #include "components/update_client/configurator.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
+std::unique_ptr<TestingPrefServiceSimple> pref(new TestingPrefServiceSimple());
+
 namespace component_updater {
 
 class ChromeComponentUpdaterConfiguratorTest : public testing::Test {
  public:
-  ChromeComponentUpdaterConfiguratorTest();
+  ChromeComponentUpdaterConfiguratorTest() {}
   ~ChromeComponentUpdaterConfiguratorTest() override{};
 
+  // Overrides from testing::Test.
+  void SetUp() override;
+
  protected:
-  PrefService* pref_service() { return pref_service_.get(); }
+  TestingPrefServiceSimple* pref_service() { return pref_service_.get(); }
 
  private:
   std::unique_ptr<TestingPrefServiceSimple> pref_service_;
 
   DISALLOW_COPY_AND_ASSIGN(ChromeComponentUpdaterConfiguratorTest);
 };
 
-ChromeComponentUpdaterConfiguratorTest::ChromeComponentUpdaterConfiguratorTest()
-    : pref_service_(new TestingPrefServiceSimple()) {}
+void ChromeComponentUpdaterConfiguratorTest::SetUp() {
+  pref_service_ = base::MakeUnique<TestingPrefServiceSimple>();
+  RegisterPrefsForChromeComponentUpdaterConfigurator(pref_service_->registry());
+}
 
 TEST_F(ChromeComponentUpdaterConfiguratorTest, TestDisablePings) {
   base::CommandLine cmdline(*base::CommandLine::ForCurrentProcess());
   cmdline.AppendSwitchASCII(switches::kComponentUpdater, "disable-pings");
 
   const auto config(MakeChromeComponentUpdaterConfigurator(&cmdline, nullptr,
                                                            pref_service()));
 
   const std::vector<GURL> pingUrls = config->PingUrl();
   EXPECT_TRUE(pingUrls.empty());
@@ skipping 90 matching lines @@
     ASSERT_STREQ(kUpdaterDefaultUrl, urls[0].spec().c_str());
     ASSERT_STREQ(kUpdaterFallbackUrl, urls[1].spec().c_str());
     ASSERT_EQ(config.UpdateUrl(), config.PingUrl());
   }
 }
 
 TEST_F(ChromeComponentUpdaterConfiguratorTest, TestEnabledComponentUpdates) {
   base::CommandLine cmdline(*base::CommandLine::ForCurrentProcess());
   const auto config(MakeChromeComponentUpdaterConfigurator(&cmdline, nullptr,
                                                            pref_service()));
+  // Tests the default is set to |true| and the component updates are enabled.
+  EXPECT_TRUE(config->EnabledComponentUpdates());
+
+  // Tests the component updates are disabled.
+  pref_service()->SetManagedPref("component_updates.enabled_component_updates",
+                                 new base::FundamentalValue(false));
+  EXPECT_FALSE(config->EnabledComponentUpdates());
+
+  // Tests the component updates are enabled.
+  pref_service()->SetManagedPref("component_updates.enabled_component_updates",
+                                 new base::FundamentalValue(true));
+  EXPECT_TRUE(config->EnabledComponentUpdates());
+
+  // Sanity check setting the preference back to |false| and then removing it.
+  pref_service()->SetManagedPref("component_updates.enabled_component_updates",
+                                 new base::FundamentalValue(false));
+  EXPECT_FALSE(config->EnabledComponentUpdates());
+  pref_service()->RemoveManagedPref(
+      "component_updates.enabled_component_updates");
   EXPECT_TRUE(config->EnabledComponentUpdates());
 }
 
 }  // namespace component_updater