chrome.webRequest support for ExtensionSettings

chrome/browser/extensions/policy_handlers_unittest.cc

 // Copyright 2013 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 <utility>
 
 #include "base/json/json_reader.h"
+#include "base/strings/stringprintf.h"
+#include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/external_policy_loader.h"
 #include "chrome/browser/extensions/policy_handlers.h"
 #include "components/policy/core/browser/policy_error_map.h"
 #include "components/policy/core/common/policy_map.h"
 #include "components/policy/core/common/policy_types.h"
 #include "components/policy/core/common/schema.h"
 #include "components/policy/policy_constants.h"
 #include "components/prefs/pref_value_map.h"
 #include "extensions/browser/pref_names.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 10 matching lines @@
 const char kTestManagementPolicy2[] =
     "{"
     "  \"abcdefghijklmnopabcdefghijklmnop\": {"
     "    \"installation_mode\": \"force_installed\","
     "    \"update_url\": \"http://example.com/app\","
     "  },"
     "  \"*\": {"
     "    \"installation_mode\": \"blocked\","
     "  },"
     "}";
+const char kTestManagementPolicy3[] =
+    "{"
+    "  \"*\": {"
+    "    \"runtime_blocked_hosts\": [\"%s\"]"
+    "  }"
+    "}";
 
 TEST(ExtensionListPolicyHandlerTest, CheckPolicySettings) {
   base::ListValue list;
   policy::PolicyMap policy_map;
   policy::PolicyErrorMap errors;
   ExtensionListPolicyHandler handler(
       policy::key::kExtensionInstallBlacklist, kTestPref, true);
 
   policy_map.Set(policy::key::kExtensionInstallBlacklist,
                  policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
@@ skipping 228 matching lines @@
   ExtensionSettingsPolicyHandler handler(chrome_schema);
 
   policy_map.Set(policy::key::kExtensionSettings,
                  policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
                  policy::POLICY_SOURCE_CLOUD, std::move(policy_value), nullptr);
   // CheckPolicySettings() fails due to missing update URL.
   EXPECT_FALSE(handler.CheckPolicySettings(policy_map, &errors));
   EXPECT_FALSE(errors.empty());
 }
 
+TEST(ExtensionSettingsPolicyHandlerTest, CheckPolicySettingsURL) {
+  std::vector<std::string> good_urls = {
+      "*://*.example.com", "*://example.com", "http://cat.example.com",
+      "https://example.*", "*://*.example.*", "<all_urls>"};
+
+  // Invalid URLPattern or with a non-standard path
+  std::vector<std::string> bad_urls = {
+      "://*.example.com",       "*://example.com/cat*",  "*://example.com/",
+      "*://*.example.com/*cat", "*://example.com/cat/*", "bad",
+      "*://example.com/*"};
+
+  // Crafts and parses a ExtensionSettings policy to test URL parsing.
+  auto URLParsesSuccessfully = [](std::string url) {
+    std::string policy =
+        base::StringPrintf(kTestManagementPolicy3, url.c_str());
+    std::string error;
+    std::unique_ptr<base::Value> policy_value =
+        base::JSONReader::ReadAndReturnError(
+            policy, base::JSONParserOptions::JSON_ALLOW_TRAILING_COMMAS, NULL,
+            &error);
+    if (!policy_value.get())
+      return false;
+
+    policy::Schema chrome_schema =
+        policy::Schema::Wrap(policy::GetChromeSchemaData());
+    policy::PolicyMap policy_map;
+    policy::PolicyErrorMap errors;
+    ExtensionSettingsPolicyHandler handler(chrome_schema);
+
+    policy_map.Set(policy::key::kExtensionSettings,
+                   policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
+                   policy::POLICY_SOURCE_CLOUD, std::move(policy_value),
+                   nullptr);
+
+    return handler.CheckPolicySettings(policy_map, &errors) && errors.empty();
+
+  };
+
+  for (std::string url : good_urls) {
+    EXPECT_TRUE(URLParsesSuccessfully(url)) << url;
+  }
+  for (std::string url : bad_urls) {
+    EXPECT_FALSE(URLParsesSuccessfully(url)) << url;
+  }
+}
+
 TEST(ExtensionSettingsPolicyHandlerTest, ApplyPolicySettings) {
   std::string error;
   std::unique_ptr<base::Value> policy_value =
       base::JSONReader::ReadAndReturnError(
           kTestManagementPolicy2,
           base::JSONParserOptions::JSON_ALLOW_TRAILING_COMMAS, NULL, &error);
   ASSERT_TRUE(policy_value.get()) << error;
 
   policy::Schema chrome_schema =
       policy::Schema::Wrap(policy::GetChromeSchemaData());
   policy::PolicyMap policy_map;
   policy::PolicyErrorMap errors;
   PrefValueMap prefs;
   ExtensionSettingsPolicyHandler handler(chrome_schema);
 
   policy_map.Set(policy::key::kExtensionSettings,
                  policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
                  policy::POLICY_SOURCE_CLOUD, policy_value->CreateDeepCopy(),
                  nullptr);
   EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors));
   handler.ApplyPolicySettings(policy_map, &prefs);
   base::Value* value = NULL;
   ASSERT_TRUE(prefs.GetValue(pref_names::kExtensionManagement, &value));
   EXPECT_TRUE(base::Value::Equals(policy_value.get(), value));
 }
 
 }  // namespace extensions