Extension white and force lists (set by policy) should have priority over auto-updated Google black…

chrome/browser/extensions/extension_service_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_service_unittest.h"
 
 #include <algorithm>
 #include <set>
 #include <vector>
 
@@ skipping 2768 matching lines @@
   // Now, the good_crx is blacklisted.
   ValidateBooleanPref(good_crx, "blacklist", true);
 
   // We can not install good_crx.
   FilePath path = data_dir_.AppendASCII("good.crx");
   InstallCRX(path, INSTALL_FAILED);
   EXPECT_EQ(0u, service_->extensions()->size());
   ValidateBooleanPref(good_crx, "blacklist", true);
 }
 
+// Unload blacklisted extension on policy change.
+TEST_F(ExtensionServiceTest, UnloadBlacklistedExtensionPolicy) {
+  InitializeEmptyExtensionService();
+
+  FilePath path = data_dir_.AppendASCII("good.crx");
+
+  const Extension* good = InstallCRX(path, INSTALL_NEW);
+  EXPECT_EQ(good_crx, good->id());
+  UpdateExtension(good_crx, path, FAILED_SILENTLY);
+  EXPECT_EQ(1u, service_->extensions()->size());
+
+  base::ListValue whitelist;
+  PrefService* prefs = service_->extension_prefs()->pref_service();
+  whitelist.Append(base::Value::CreateStringValue(good_crx));
+  prefs->Set(prefs::kExtensionInstallAllowList, whitelist);
+
+  std::vector<std::string> blacklist;
+  blacklist.push_back(good_crx);
+  service_->UpdateExtensionBlacklist(blacklist);
+  // Make sure pref is updated

[Mattias Nissler (ping if slow), 2012/08/08 08:08:37]

Missing period.
Also, pref read writes don't require message loop iterations. Can you clarify
what actually does require it?

[qfel, 2012/08/13 16:03:02]

Uh, I was going to ask for the same, and forgot to put it in the message. 

+  loop_.RunAllPending();
+
+  // Now, the good_crx is blacklisted but whitelist negates it.
+  ValidateBooleanPref(good_crx, "blacklist", true);
+  EXPECT_EQ(1u, service_->extensions()->size());
+
+  whitelist.Clear();
+  prefs->Set(prefs::kExtensionInstallAllowList, whitelist);
+  loop_.RunAllPending();
+
+  // Now, the good_crx is blacklisted for good.
+  ValidateBooleanPref(good_crx, "blacklist", true);
+  EXPECT_EQ(0u, service_->extensions()->size());
+}
+
+// Allow Google-blacklisted extension if policy explicitly allows it (blacklist
+// then set policy).
+TEST_F(ExtensionServiceTest, WhitelistGoogleBlacklistedExtension) {
+  InitializeEmptyExtensionService();
+
+  std::vector<std::string> blacklist;
+  blacklist.push_back(good_crx);
+  EXPECT_FALSE(IsPrefExist(good_crx, "blacklist"));
+  service_->UpdateExtensionBlacklist(blacklist);
+  loop_.RunAllPending();
+  EXPECT_TRUE(IsPrefExist(good_crx, "blacklist"));
+
+  PrefService* prefs = service_->extension_prefs()->pref_service();
+  base::ListValue whitelist;
+  whitelist.Append(base::Value::CreateStringValue(good_crx));
+  prefs->Set(prefs::kExtensionInstallAllowList, whitelist);
+  loop_.RunAllPending();
+  EXPECT_FALSE(service_->extension_prefs()->IsExtensionBlacklisted(good_crx));
+}
+
+// Allow Google-blacklisted extension if policy requires it (blacklist then set
+// policy).
+TEST_F(ExtensionServiceTest, ForcelistGoogleBlacklistedExtension) {
+  InitializeEmptyExtensionService();
+
+  std::vector<std::string> blacklist;
+  blacklist.push_back(good_crx);
+  EXPECT_FALSE(IsPrefExist(good_crx, "blacklist"));
+  service_->UpdateExtensionBlacklist(blacklist);
+  loop_.RunAllPending();
+  EXPECT_TRUE(IsPrefExist(good_crx, "blacklist"));
+
+  PrefService* prefs = service_->extension_prefs()->pref_service();
+  base::ListValue forcelist;
+  forcelist.Append(base::Value::CreateStringValue(good_crx));
+  prefs->Set(prefs::kExtensionInstallForceList, forcelist);
+  loop_.RunAllPending();
+  EXPECT_FALSE(service_->extension_prefs()->IsExtensionBlacklisted(good_crx));
+}
+
+// Allow Google-blacklisted extension if policy explicitly allows it (set policy
+// then blacklist).

[Mattias Nissler (ping if slow), 2012/08/08 08:08:37]

I'm not sure we really need to test different order here.

[qfel, 2012/08/13 16:03:02]

Considering you knowing the implementation, it's a bit biased statement:)
If it was more callback-based (as opposed to current read on demand) it would be
quite ease to get one ordering right and don't refresh on the other.

+TEST_F(ExtensionServiceTest, GoogleBlacklistWhitelistedExtension) {
+  InitializeEmptyExtensionService();
+
+  PrefService* prefs = service_->extension_prefs()->pref_service();
+  base::ListValue whitelist;
+  whitelist.Append(base::Value::CreateStringValue(good_crx));
+  prefs->Set(prefs::kExtensionInstallAllowList, whitelist);
+  loop_.RunAllPending();
+
+  std::vector<std::string> blacklist;
+  blacklist.push_back(good_crx);
+  EXPECT_FALSE(IsPrefExist(good_crx, "blacklist"));
+  service_->UpdateExtensionBlacklist(blacklist);
+  loop_.RunAllPending();
+  EXPECT_FALSE(service_->extension_prefs()->IsExtensionBlacklisted(good_crx));
+}
+
+// Allow Google-blacklisted extension if policy requires it (set policy then
+// blacklist).
+TEST_F(ExtensionServiceTest, GoogleBlacklistForcelistedExtension) {
+  InitializeEmptyExtensionService();
+
+  PrefService* prefs = service_->extension_prefs()->pref_service();
+  base::ListValue forcelist;
+  forcelist.Append(base::Value::CreateStringValue(good_crx));
+  prefs->Set(prefs::kExtensionInstallForceList, forcelist);
+  loop_.RunAllPending();
+
+  std::vector<std::string> blacklist;
+  blacklist.push_back(good_crx);
+  EXPECT_FALSE(IsPrefExist(good_crx, "blacklist"));
+  service_->UpdateExtensionBlacklist(blacklist);
+  loop_.RunAllPending();
+  EXPECT_FALSE(service_->extension_prefs()->IsExtensionBlacklisted(good_crx));
+}
+
 // Test loading extensions from the profile directory, except
 // blacklisted ones.
 TEST_F(ExtensionServiceTest, WillNotLoadBlacklistedExtensionsFromDirectory) {
   // Initialize the test dir with a good Preferences/extensions.
   FilePath source_install_dir = data_dir_
       .AppendASCII("good")
       .AppendASCII("Extensions");
   FilePath pref_path = source_install_dir
       .DirName()
       .AppendASCII("Preferences");
@@ skipping 2447 matching lines @@
   // This should NOT trigger an alert.
   provider->UpdateOrAddExtension(hosted_app, "1.0.0.0",
                                  data_dir_.AppendASCII("hosted_app.crx"));
 
   service_->CheckForExternalUpdates();
   loop_.RunAllPending();
 
   ASSERT_TRUE(service_->PopulateExtensionErrorUI(extension_error_ui.get()));
   ASSERT_EQ(1u, extension_error_ui->get_external_extension_ids()->size());
 }