Only show the checkbox for all urls when the switch is on

chrome/browser/extensions/extension_ui_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/command_line.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/strings/string_util.h"
 #include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/extensions/extension_util.h"
+#include "chrome/browser/extensions/permissions_updater.h"
 #include "chrome/browser/extensions/test_extension_system.h"
 #include "chrome/browser/ui/webui/extensions/extension_settings_handler.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/test/test_browser_thread.h"
+#include "extensions/browser/extension_registry.h"
 #include "extensions/browser/management_policy.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension.h"
+#include "extensions/common/extension_builder.h"
+#include "extensions/common/feature_switch.h"
+#include "extensions/common/id_util.h"
+#include "extensions/common/value_builder.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/users/scoped_test_user_manager.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/device_settings_service.h"
 #endif
 
 namespace extensions {
 
+namespace {
+const char kAllHostsPermission[] = "*://*/*";
+}
+
 class ExtensionUITest : public testing::Test {
  public:
   ExtensionUITest()
       : ui_thread_(content::BrowserThread::UI, &message_loop_),
         file_thread_(content::BrowserThread::FILE, &message_loop_)  {}
 
  protected:
   virtual void SetUp() OVERRIDE {
     // Create an ExtensionService and ManagementPolicy to inject into the
     // ExtensionSettingsHandler.
@@ skipping 19 matching lines @@
       const base::FilePath& path,
       std::string *error) {
     base::Value* value;
 
     JSONFileValueSerializer serializer(path);
     value = serializer.Deserialize(NULL, error);
 
     return static_cast<base::DictionaryValue*>(value);
   }
 
+  const Extension* CreateExtension(const std::string& name,
+                                   ListBuilder& permissions) {
+    const std::string kId = id_util::GenerateId(name);
+    scoped_refptr<const Extension> extension =
+        ExtensionBuilder().SetManifest(
+                               DictionaryBuilder()
+                                   .Set("name", name)
+                                   .Set("description", "an extension")
+                                   .Set("manifest_version", 2)
+                                   .Set("version", "1.0.0")
+                                   .Set("permissions", permissions))
+                          .SetLocation(Manifest::INTERNAL)
+                          .SetID(kId)
+                          .Build();
+
+    ExtensionRegistry::Get(profile())->AddEnabled(extension);
+    PermissionsUpdater(profile()).InitializePermissions(extension);
+    return extension;
+  }
+
   base::DictionaryValue* CreateExtensionDetailViewFromPath(
       const base::FilePath& extension_path,
       const std::vector<ExtensionPage>& pages,
       Manifest::Location location) {
     std::string error;
 
     base::FilePath manifest_path = extension_path.Append(kManifestFilename);
     scoped_ptr<base::DictionaryValue> extension_data(DeserializeJSONTestData(
         manifest_path, &error));
     EXPECT_EQ("", error);
@@ skipping 31 matching lines @@
          !field.IsAtEnd(); field.Advance()) {
       const base::Value* expected_value = &field.value();
       base::Value* actual_value = NULL;
       EXPECT_TRUE(actual_output_data->Get(field.key(), &actual_value)) <<
           field.key() + " is missing" + paths_details;
       EXPECT_TRUE(expected_value->Equals(actual_value)) << field.key() +
           paths_details;
     }
   }
 
+  Profile* profile() { return profile_.get(); }
+  ExtensionSettingsHandler* handler() { return handler_.get(); }
+
   base::MessageLoop message_loop_;
   content::TestBrowserThread ui_thread_;
   content::TestBrowserThread file_thread_;
   scoped_ptr<TestingProfile> profile_;
   ExtensionService* extension_service_;
   ManagementPolicy* management_policy_;
   scoped_ptr<ExtensionSettingsHandler> handler_;
 
 #if defined OS_CHROMEOS
   chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
@@ skipping 142 matching lines @@
   scoped_ptr<base::DictionaryValue> extension_details(
       CreateExtensionDetailViewFromPath(
           extension_path, pages, Manifest::UNPACKED));
 
   base::FilePath::StringType ui_path;
 
   EXPECT_TRUE(extension_details->GetString("path", &ui_path));
   EXPECT_EQ(extension_path, base::FilePath(ui_path));
 }
 
+// Test that the all_urls checkbox only shows up for extensions that want all
+// urls, and only when the switch is on.
+TEST_F(ExtensionUITest, ExtensionUIAllUrlsCheckbox) {
+  // Start with the switch enabled.
+  scoped_ptr<FeatureSwitch::ScopedOverride> enable_scripts_switch(
+      new FeatureSwitch::ScopedOverride(
+          FeatureSwitch::scripts_require_action(), true));
+  // Two extensions - one with all urls, one without.
+  const Extension* all_urls_extension =
+      CreateExtension("all_urls",
+                      ListBuilder().Append(kAllHostsPermission).Pass());
+  const Extension* no_urls_extension =
+      CreateExtension("no urls", ListBuilder().Pass());
+
+  scoped_ptr<base::DictionaryValue> value(
+      handler()->CreateExtensionDetailValue(
+          all_urls_extension, std::vector<ExtensionPage>(), NULL));
+  bool result = false;
+  const std::string kWantsAllUrls = "wantsAllUrls";
+  const std::string kAllowAllUrls = "allowAllUrls";
+
+  // The extension should want all urls, but not currently have it.
+  EXPECT_TRUE(value->GetBoolean(kWantsAllUrls, &result));
+  EXPECT_TRUE(result);
+  EXPECT_TRUE(value->GetBoolean(kAllowAllUrls, &result));
+  EXPECT_FALSE(result);
+
+  // Give the extension all urls.
+  util::SetAllowedScriptingOnAllUrls(
+      all_urls_extension->id(), profile(), true);
+
+  // Now the extension should both want and have all urls.
+  value.reset(handler()->CreateExtensionDetailValue(
+      all_urls_extension, std::vector<ExtensionPage>(), NULL));
+  EXPECT_TRUE(value->GetBoolean(kWantsAllUrls, &result));
+  EXPECT_TRUE(result);
+  EXPECT_TRUE(value->GetBoolean(kAllowAllUrls, &result));
+  EXPECT_TRUE(result);
+
+  // The other extension should neither want nor have all urls.
+  value.reset(handler()->CreateExtensionDetailValue(
+      no_urls_extension, std::vector<ExtensionPage>(), NULL));
+  EXPECT_TRUE(value->GetBoolean(kWantsAllUrls, &result));
+  EXPECT_FALSE(result);
+  EXPECT_TRUE(value->GetBoolean(kAllowAllUrls, &result));
+  EXPECT_FALSE(result);
+
+  // Turn off the switch and load another extension (so permissions are
+  // re-initialized).
+  enable_scripts_switch.reset();
+
+  // Even though the extension has the all urls preference, the checkbox
+  // shouldn't show up with the switch off.
+  value.reset(handler()->CreateExtensionDetailValue(
+      all_urls_extension, std::vector<ExtensionPage>(), NULL));
+  EXPECT_TRUE(value->GetBoolean(kWantsAllUrls, &result));
+  EXPECT_FALSE(result);
+  EXPECT_TRUE(value->GetBoolean(kAllowAllUrls, &result));
+  EXPECT_TRUE(result);
+
+  // Load another extension with all urls (so permissions get re-init'd).
+  all_urls_extension = CreateExtension(
+      "all_urls_II", ListBuilder().Append(kAllHostsPermission).Pass());
+
+  // Even though the extension has all_urls permission, the checkbox shouldn't
+  // show up without the switch.
+  value.reset(handler()->CreateExtensionDetailValue(
+      all_urls_extension, std::vector<ExtensionPage>(), NULL));
+  EXPECT_TRUE(value->GetBoolean(kWantsAllUrls, &result));
+  EXPECT_FALSE(result);
+  EXPECT_TRUE(value->GetBoolean(kAllowAllUrls, &result));
+  EXPECT_FALSE(result);
+}
+
 }  // namespace extensions