Add management policy function allowing extensions to be disabled.

chrome/browser/extensions/management_policy_unittest.cc

 // Copyright (c) 2012 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 "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/management_policy.h"
 #include "chrome/browser/extensions/test_management_policy.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 typedef extensions::TestManagementPolicyProvider TestProvider;
 using extensions::Extension;
 
 class ManagementPolicyTest : public testing::Test {
  public:
   virtual void SetUp() {
     allow_all_.SetProhibitedActions(TestProvider::ALLOW_ALL);
     no_modify_status_.SetProhibitedActions(
         TestProvider::PROHIBIT_MODIFY_STATUS);
     no_load_.SetProhibitedActions(TestProvider::PROHIBIT_LOAD);
     must_remain_enabled_.SetProhibitedActions(
         TestProvider::MUST_REMAIN_ENABLED);
+    must_remain_disabled_.SetProhibitedActions(
+        TestProvider::MUST_REMAIN_DISABLED);
+    must_remain_disabled_.SetDisableReason(Extension::DISABLE_SIDELOAD_WIPEOUT);
     restrict_all_.SetProhibitedActions(TestProvider::PROHIBIT_MODIFY_STATUS |
                                        TestProvider::PROHIBIT_LOAD |
                                        TestProvider::MUST_REMAIN_ENABLED);
   }
 
  protected:
   extensions::ManagementPolicy policy_;
 
   TestProvider allow_all_;
   TestProvider no_modify_status_;
   TestProvider no_load_;
   TestProvider must_remain_enabled_;
+  TestProvider must_remain_disabled_;
   TestProvider restrict_all_;
 };
 
 TEST_F(ManagementPolicyTest, RegisterAndUnregister) {
   EXPECT_EQ(0, policy_.GetNumProviders());
   policy_.RegisterProvider(&allow_all_);
   EXPECT_EQ(1, policy_.GetNumProviders());
   policy_.RegisterProvider(&allow_all_);
   EXPECT_EQ(1, policy_.GetNumProviders());
 
@@ skipping 91 matching lines @@
   EXPECT_TRUE(policy_.MustRemainEnabled(NULL, &error));
   EXPECT_FALSE(error.empty());
 
   // Remove the restriction.
   policy_.UnregisterProvider(&must_remain_enabled_);
   error.clear();
   EXPECT_FALSE(policy_.MustRemainEnabled(NULL, &error));
   EXPECT_TRUE(error.empty());
 }
 
+TEST_F(ManagementPolicyTest, MustRemainDisabled) {
+  // No providers registered.
+  string16 error;
+  EXPECT_FALSE(policy_.MustRemainDisabled(NULL, NULL, &error));
+  EXPECT_TRUE(error.empty());
+
+  // One provider, no relevant restriction.
+  policy_.RegisterProvider(&allow_all_);
+  EXPECT_FALSE(policy_.MustRemainDisabled(NULL, NULL, &error));
+  EXPECT_TRUE(error.empty());
+
+  // Two providers, no relevant restrictions.
+  policy_.RegisterProvider(&no_modify_status_);
+  EXPECT_FALSE(policy_.MustRemainDisabled(NULL, NULL, &error));
+  EXPECT_TRUE(error.empty());
+
+  // Three providers, one with a relevant restriction.
+  Extension::DisableReason reason = Extension::DISABLE_NONE;
+  policy_.RegisterProvider(&must_remain_disabled_);
+  EXPECT_TRUE(policy_.MustRemainDisabled(NULL, &reason, &error));
+  EXPECT_FALSE(error.empty());
+  EXPECT_EQ(Extension::DISABLE_SIDELOAD_WIPEOUT, reason);
+
+  // Remove the restriction.
+  policy_.UnregisterProvider(&must_remain_disabled_);
+  error.clear();
+  EXPECT_FALSE(policy_.MustRemainDisabled(NULL, NULL, &error));
+  EXPECT_TRUE(error.empty());
+}
+
 // Tests error handling in the ManagementPolicy.
 TEST_F(ManagementPolicyTest, ErrorHandling) {
   // The error parameter should be unchanged if no restriction was found.
   std::string original_error = "Ceci est en effet une erreur.";
   string16 original_error16 = UTF8ToUTF16(original_error);
   string16 error = original_error16;
   EXPECT_TRUE(policy_.UserMayLoad(NULL, &error));
   EXPECT_EQ(original_error, UTF16ToUTF8(error));
   EXPECT_TRUE(policy_.UserMayModifySettings(NULL, &error));
   EXPECT_EQ(original_error, UTF16ToUTF8(error));
@@ skipping 13 matching lines @@
   error = original_error16;
   EXPECT_FALSE(policy_.UserMayLoad(NULL, &error));
   EXPECT_EQ(UTF8ToUTF16(TestProvider::expected_error()), error);
   error = original_error16;
   EXPECT_FALSE(policy_.UserMayModifySettings(NULL, &error));
   EXPECT_EQ(UTF8ToUTF16(TestProvider::expected_error()), error);
   error = original_error16;
   EXPECT_TRUE(policy_.MustRemainEnabled(NULL, &error));
   EXPECT_EQ(UTF8ToUTF16(TestProvider::expected_error()), error);
 }