Prevent unnecessary prompts when unpacked extensions use chrome.permissions.request.

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 2299 matching lines @@
   extensions::UnpackedInstaller::Create(service_)->Load(extension_path);
   loop_.RunAllPending();
 
   EXPECT_EQ(0u, GetErrors().size());
   ASSERT_EQ(1u, loaded_.size());
   EXPECT_EQ(Extension::LOAD, loaded_[0]->location());
   EXPECT_EQ(1u, service_->extensions()->size());
   EXPECT_EQ("1.0", loaded_[0]->VersionString());
 }
 
+#if !defined(OS_CHROMEOS)
+// LOAD extensions with plugins require approval.
+TEST_F(ExtensionServiceTest, LoadExtensionsWithPlugins) {
+  FilePath extension_with_plugin_path = data_dir_
+      .AppendASCII("good")
+      .AppendASCII("Extensions")
+      .AppendASCII(good1)
+      .AppendASCII("2");
+  FilePath extension_no_plugin_path = data_dir_
+      .AppendASCII("good")
+      .AppendASCII("Extensions")
+      .AppendASCII(good2)
+      .AppendASCII("1.0");
+
+  PluginService::GetInstance()->Init();
+  InitializeEmptyExtensionService();
+  InitializeExtensionProcessManager();
+  service_->set_show_extensions_prompts(true);
+
+  // Start by canceling any install prompts.
+  CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      switches::kAppsGalleryInstallAutoConfirmForTests,
+      "cancel");
+
+  // The extension that has a plugin should not install.
+  extensions::UnpackedInstaller::Create(service_)->Load(
+      extension_with_plugin_path);
+  loop_.RunAllPending();
+  EXPECT_EQ(0u, GetErrors().size());
+  EXPECT_EQ(0u, loaded_.size());
+  EXPECT_EQ(0u, service_->extensions()->size());
+  EXPECT_EQ(0u, service_->disabled_extensions()->size());
+
+  // But the extension with no plugin should since there's no prompt.
+  extensions::UnpackedInstaller::Create(service_)->Load(
+      extension_no_plugin_path);
+  loop_.RunAllPending();
+  EXPECT_EQ(0u, GetErrors().size());
+  EXPECT_EQ(1u, loaded_.size());
+  EXPECT_EQ(1u, service_->extensions()->size());
+  EXPECT_EQ(0u, service_->disabled_extensions()->size());
+  EXPECT_TRUE(service_->extensions()->Contains(good2));
+
+  // The plugin extension should install if we accept the dialog.
+  CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      switches::kAppsGalleryInstallAutoConfirmForTests,
+      "accept");
+
+  extensions::UnpackedInstaller::Create(service_)->Load(
+      extension_with_plugin_path);
+  loop_.RunAllPending();
+  EXPECT_EQ(0u, GetErrors().size());
+  EXPECT_EQ(2u, loaded_.size());
+  EXPECT_EQ(2u, service_->extensions()->size());
+  EXPECT_EQ(0u, service_->disabled_extensions()->size());
+  EXPECT_TRUE(service_->extensions()->Contains(good1));
+  EXPECT_TRUE(service_->extensions()->Contains(good2));
+
+  // Make sure the granted permissions have been setup.
+  scoped_refptr<ExtensionPermissionSet> permissions(
+      service_->extension_prefs()->GetGrantedPermissions(good1));
+  EXPECT_FALSE(permissions->IsEmpty());
+  EXPECT_TRUE(permissions->HasEffectiveFullAccess());
+  EXPECT_FALSE(permissions->apis().empty());
+  EXPECT_TRUE(permissions->HasAPIPermission(ExtensionAPIPermission::kPlugin));
+
+  // We should be able to reload the extension without getting another prompt.
+  loaded_.clear();
+  CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+      switches::kAppsGalleryInstallAutoConfirmForTests,
+      "cancel");
+
+  service_->ReloadExtension(good1);
+  loop_.RunAllPending();
+  EXPECT_EQ(1u, loaded_.size());
+  EXPECT_EQ(2u, service_->extensions()->size());
+  EXPECT_EQ(0u, service_->disabled_extensions()->size());
+}
+#endif
+
 namespace {
 
 bool IsExtension(const Extension& extension) {
   return extension.GetType() == Extension::TYPE_EXTENSION;
 }
 
 }  // namespace
 
 // Test adding a pending extension.
 TEST_F(ExtensionServiceTest, AddPendingExtensionFromSync) {
@@ skipping 2333 matching lines @@
   ASSERT_FALSE(AddPendingSyncInstall());
 
   // Wait for the external source to install.
   WaitForCrxInstall(crx_path_, INSTALL_NEW);
   ASSERT_TRUE(IsCrxInstalled());
 
   // Now that the extension is installed, sync request should fail
   // because the extension is already installed.
   ASSERT_FALSE(AddPendingSyncInstall());
 }