Enable/disable extensions upon changes in blacklist.

chrome/browser/extensions/extension_prefs_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 "chrome/browser/extensions/./extension_prefs_unittest.h"
 
 #include "base/basictypes.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/path_service.h"
 #include "base/prefs/mock_pref_change_callback.h"
@@ skipping 834 matching lines @@
   }
 
  private:
   scoped_refptr<const Extension> extension_a_;
   scoped_refptr<const Extension> extension_b_;
   scoped_refptr<const Extension> extension_c_;
 };
 TEST_F(ExtensionPrefsBlacklistedExtensions,
        ExtensionPrefsBlacklistedExtensions) {}
 
+// Tests the blacklist state. Old "blacklist" preference should take precedence
+// over new "blacklist_state".
+class ExtensionPrefsBlacklistState : public ExtensionPrefsTest {
+ public:
+  virtual ~ExtensionPrefsBlacklistState() {}
+
+  virtual void Initialize() OVERRIDE {
+    extension_a_ = prefs_.AddExtension("a");
+  }
+
+  virtual void Verify() OVERRIDE {
+    ExtensionIdSet empty_ids;
+    EXPECT_EQ(empty_ids, prefs()->GetBlacklistedExtensions());
+
+    prefs()->SetExtensionBlacklisted(extension_a_->id(), true);
+    EXPECT_EQ(BLACKLISTED_MALWARE,
+              prefs()->GetExtensionBlacklistState(extension_a_->id()));
+
+    prefs()->SetExtensionBlacklistState(extension_a_->id(),
+                                        BLACKLISTED_POTENTIALLY_UNWANTED);
+    EXPECT_EQ(BLACKLISTED_POTENTIALLY_UNWANTED,
+              prefs()->GetExtensionBlacklistState(extension_a_->id()));
+    EXPECT_FALSE(prefs()->IsExtensionBlacklisted(extension_a_->id()));
+    EXPECT_EQ(empty_ids, prefs()->GetBlacklistedExtensions());
+
+    prefs()->SetExtensionBlacklisted(extension_a_->id(), true);
+    EXPECT_TRUE(prefs()->IsExtensionBlacklisted(extension_a_->id()));
+    EXPECT_EQ(BLACKLISTED_MALWARE,
+              prefs()->GetExtensionBlacklistState(extension_a_->id()));
+    EXPECT_EQ(1u, prefs()->GetBlacklistedExtensions().size());
+
+    prefs()->SetExtensionBlacklistState(extension_a_->id(),
+                                        NOT_BLACKLISTED);
+    EXPECT_EQ(NOT_BLACKLISTED,
+              prefs()->GetExtensionBlacklistState(extension_a_->id()));
+    EXPECT_FALSE(prefs()->IsExtensionBlacklisted(extension_a_->id()));
+    EXPECT_EQ(empty_ids, prefs()->GetBlacklistedExtensions());
+  }
+
+ private:
+  scoped_refptr<const Extension> extension_a_;
+};
+TEST_F(ExtensionPrefsBlacklistState, ExtensionPrefsBlacklistState) {}
+
 }  // namespace extensions