Add update url controlled extension management settings

chrome/browser/extensions/extension_management_unittest.cc

 // Copyright 2014 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 <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/json/json_parser.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 15 matching lines @@
 #include "url/gurl.h"
 
 namespace extensions {
 
 namespace {
 
 const char kTargetExtension[] = "abcdefghijklmnopabcdefghijklmnop";
 const char kTargetExtension2[] = "bcdefghijklmnopabcdefghijklmnopa";
 const char kTargetExtension3[] = "cdefghijklmnopabcdefghijklmnopab";
 const char kTargetExtension4[] = "defghijklmnopabcdefghijklmnopabc";
-const char kOtherExtension[] = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
 const char kExampleUpdateUrl[] = "http://example.com/update_url";
 
+const char kNonExistingExtension[] = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
+const char kNonExistingUpdateUrl[] = "http://example.net/update.xml";
+
 const char kExampleDictPreference[] =
     "{"
     "  \"abcdefghijklmnopabcdefghijklmnop\": {"  // kTargetExtension
     "    \"installation_mode\": \"allowed\","
     "    \"blocked_permissions\": [\"fileSystem\", \"bookmarks\"],"
     "  },"
     "  \"bcdefghijklmnopabcdefghijklmnopa\": {"  // kTargetExtension2
     "    \"installation_mode\": \"force_installed\","
     "    \"update_url\": \"http://example.com/update_url\","
     "    \"allowed_permissions\": [\"fileSystem\", \"bookmarks\"],"
     "  },"
     "  \"cdefghijklmnopabcdefghijklmnopab\": {"  // kTargetExtension3
     "    \"installation_mode\": \"normal_installed\","
     "    \"update_url\": \"http://example.com/update_url\","
     "    \"allowed_permissions\": [\"fileSystem\", \"downloads\"],"
     "    \"blocked_permissions\": [\"fileSystem\", \"history\"],"
     "  },"
     "  \"defghijklmnopabcdefghijklmnopabc\": {"  // kTargetExtension4
     "    \"installation_mode\": \"blocked\","
     "  },"
+    "  \"update_url:http://example.com/update_url\": {"  // kExampleUpdateUrl
+    "    \"installation_mode\": \"allowed\","
+    "    \"allowed_permissions\": [\"downloads\"],"
+    "    \"blocked_permissions\": [\"bookmarks\"],"
+    "  },"
     "  \"*\": {"
     "    \"installation_mode\": \"blocked\","
     "    \"install_sources\": [\"*://foo.com/*\"],"
     "    \"allowed_types\": [\"theme\", \"user_script\"],"
     "    \"blocked_permissions\": [\"fileSystem\", \"downloads\"],"
     "  },"
     "}";
 
 }  // namespace
 
@@ skipping 29 matching lines @@
       pref_service_->SetUserPref(path, value);
   }
 
   void RemovePref(bool managed, const char* path) {
     if (managed)
       pref_service_->RemoveManagedPref(path);
     else
       pref_service_->RemoveUserPref(path);
   }
 
-  const internal::IndividualSettings* ReadById(const ExtensionId& id) {
-    return extension_management_->ReadById(id);
+  const internal::GlobalSettings* ReadGlobalSettings() {
+    return extension_management_->global_settings_.get();
   }
 
-  const internal::GlobalSettings* ReadGlobalSettings() {
-    return extension_management_->ReadGlobalSettings();
+  ExtensionManagement::InstallationMode GetInstallationModeById(
+      const std::string& id) {
+    return GetInstallationMode(id, kNonExistingUpdateUrl);
+  }
+
+  ExtensionManagement::InstallationMode GetInstallationModeByUpdateUrl(
+      const std::string& update_url) {
+    return GetInstallationMode(kNonExistingExtension, update_url);
+  }
+
+  void CheckAutomaticallyInstalledUpdateUrl(const std::string& id,
+                                            const std::string& update_url) {
+    auto iter = extension_management_->settings_by_id_.find(id);
+    ASSERT_TRUE(iter != extension_management_->settings_by_id_.end());
+    ASSERT_TRUE((iter->second->installation_mode ==
+                 ExtensionManagement::INSTALLATION_FORCED) ||
+                (iter->second->installation_mode ==
+                 ExtensionManagement::INSTALLATION_RECOMMENDED));
+    EXPECT_EQ(iter->second->update_url, update_url);
+  }
+
+  APIPermissionSet GetBlockedAPIPermissionsById(const std::string& id) {
+    return GetBlockedAPIPermissions(id, kNonExistingUpdateUrl);
+  }
+
+  APIPermissionSet GetBlockedAPIPermissionsByUpdateUrl(
+      const std::string& update_url) {
+    return GetBlockedAPIPermissions(kNonExistingExtension, update_url);
   }
 
   void SetExampleDictPref() {
     std::string error_msg;
     scoped_ptr<base::Value> parsed(base::JSONReader::ReadAndReturnError(
         kExampleDictPreference,
         base::JSONParserOptions::JSON_ALLOW_TRAILING_COMMAS,
         NULL,
         &error_msg));
     ASSERT_TRUE(parsed && parsed->IsType(base::Value::TYPE_DICTIONARY))
         << error_msg;
     SetPref(true, pref_names::kExtensionManagement, parsed.release());
   }
 
  protected:

[Joao da Silva, 2014/11/04 14:06:09]

Why are some of the helper methods public and others protected?

[binjin, 2014/11/05 12:52:42]

Done. Moving helper functions to public and internal one to private instead.

+  ExtensionManagement::InstallationMode GetInstallationMode(
+      const std::string& id,
+      const std::string& update_url) {
+    scoped_refptr<const Extension> extension =
+        CreateExtensionWithIdAndUpdateUrl(Manifest::UNPACKED, id, update_url);
+    return extension_management_->GetInstallationMode(extension.get());
+  }
+
+  APIPermissionSet GetBlockedAPIPermissions(const std::string& id,
+                                            const std::string& update_url) {
+    scoped_refptr<const Extension> extension =
+        CreateExtensionWithIdAndUpdateUrl(Manifest::UNPACKED, id, update_url);
+    return extension_management_->GetBlockedAPIPermissions(extension.get());
+  }
+
+  // Create an extension with specified |location|, |id| and |update_url|.
+  scoped_refptr<const Extension> CreateExtensionWithIdAndUpdateUrl(
+      Manifest::Location location,
+      const std::string& id,
+      const std::string& update_url) {
+    base::DictionaryValue manifest_dict;
+    manifest_dict.SetString(manifest_keys::kName, "test");
+    manifest_dict.SetString(manifest_keys::kVersion, "0.1");
+    manifest_dict.SetString(manifest_keys::kUpdateURL, update_url);
+    std::string error;
+    scoped_refptr<const Extension> extension =
+        Extension::Create(base::FilePath(), location, manifest_dict,
+                          Extension::NO_FLAGS, id, &error);
+    CHECK(extension.get()) << error;
+    return extension;
+  }
+
   scoped_ptr<TestingPrefServiceSimple> pref_service_;
   scoped_ptr<ExtensionManagement> extension_management_;
 };
 
 class ExtensionAdminPolicyTest : public ExtensionManagementServiceTest {
  public:
   ExtensionAdminPolicyTest() {}
   ~ExtensionAdminPolicyTest() override {}
 
   void SetUpPolicyProvider() {
@@ skipping 121 matching lines @@
                    Manifest::TYPE_USER_SCRIPT) != allowed_types.end());
 }
 
 // Verify that preference controlled by legacy ExtensionInstallBlacklist policy
 // is handled well.
 TEST_F(ExtensionManagementServiceTest, LegacyInstallBlacklist) {
   base::ListValue denied_list_pref;
   denied_list_pref.AppendString(kTargetExtension);
 
   SetPref(true, pref_names::kInstallDenyList, denied_list_pref.DeepCopy());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 }
 
 // Verify that preference controlled by legacy ExtensionInstallWhitelist policy
 // is handled well.
 TEST_F(ExtensionManagementServiceTest, LegacyInstallWhitelist) {
   base::ListValue denied_list_pref;
   denied_list_pref.AppendString("*");
   base::ListValue allowed_list_pref;
   allowed_list_pref.AppendString(kTargetExtension);
 
   SetPref(true, pref_names::kInstallDenyList, denied_list_pref.DeepCopy());
   SetPref(true, pref_names::kInstallAllowList, allowed_list_pref.DeepCopy());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
 
   // Verify that install whitelist preference set by user is ignored.
   RemovePref(true, pref_names::kInstallAllowList);
   SetPref(false, pref_names::kInstallAllowList, allowed_list_pref.DeepCopy());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
 }
 
 // Verify that preference controlled by legacy ExtensionInstallForcelist policy
 // is handled well.
 TEST_F(ExtensionManagementServiceTest, LegacyInstallForcelist) {
   base::DictionaryValue forced_list_pref;
   ExternalPolicyLoader::AddExtension(
       &forced_list_pref, kTargetExtension, kExampleUpdateUrl);
 
   SetPref(true, pref_names::kInstallForceList, forced_list_pref.DeepCopy());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_FORCED);
-  EXPECT_EQ(ReadById(kTargetExtension)->update_url, kExampleUpdateUrl);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  CheckAutomaticallyInstalledUpdateUrl(kTargetExtension, kExampleUpdateUrl);
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 
   // Verify that install forcelist preference set by user is ignored.
   RemovePref(true, pref_names::kInstallForceList);
   SetPref(false, pref_names::kInstallForceList, forced_list_pref.DeepCopy());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 }
 
 // Tests parsing of new dictionary preference.
 TEST_F(ExtensionManagementServiceTest, PreferenceParsing) {
   SetExampleDictPref();
 
   // Verifies the installation mode settings.
   EXPECT_TRUE(extension_management_->BlacklistedByDefault());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
-  EXPECT_EQ(ReadById(kTargetExtension2)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension2),
             ExtensionManagement::INSTALLATION_FORCED);
-  EXPECT_EQ(ReadById(kTargetExtension2)->update_url, kExampleUpdateUrl);
-  EXPECT_EQ(ReadById(kTargetExtension3)->installation_mode,
+  CheckAutomaticallyInstalledUpdateUrl(kTargetExtension2, kExampleUpdateUrl);
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension3),
             ExtensionManagement::INSTALLATION_RECOMMENDED);
-  EXPECT_EQ(ReadById(kTargetExtension3)->update_url, kExampleUpdateUrl);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  CheckAutomaticallyInstalledUpdateUrl(kTargetExtension3, kExampleUpdateUrl);
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
+  EXPECT_EQ(GetInstallationModeByUpdateUrl(kExampleUpdateUrl),
+            ExtensionManagement::INSTALLATION_ALLOWED);
 
   // Verifies global settings.
   EXPECT_TRUE(ReadGlobalSettings()->has_restricted_install_sources);
   const URLPatternSet& allowed_sites = ReadGlobalSettings()->install_sources;
   EXPECT_EQ(allowed_sites.size(), 1u);
   EXPECT_TRUE(allowed_sites.MatchesURL(GURL("http://foo.com/entry")));
   EXPECT_FALSE(allowed_sites.MatchesURL(GURL("http://bar.com/entry")));
 
   EXPECT_TRUE(ReadGlobalSettings()->has_restricted_allowed_types);
   const std::vector<Manifest::Type>& allowed_types =
       ReadGlobalSettings()->allowed_types;
   EXPECT_EQ(allowed_types.size(), 2u);
   EXPECT_TRUE(std::find(allowed_types.begin(),
                         allowed_types.end(),
                         Manifest::TYPE_THEME) != allowed_types.end());
   EXPECT_TRUE(std::find(allowed_types.begin(),
                         allowed_types.end(),
                         Manifest::TYPE_USER_SCRIPT) != allowed_types.end());
 
   // Verifies blocked permission list settings.
   APIPermissionSet api_permission_set;
   api_permission_set.clear();
   api_permission_set.insert(APIPermission::kFileSystem);
   api_permission_set.insert(APIPermission::kDownloads);
   EXPECT_EQ(api_permission_set,
-            extension_management_->GetBlockedAPIPermissions(kOtherExtension));
+            GetBlockedAPIPermissionsById(kNonExistingExtension));
 
   api_permission_set.clear();
   api_permission_set.insert(APIPermission::kFileSystem);
   api_permission_set.insert(APIPermission::kDownloads);
   api_permission_set.insert(APIPermission::kBookmark);
-  EXPECT_EQ(api_permission_set,
-            extension_management_->GetBlockedAPIPermissions(kTargetExtension));
+  EXPECT_EQ(api_permission_set, GetBlockedAPIPermissionsById(kTargetExtension));
 
   api_permission_set.clear();
   api_permission_set.insert(APIPermission::kDownloads);
   EXPECT_EQ(api_permission_set,
-            extension_management_->GetBlockedAPIPermissions(kTargetExtension2));
+            GetBlockedAPIPermissionsById(kTargetExtension2));
 
   api_permission_set.clear();
   api_permission_set.insert(APIPermission::kFileSystem);
   api_permission_set.insert(APIPermission::kHistory);
   EXPECT_EQ(api_permission_set,
-            extension_management_->GetBlockedAPIPermissions(kTargetExtension3));
+            GetBlockedAPIPermissionsById(kTargetExtension3));
+
+  api_permission_set.clear();
+  api_permission_set.insert(APIPermission::kFileSystem);
+  api_permission_set.insert(APIPermission::kBookmark);
+  EXPECT_EQ(api_permission_set,
+            GetBlockedAPIPermissionsByUpdateUrl(kExampleUpdateUrl));
+}
+
+// Tests the handling of installation mode in case it's specified in both
+// per-extension and per-update-url settings.
+TEST_F(ExtensionManagementServiceTest, InstallationModeConflictionHandling) {

[Joao da Silva, 2014/11/04 14:06:09]

InstallationModeConflicHandling

[binjin, 2014/11/05 12:52:42]

Done.

+  SetExampleDictPref();
+
+  // Per-extension installation mode settings should always override
+  // per-update-url settings.
+  EXPECT_EQ(GetInstallationMode(kTargetExtension, kExampleUpdateUrl),
+            ExtensionManagement::INSTALLATION_ALLOWED);
+  EXPECT_EQ(GetInstallationMode(kTargetExtension2, kExampleUpdateUrl),
+            ExtensionManagement::INSTALLATION_FORCED);
+  EXPECT_EQ(GetInstallationMode(kTargetExtension3, kExampleUpdateUrl),
+            ExtensionManagement::INSTALLATION_RECOMMENDED);
+}
+
+// Tests the handling of blocked permissions in case it's specified in both
+// per-extension and per-update-url settings.
+TEST_F(ExtensionManagementServiceTest, BlockedPermissionsConflictionHandling) {

[Joao da Silva, 2014/11/04 14:06:10]

BlockedPermissionsConflictHandling

[binjin, 2014/11/05 12:52:42]

Done.

+  SetExampleDictPref();
+
+  // Both settings should be enforced.
+  APIPermissionSet blocked_permissions_for_update_url;
+  blocked_permissions_for_update_url.insert(APIPermission::kFileSystem);
+  blocked_permissions_for_update_url.insert(APIPermission::kBookmark);
+
+  APIPermissionSet api_permission_set;
+
+  api_permission_set = blocked_permissions_for_update_url;
+  api_permission_set.insert(APIPermission::kFileSystem);
+  api_permission_set.insert(APIPermission::kDownloads);
+  api_permission_set.insert(APIPermission::kBookmark);
+  EXPECT_EQ(api_permission_set,
+            GetBlockedAPIPermissions(kTargetExtension, kExampleUpdateUrl));
+
+  api_permission_set = blocked_permissions_for_update_url;
+  api_permission_set.insert(APIPermission::kDownloads);
+  EXPECT_EQ(api_permission_set,
+            GetBlockedAPIPermissions(kTargetExtension2, kExampleUpdateUrl));
+
+  api_permission_set = blocked_permissions_for_update_url;
+  api_permission_set.insert(APIPermission::kFileSystem);
+  api_permission_set.insert(APIPermission::kHistory);
+  EXPECT_EQ(api_permission_set,
+            GetBlockedAPIPermissions(kTargetExtension3, kExampleUpdateUrl));
 }
 
 // Tests functionality of new preference as to deprecate legacy
 // ExtensionInstallSources policy.
 TEST_F(ExtensionManagementServiceTest, NewInstallSources) {
   // Set the legacy preference, and verifies that it works.
   base::ListValue allowed_sites_pref;
   allowed_sites_pref.AppendString("https://www.example.com/foo");
   SetPref(
       true, pref_names::kAllowedInstallSites, allowed_sites_pref.DeepCopy());
@@ skipping 53 matching lines @@
 
 // Tests functionality of new preference as to deprecate legacy
 // ExtensionInstallBlacklist policy.
 TEST_F(ExtensionManagementServiceTest, NewInstallBlacklist) {
   // Set the new dictionary preference.
   {
     PrefUpdater updater(pref_service_.get());
     updater.SetBlacklistedByDefault(false);  // Allowed by default.
     updater.SetIndividualExtensionInstallationAllowed(kTargetExtension, false);
     updater.ClearPerExtensionSettings(kTargetExtension2);
-    updater.ClearPerExtensionSettings(kOtherExtension);
   }
   EXPECT_FALSE(extension_management_->BlacklistedByDefault());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 
   // Set legacy preference.
   base::ListValue denied_list_pref;
   denied_list_pref.AppendString("*");
   denied_list_pref.AppendString(kTargetExtension2);
   SetPref(true, pref_names::kInstallDenyList, denied_list_pref.DeepCopy());
 
   base::ListValue allowed_list_pref;
   allowed_list_pref.AppendString(kTargetExtension);
   SetPref(true, pref_names::kInstallAllowList, allowed_list_pref.DeepCopy());
 
   // Verifies that the new one have higher priority over the legacy ones.
   EXPECT_FALSE(extension_management_->BlacklistedByDefault());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
-  EXPECT_EQ(ReadById(kTargetExtension2)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension2),
             ExtensionManagement::INSTALLATION_BLOCKED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 }
 
 // Tests functionality of new preference as to deprecate legacy
 // ExtensionInstallWhitelist policy.
 TEST_F(ExtensionManagementServiceTest, NewInstallWhitelist) {
   // Set the new dictionary preference.
   {
     PrefUpdater updater(pref_service_.get());
     updater.SetBlacklistedByDefault(true);  // Disallowed by default.
     updater.SetIndividualExtensionInstallationAllowed(kTargetExtension, true);
     updater.ClearPerExtensionSettings(kTargetExtension2);
-    updater.ClearPerExtensionSettings(kOtherExtension);
   }
   EXPECT_TRUE(extension_management_->BlacklistedByDefault());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
 
   // Set legacy preference.
   base::ListValue denied_list_pref;
   denied_list_pref.AppendString(kTargetExtension);
   SetPref(true, pref_names::kInstallDenyList, denied_list_pref.DeepCopy());
 
   base::ListValue allowed_list_pref;
   allowed_list_pref.AppendString(kTargetExtension2);
   SetPref(true, pref_names::kInstallAllowList, allowed_list_pref.DeepCopy());
 
   // Verifies that the new one have higher priority over the legacy ones.
   EXPECT_TRUE(extension_management_->BlacklistedByDefault());
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
-  EXPECT_EQ(ReadById(kTargetExtension2)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension2),
             ExtensionManagement::INSTALLATION_ALLOWED);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_BLOCKED);
 }
 
 // Tests functionality of new preference as to deprecate legacy
 // ExtensionInstallForcelist policy.
 TEST_F(ExtensionManagementServiceTest, NewInstallForcelist) {
   // Set some legacy preferences, to verify that the new one overrides the
   // legacy ones.
   base::ListValue denied_list_pref;
   denied_list_pref.AppendString(kTargetExtension);
   SetPref(true, pref_names::kInstallDenyList, denied_list_pref.DeepCopy());
 
   // Set the new dictionary preference.
   {
     PrefUpdater updater(pref_service_.get());
     updater.SetIndividualExtensionAutoInstalled(
         kTargetExtension, kExampleUpdateUrl, true);
-    updater.ClearPerExtensionSettings(kOtherExtension);
   }
-  EXPECT_EQ(ReadById(kTargetExtension)->installation_mode,
+  EXPECT_EQ(GetInstallationModeById(kTargetExtension),
             ExtensionManagement::INSTALLATION_FORCED);
-  EXPECT_EQ(ReadById(kTargetExtension)->update_url, kExampleUpdateUrl);
-  EXPECT_EQ(ReadById(kOtherExtension)->installation_mode,
+  CheckAutomaticallyInstalledUpdateUrl(kTargetExtension, kExampleUpdateUrl);
+  EXPECT_EQ(GetInstallationModeById(kNonExistingExtension),
             ExtensionManagement::INSTALLATION_ALLOWED);
 }
 
 // Tests the behavior of IsInstallationExplicitlyAllowed().
 TEST_F(ExtensionManagementServiceTest, IsInstallationExplicitlyAllowed) {
   SetExampleDictPref();
 
   // Constant name indicates the installation_mode of extensions in example
   // preference.
   const char* allowed = kTargetExtension;
   const char* forced  = kTargetExtension2;
   const char* recommended = kTargetExtension3;
   const char* blocked = kTargetExtension4;
-  const char* not_specified = kOtherExtension;
+  const char* not_specified = kNonExistingExtension;
 
   // BlacklistedByDefault() is true in example preference.
   EXPECT_TRUE(extension_management_->IsInstallationExplicitlyAllowed(allowed));
   EXPECT_TRUE(extension_management_->IsInstallationExplicitlyAllowed(forced));
   EXPECT_TRUE(
       extension_management_->IsInstallationExplicitlyAllowed(recommended));
   EXPECT_FALSE(extension_management_->IsInstallationExplicitlyAllowed(blocked));
   EXPECT_FALSE(
       extension_management_->IsInstallationExplicitlyAllowed(not_specified));
 
@@ skipping 11 matching lines @@
   EXPECT_FALSE(extension_management_->IsInstallationExplicitlyAllowed(blocked));
   EXPECT_FALSE(
       extension_management_->IsInstallationExplicitlyAllowed(not_specified));
 }
 
 // Tests the flag value indicating that extensions are blacklisted by default.
 TEST_F(ExtensionAdminPolicyTest, BlacklistedByDefault) {
   EXPECT_FALSE(BlacklistedByDefault(NULL));
 
   base::ListValue blacklist;
-  blacklist.Append(new base::StringValue(kOtherExtension));
+  blacklist.Append(new base::StringValue(kNonExistingExtension));
   EXPECT_FALSE(BlacklistedByDefault(&blacklist));
   blacklist.Append(new base::StringValue("*"));
   EXPECT_TRUE(BlacklistedByDefault(&blacklist));
 
   blacklist.Clear();
   blacklist.Append(new base::StringValue("*"));
   EXPECT_TRUE(BlacklistedByDefault(&blacklist));
 }
 
 // Tests UserMayLoad for required extensions.
@@ skipping 123 matching lines @@
   EXPECT_FALSE(error.empty());
 
   CreateExtension(Manifest::INTERNAL);
   error.clear();
   EXPECT_FALSE(MustRemainEnabled(extension_.get(), NULL));
   EXPECT_FALSE(MustRemainEnabled(extension_.get(), &error));
   EXPECT_TRUE(error.empty());
 }
 
 }  // namespace extensions